Files
Auto_Bangumi/backend/src/module/update/data_migration.py
2023-09-06 23:49:20 +08:00

25 lines
688 B
Python

from module.conf import LEGACY_DATA_PATH
from module.rss import RSSEngine
from module.models import Bangumi
from module.utils import json_config
def data_migration():
if not LEGACY_DATA_PATH.exists():
return False
old_data = json_config.load(LEGACY_DATA_PATH)
infos = old_data["bangumi_info"]
rss_link = old_data["rss_link"]
new_data = []
for info in infos:
new_data.append(Bangumi(**info, rss_link=rss_link))
with RSSEngine() as engine:
engine.bangumi.add_all(new_data)
engine.add_rss(rss_link)
LEGACY_DATA_PATH.unlink(missing_ok=True)
def database_migration():
with RSSEngine() as engine:
engine.migrate()