♻️ 使用asdict替代__dict__以序列化数据类

Author: shadow3aaa

src/nonebot_plugin_nyaturingtest/image_manager.py

@@ -1,5 +1,5 @@
 import base64
-from dataclasses import dataclass
+from dataclasses import asdict, dataclass
 import hashlib
 import io
 import json
@@ -40,7 +40,7 @@ def to_json(self) -> str:
         """
         将对象转换为JSON字符串
         """
-        return json.dumps(self.__dict__, ensure_ascii=False)
+        return json.dumps(asdict(self), ensure_ascii=False)
 
     @staticmethod
     def from_json(json_str: str) -> "ImageWithDescription":

src/nonebot_plugin_nyaturingtest/presets.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass, field
+from dataclasses import asdict, dataclass, field
 import json
 import os
 
@@ -65,7 +65,7 @@ def _load_presets_from_directory(directory: str = f"{store.get_plugin_config_dir
     if not os.path.exists(directory):
         os.makedirs(directory)
         with open(os.path.join(directory, "喵喵.json"), "w", encoding="utf-8") as f:
-            json.dump(_猫娘预设.__dict__, f, ensure_ascii=False, indent=4)
+            json.dump(asdict(_猫娘预设), f, ensure_ascii=False, indent=4)
     for filename in os.listdir(directory):
         if filename.endswith(".json"):
             path = os.path.join(directory, filename)

src/nonebot_plugin_nyaturingtest/session.py

@@ -1,6 +1,6 @@
 from collections import deque
 from collections.abc import Awaitable, Callable
-from dataclasses import dataclass
+from dataclasses import asdict, dataclass
 from datetime import datetime
 from enum import Enum
 import json
@@ -420,7 +420,7 @@ async def __feedback_stage(
             [
                 {
                     "user_name": profile.user_id,
-                    "emotion_tends_to_user": profile.emotion.__dict__,
+                    "emotion_tends_to_user": asdict(profile.emotion),
                 }
                 for profile in related_profiles
             ],
@@ -751,7 +751,7 @@ async def __chat_stage(
             [
                 {
                     "user_name": profile.user_id,
-                    "emotion_tends_to_user": profile.emotion.__dict__,
+                    "emotion_tends_to_user": asdict(profile.emotion),
                 }
                 for profile in related_profiles
             ],