fix: old data support problem.

This commit is contained in:
EstrellaXD
2023-08-07 20:14:45 +08:00
parent 84d5dbaceb
commit 1c4e8dc293
18 changed files with 136 additions and 66 deletions

View File

@@ -1 +1,2 @@
from .data_migration import data_migration
from .startup import start_up

View File

@@ -1,7 +1,5 @@
import os
from module.conf import LEGACY_DATA_PATH
from module.database import Database
from module.rss import RSSEngine
from module.models import Bangumi
from module.utils import json_config
@@ -15,8 +13,9 @@ def data_migration():
new_data = []
for info in infos:
new_data.append(Bangumi(**info, rss_link=[rss_link]))
with Database() as db:
db.create_table()
db.bangumi.add_all(new_data)
with RSSEngine() as engine:
engine.create_table()
engine.bangumi.add_all(new_data)
engine.user.add_default_user()
engine.add_rss(rss_link)
LEGACY_DATA_PATH.unlink(missing_ok=True)

View File

@@ -0,0 +1,23 @@
import logging
from module.rss import RSSEngine
from module.conf import settings
logger = logging.getLogger(__name__)
def start_up(first_run):
with RSSEngine() as engine:
engine.create_table()
engine.user.add_default_user()
if not first_run:
main_rss = engine.rss.search_id(1)
if not main_rss:
engine.add_rss(settings.rss_link, name="Mikan RSS", combine=True)
elif main_rss.url != settings.rss_link:
main_rss.url = settings.rss_link
engine.rss.update(main_rss)
if __name__ == "__main__":
start_up(False)