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

use mongodb search

Browse Source
This commit is contained in:
Benny
2024-12-06 22:28:08 +01:00
parent 2d5d55bf0c
commit 87ada1229f
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
yyetsweb/databases/resources.py
View File

@@ -47,7 +47,6 @@ class Resource(SearchEngine):
def search_resource(self, keyword: str, search_type: "str") -> dict:
# search_type: default,subtitle,douban,comment
# TODO
if os.getenv("MEILISEARCH"):
return self.meili_search(keyword, search_type)
else:
@@ -63,7 +62,18 @@ class Resource(SearchEngine):
comment_data = hide_phone(self.search_comment(keyword))
if search_type == "subtitle":
subtitle_data = self.search_subtitle(keyword)
# TODO: just get data from mongodb for now.
subtitle_data = list(
self.db["subtitle"].find(
{
"$or": [
{"cnname": {"$regex": f".*{keyword}.*", "$options": "i"}},
{"enname": {"$regex": f".*{keyword}.*", "$options": "i"}},
]
},
{"_id": False},
)
)
if search_type == "default":
resource_data = self.search_yyets(keyword)