injectlib/YYeTsBot
1
0
Fork 0
mirror of https://github.com/tgbot-collection/YYeTsBot.git synced 2025-11-25 11:29:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

bot add search in comments

Browse Source
This commit is contained in:
BennyThink
2021-12-11 14:18:23 +08:00
parent cf9fc350ba
commit 4cf1cfcfc5
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
yyetsbot/fansub.py
View File

@@ -172,7 +172,8 @@ class YYeTsOffline(BaseFansub):
def __init__(self, db="zimuzu", col="yyets"): def __init__(self, db="zimuzu", col="yyets"):
super().__init__() super().__init__()
self.mongo = pymongo.MongoClient(host=MONGO) self.mongo = pymongo.MongoClient(host=MONGO)
self.collection = self.mongo[db][col] self.db = self.mongo[db]
self.collection = self.db[col]
@Redis.preview_cache(60) @Redis.preview_cache(60)
def search_preview(self, search_text: str) -> dict: def search_preview(self, search_text: str) -> dict:
@@ -199,7 +200,20 @@ class YYeTsOffline(BaseFansub):
"class": self.__class__.__name__ "class": self.__class__.__name__
} }
logging.info("[%s] Offline search complete", self.__class__.__name__) logging.info("[%s] Offline resource search complete", self.__class__.__name__)
comments = self.db["comment"].find({"content": {"$regex": f".*{search_text}.*", "$options": "-i"}})
for c in comments:
url = "https://yyets.dmesg.app/resource.html?id={}#{}".format(c["resource_id"], str(c["_id"]))
url_hash = hashlib.sha1(url.encode('u8')).hexdigest()
all_name = c["content"]
results[url_hash] = {
"name": all_name,
"url": url,
"class": self.__class__.__name__
}
logging.info("[%s] Offline comment search complete", self.__class__.__name__)
results["class"] = self.__class__.__name__ results["class"] = self.__class__.__name__
return results return results