mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-05-11 10:35:50 +08:00
fix: old data support problem.
This commit is contained in:
@@ -1 +1,2 @@
|
||||
from .data_migration import data_migration
|
||||
from .startup import start_up
|
||||
|
||||
@@ -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)
|
||||
|
||||
23
backend/src/module/update/startup.py
Normal file
23
backend/src/module/update/startup.py
Normal 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)
|
||||
Reference in New Issue
Block a user