diff --git a/docker-compose.yml b/docker-compose.yml index de0ad55..825cfac 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -38,3 +38,5 @@ services: - ./sg_data/mongodb:/data/db logging: driver: none + ports: + - "127.0.0.1:27017:27017" diff --git a/searchgram/__init__.py b/searchgram/__init__.py index ec35395..c2ba6f2 100644 --- a/searchgram/__init__.py +++ b/searchgram/__init__.py @@ -7,6 +7,8 @@ from config import ENGINE if ENGINE == "meili": + print("Using MeiliSearch as search engine") from meili import SearchEngine elif ENGINE == "mongo": + print("Using MongoDB as search engine") from mongo import SearchEngine diff --git a/searchgram/mongo.py b/searchgram/mongo.py index 3fd136f..1322734 100644 --- a/searchgram/mongo.py +++ b/searchgram/mongo.py @@ -23,14 +23,16 @@ class SearchEngine(BasicSearchEngine): def __init__(self): self.client = pymongo.MongoClient(host=MONGO_HOST, connect=False, connectTimeoutMS=5000, serverSelectionTimeoutMS=5000) self.db = self.client["telegram"] - self.col = self.db["chat"] - self.history = self.db["history"] + self.chat = self.db["chat"] def __del__(self): self.client.close() def upsert(self, message): - pass + if self.check_ignore(message): + return + data = self.set_uid(message) + self.chat.update_one({"ID": data["ID"]}, {"$set": data}, upsert=True) def search(self, keyword, _type=None, user=None, page=1, mode=None): pass