feat: store deleted messages on our DB

This commit is contained in:
2025-11-29 22:26:43 +01:00
parent 8d215f56f3
commit 0b5308d1e9
4 changed files with 132 additions and 54 deletions

View File

@@ -1,13 +1,12 @@
import os
from bot.log import logger
from bot.mongodb.load_db import load_chat_messages_from_db
from database import get_patterns
from pymongo import MongoClient
MONGO_URI = os.getenv("MONGO_URI")
message_patterns = get_patterns()
def update_patterns(regex: str, response: str):
"""update patterns in memory."""
@@ -15,55 +14,8 @@ def update_patterns(regex: str, response: str):
logger.info(f"Pattern added in memory: {regex}")
def load_chat_messages_from_db(
mongo_uri="mongodb://localhost:27017",
database="discord_bot",
collection="messages",
):
"""
Loads all chat messages from MongoDB.
# load global variables
Args:
mongo_uri (str): MongoDB connection URI
database (str): Name of the MongoDB database
collection (str): Name of the collection
message_patterns = get_patterns()
Returns:
list: list of message strings
"""
try:
client = MongoClient(mongo_uri)
db = client[database]
col = db[collection]
logger.debug(
f"Connecting to MongoDB at {mongo_uri}, DB='{database}', Collection='{collection}'"
)
cursor = col.find({}, {"message": 1})
messages = [doc["message"] for doc in cursor if "message" in doc]
logger.info(f"Loaded {len(messages)} messages from MongoDB")
return messages
except Exception as e:
logger.error(f"Failed to load messages from MongoDB: {e}")
return []
def load_mongodb():
messages = []
if not MONGO_URI:
logger.error("MONGO_URI is not set. Please contact the administrator.")
return
messages = load_chat_messages_from_db(MONGO_URI)
if not messages:
logger.warning("messages collection is empty after loading from MongoDB!")
return messages
schizo_messages = load_mongodb()
schizo_messages = load_chat_messages_from_db()