Skip to content

Commit

Permalink
fix find meme name
Browse files Browse the repository at this point in the history
  • Loading branch information
MeetWq committed Sep 8, 2024
1 parent 206fcc2 commit ad65f14
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 19 deletions.
30 changes: 12 additions & 18 deletions nonebot_plugin_memes_api/manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,32 +163,26 @@ def __dump(self):

def __refresh_names(self):
self.__meme_names = {}

def add(key: str, meme: MemeInfo):
key = key.lower()
if key not in self.__meme_names:
self.__meme_names[key] = []
self.__meme_names[key].append(meme)

for meme in self.__meme_dict.values():
add(meme.key, meme)
names = set()
names.add(meme.key.lower())
for keyword in meme.keywords:
add(keyword, meme)
names.add(keyword.lower())
for shortcut in meme.shortcuts:
add(shortcut.humanized or shortcut.key, meme)
names.add((shortcut.humanized or shortcut.key).lower())
for name in names:
if name not in self.__meme_names:
self.__meme_names[name] = []
self.__meme_names[name].append(meme)

def __refresh_tags(self):
self.__meme_tags = {}

def add(tag: str, meme: MemeInfo):
tag = tag.lower()
if tag not in self.__meme_tags:
self.__meme_tags[tag] = []
self.__meme_tags[tag].append(meme)

for meme in self.__meme_dict.values():
for tag in meme.tags:
add(tag, meme)
tag = tag.lower()
if tag not in self.__meme_tags:
self.__meme_tags[tag] = []
self.__meme_tags[tag].append(meme)


meme_manager = MemeManager()
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "nonebot_plugin_memes_api"
version = "0.4.9"
version = "0.4.10"
description = "Nonebot2 plugin for making memes"
authors = ["meetwq <[email protected]>"]
license = "MIT"
Expand Down

0 comments on commit ad65f14

Please sign in to comment.