klesh/ss
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: handle Pydantic v2 model serialization in save_subscriptions method

Browse Source
This commit is contained in:
Klesh Wong
2026-03-06 11:25:41 +08:00
parent 9466fb73ef
commit 9309390182
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ss/storage.py
View File

@ -53,7 +53,11 @@ class StorageManager:
try: try:
with open(self.subscriptions_file, 'w', encoding='utf-8') as f: with open(self.subscriptions_file, 'w', encoding='utf-8') as f:
# Convert Pydantic model to dict for YAML serialization # Convert Pydantic model to dict for YAML serialization
data = subscriptions.dict() if hasattr(subscriptions, 'model_dump'):
data = subscriptions.model_dump(mode='json')
else:
# Fallback for Pydantic v1
data = subscriptions.dict()
yaml.dump(data, f, default_flow_style=False, allow_unicode=True) yaml.dump(data, f, default_flow_style=False, allow_unicode=True)
return True return True
except (yaml.YAMLError, IOError, ValueError) as e: except (yaml.YAMLError, IOError, ValueError) as e: