suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-07 20:33:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

use deprecation marker for legacy db

Browse Source
This commit is contained in:
felinae98 2022-10-10 22:29:11 +08:00
parent 9a033aed1d
commit b34f93b313
No known key found for this signature in database
GPG Key ID: 00C8B010587FF610
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/plugins/nonebot_bison/config/config_legacy.py
View File

@ -1,5 +1,7 @@
import json
import os import os
from collections import defaultdict from collections import defaultdict
from datetime import datetime
from os import path from os import path
from pathlib import Path from pathlib import Path
from typing import DefaultDict, Literal, Mapping, TypedDict from typing import DefaultDict, Literal, Mapping, TypedDict
@ -17,7 +19,7 @@ from .utils import NoSuchSubscribeException, NoSuchUserException
supported_target_type = platform_manager.keys() supported_target_type = platform_manager.keys()
def get_config_path() -> str: def get_config_path() -> tuple[str, str]:
if plugin_config.bison_config_path: if plugin_config.bison_config_path:
data_dir = plugin_config.bison_config_path data_dir = plugin_config.bison_config_path
else: else:
@ -27,9 +29,10 @@ def get_config_path() -> str:
os.makedirs(data_dir) os.makedirs(data_dir)
old_path = path.join(data_dir, "hk_reporter.json") old_path = path.join(data_dir, "hk_reporter.json")
new_path = path.join(data_dir, "bison.json") new_path = path.join(data_dir, "bison.json")
deprecated_maker_path = path.join(data_dir, "bison.json.deprecated")
if os.path.exists(old_path) and not os.path.exists(new_path): if os.path.exists(old_path) and not os.path.exists(new_path):
os.rename(old_path, new_path) os.rename(old_path, new_path)
return new_path return new_path, deprecated_maker_path
def drop(): def drop():
@ -39,11 +42,15 @@ def drop():
working_dir = os.getcwd() working_dir = os.getcwd()
data_dir = path.join(working_dir, "data") data_dir = path.join(working_dir, "data")
old_path = path.join(data_dir, "bison.json") old_path = path.join(data_dir, "bison.json")
new_path = path.join(data_dir, "bison-legacy.json") deprecated_marker_path = path.join(data_dir, "bison.json.deprecated")
if os.path.exists(old_path): if os.path.exists(old_path):
config.db.close() config.db.close()
config.available = False config.available = False
os.rename(old_path, new_path) with open(deprecated_marker_path, "w") as file:
content = {
"migration_time": datetime.now().isoformat(),
}
file.write(json.dumps(content))
return True return True
return False return False
@ -71,10 +78,12 @@ class Config(metaclass=Singleton):
self._do_init() self._do_init()
def _do_init(self): def _do_init(self):
path = get_config_path() path, deprecated_marker_path = get_config_path()
if Path(path).exists(): if Path(deprecated_marker_path).exists():
self.available = False
elif Path(path).exists():
self.available = True self.available = True
self.db = TinyDB(get_config_path(), encoding="utf-8") self.db = TinyDB(path, encoding="utf-8")
self.kv_config = self.db.table("kv") self.kv_config = self.db.table("kv")
self.user_target = self.db.table("user_target") self.user_target = self.db.table("user_target")
self.target_user_cache: dict[str, defaultdict[Target, list[User]]] = {} self.target_user_cache: dict[str, defaultdict[Target, list[User]]] = {}