From ad65f147e73d1a18fcf913004c9ceb1f22509b1e Mon Sep 17 00:00:00 2001
From: meetwq <meetwq@gmail.com>
Date: Sun, 8 Sep 2024 14:13:05 +0800
Subject: [PATCH] fix find meme name

---
 nonebot_plugin_memes_api/manager.py | 30 ++++++++++++-----------------
 pyproject.toml                      |  2 +-
 2 files changed, 13 insertions(+), 19 deletions(-)

diff --git a/nonebot_plugin_memes_api/manager.py b/nonebot_plugin_memes_api/manager.py
index 6e4722d..d1b2e5f 100644
--- a/nonebot_plugin_memes_api/manager.py
+++ b/nonebot_plugin_memes_api/manager.py
@@ -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()
diff --git a/pyproject.toml b/pyproject.toml
index 4fa8029..50dd700 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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 <meetwq@gmail.com>"]
 license = "MIT"