Fix rss token can not be hot changed

This commit is contained in:
EstrellaXD
2023-04-26 21:36:50 +08:00
parent c07d9466d5
commit cfe1fba9c7
5 changed files with 26 additions and 29 deletions

View File

@@ -3,7 +3,7 @@ import time
import logging
from module.conf import settings, setup_logger, LOG_PATH, DATA_PATH, RSSLink, VERSION
from module.utils import load_program_data, save_program_data, json_config
from module.utils import load_program_data, save_program_data
from module.core import DownloadClient
from module.manager import Renamer, FullSeasonGet
@@ -12,7 +12,6 @@ from module.models import ProgramData
logger = logging.getLogger(__name__)
RSS_LINK = RSSLink()
def reset_log():
@@ -20,9 +19,9 @@ def reset_log():
os.remove(LOG_PATH)
def load_data_file() -> ProgramData:
def load_data_file(rss_link: str) -> ProgramData:
empty_data = ProgramData(
rss_link=RSS_LINK,
rss_link=rss_link,
data_version=settings.data_version,
)
if not os.path.exists(DATA_PATH):
@@ -31,20 +30,19 @@ def load_data_file() -> ProgramData:
logger.info("Building data information...")
else:
program_data = load_program_data(DATA_PATH)
if program_data.rss_link != RSS_LINK or program_data.data_version != settings.data_version:
if program_data.rss_link != rss_link or program_data.data_version != settings.data_version:
program_data = empty_data
logger.info("Rebuilding data information...")
return program_data
def main_process(program_data: ProgramData, download_client: DownloadClient):
rename = Renamer(download_client)
rss_analyser = RSSAnalyser()
while True:
times = 0
if settings.rss_parser.enable:
rss_analyser.run(program_data.bangumi_info, download_client)
rss_analyser.run(program_data.bangumi_info, download_client, program_data.rss_link)
if settings.bangumi_manage.eps_complete and program_data.bangumi_info != []:
FullSeasonGet().eps_complete(program_data.bangumi_info, download_client)
logger.info("Running....")
@@ -68,16 +66,17 @@ def show_info():
def run():
# 初始化
settings.reload()
rss_link = RSSLink()
reset_log()
setup_logger()
show_info()
download_client = DownloadClient()
download_client.auth()
download_client.init_downloader()
if settings.rss_parser.token in ["", "token", None]:
logger.error("Please set your RSS token in config file.")
exit(1)
download_client.rss_feed()
bangumi_data = load_data_file()
download_client = DownloadClient()
download_client.auth()
download_client.init_downloader()
download_client.rss_feed(rss_link)
bangumi_data = load_data_file(rss_link)
# 主程序循环
main_process(bangumi_data, download_client)