✨ 轻雪天气更新

liteyuki/utils/base/config.py

@@ -4,7 +4,7 @@ import nonebot
 import yaml
 from pydantic import BaseModel
 
-from .data_manager import StoredConfig, common_db
+from .data_manager import StoredConfig, TempConfig, common_db
 from .ly_typing import T_Bot
 from ..message.tools import random_hex_string
 
@@ -64,6 +64,12 @@ def get_config(key: str, default=None):
         return default
 
 
+def set_stored_config(key: str, value):
+    temp_config: TempConfig = common_db.first(TempConfig(), default=TempConfig())
+    temp_config.data[key] = value
+    common_db.save(temp_config)
+
+
 def init_conf(conf: dict) -> dict:
     """
     初始化配置文件，确保配置文件中的必要字段存在，且不会冲突

liteyuki/utils/base/data.py

@@ -78,7 +78,7 @@ class Database:
         else:
             return [model_type(**self._load(dict(zip(fields, result)))) for result in results]
 
-    def upsert(self, *args: LiteModel):
+    def save(self, *args: LiteModel):
         """增/改操作
         Args:
             *args:

liteyuki/utils/base/data_manager.py

@@ -11,6 +11,11 @@ group_db: Database = Database(os.path.join(DATA_PATH, "groups.ldb"))
 plugin_db: Database = Database(os.path.join(DATA_PATH, "plugins.ldb"))
 common_db: Database = Database(os.path.join(DATA_PATH, "common.ldb"))
 
+# 内存数据库，临时用于存储数据
+memory_database = {
+
+}
+
 
 class User(LiteModel):
     TABLE_NAME = "user"
@@ -60,3 +65,31 @@ def auto_migrate():
     group_db.auto_migrate(Group())
     plugin_db.auto_migrate(InstalledPlugin(), GlobalPlugin())
     common_db.auto_migrate(GlobalPlugin(), StoredConfig(), TempConfig())
+
+
+def set_memory_data(key: str, value) -> None:
+    """
+    设置内存数据库的数据，类似于redis
+    Args:
+        key:
+        value:
+
+    Returns:
+
+    """
+    return memory_database.update({
+                                          key: value
+                                          })
+
+
+def get_memory_data(key: str, default=None) -> any:
+    """
+    获取内存数据库的数据,类似于redis
+    Args:
+        key:
+        default:
+
+    Returns:
+
+    """
+    return memory_database.get(key, default)

liteyuki/utils/base/language.py

@@ -149,7 +149,7 @@ def change_user_lang(user_id: str, lang_code: str):
     """
     user = user_db.first(User(), "user_id = ?", user_id, default=User(user_id=user_id))
     user.profile["lang"] = lang_code
-    user_db.upsert(user)
+    user_db.save(user)
     _user_lang[user_id] = lang_code
 
 
@@ -160,6 +160,7 @@ def get_user_lang(user_id: str) -> Language:
     user_id = str(user_id)
 
     if user_id not in _user_lang:
+        nonebot.logger.debug(f"Loading user language for {user_id}")
         user = user_db.first(
             User(), "user_id = ?", user_id, default=User(
                 user_id=user_id,

liteyuki/utils/base/ly_typing.py

@@ -4,4 +4,4 @@ T_Bot = v11.Bot | v12.Bot
 T_GroupMessageEvent = v11.GroupMessageEvent | v12.GroupMessageEvent
 T_PrivateMessageEvent = v11.PrivateMessageEvent | v12.PrivateMessageEvent
 T_MessageEvent = v11.MessageEvent | v12.MessageEvent
-T_Message = v11.Message | v12.Message
+T_Message = v11.Message | v12.Message