import json import os from werkzeug.security import generate_password_hash from app.helper import DbHelper from app.plugins import PluginManager from app.utils import StringUtils, ExceptionUtils from config import Config def check_config(): """ 检查配置文件，如有错误进行日志输出 """ # 检查日志输出 if Config().get_config('app'): logtype = Config().get_config('app').get('logtype') if logtype: print("日志输出类型为：%s" % logtype) if logtype == "server": logserver = Config().get_config('app').get('logserver') if not logserver: print("【Config】日志中心地址未配置，无法正常输出日志") else: print("日志将上送到服务器：%s" % logserver) elif logtype == "file": logpath = Config().get_config('app').get('logpath') if not logpath: print("【Config】日志文件路径未配置，无法正常输出日志") else: print("日志将写入文件：%s" % logpath) # 检查WEB端口 web_port = Config().get_config('app').get('web_port') if not web_port: print("WEB服务端口未设置，将使用默认3000端口") # 检查登录用户和密码 login_user = Config().get_config('app').get('login_user') login_password = Config().get_config('app').get('login_password') if not login_user or not login_password: print("WEB管理用户或密码未设置，将使用默认用户：admin，密码：password") else: print("WEB管理页面用户：%s" % str(login_user)) # 检查HTTPS ssl_cert = Config().get_config('app').get('ssl_cert') ssl_key = Config().get_config('app').get('ssl_key') if not ssl_cert or not ssl_key: print("未启用https，请使用 http://IP:%s 访问管理页面" % str(web_port)) else: if not os.path.exists(ssl_cert): print("ssl_cert文件不存在：%s" % ssl_cert) if not os.path.exists(ssl_key): print("ssl_key文件不存在：%s" % ssl_key) print("已启用https，请使用 https://IP:%s 访问管理页面" % str(web_port)) rmt_tmdbkey = Config().get_config('app').get('rmt_tmdbkey') if not rmt_tmdbkey: print("TMDB API Key未配置，媒体整理、搜索下载等功能将无法正常运行！") rmt_match_mode = Config().get_config('app').get('rmt_match_mode') if rmt_match_mode: rmt_match_mode = rmt_match_mode.upper() else: rmt_match_mode = "NORMAL" if rmt_match_mode == "STRICT": print("TMDB匹配模式：严格模式") else: print("TMDB匹配模式：正常模式") else: print("配置文件格式错误，找不到app配置项！") # 检查媒体库目录路径 if Config().get_config('media'): media_server = Config().get_config('media').get('media_server') if media_server: print("媒体管理软件设置为：%s" % media_server) if media_server == "jellyfin": if not Config().get_config('jellyfin'): print("jellyfin未配置") else: if not Config().get_config('jellyfin').get('host') \ or not Config().get_config('jellyfin').get('api_key'): print("jellyfin配置不完整") elif media_server == "plex": if not Config().get_config('plex'): print("plex未配置") else: if not Config().get_config('plex').get('token') \ and not Config().get_config('plex').get('username'): print("plex配置不完整") else: if not Config().get_config('emby'): print("emby未配置") else: if not Config().get_config('emby').get('host') \ or not Config().get_config('emby').get('api_key'): print("emby配置不完整") movie_paths = Config().get_config('media').get('movie_path') if not movie_paths: print("未配置电影媒体库目录") else: if not isinstance(movie_paths, list): movie_paths = [movie_paths] for movie_path in movie_paths: if not os.path.exists(movie_path): print("电影媒体库目录不存在：%s" % movie_path) tv_paths = Config().get_config('media').get('tv_path') if not tv_paths: print("未配置电视剧媒体库目录") else: if not isinstance(tv_paths, list): tv_paths = [tv_paths] for tv_path in tv_paths: if not os.path.exists(tv_path): print("电视剧媒体库目录不存在：%s" % tv_path) anime_paths = Config().get_config('media').get('anime_path') if anime_paths: if not isinstance(anime_paths, list): anime_paths = [anime_paths] for anime_path in anime_paths: if not os.path.exists(anime_path): print("动漫媒体库目录不存在：%s" % anime_path) category = Config().get_config('media').get('category') if not category: print("未配置分类策略") else: print("配置文件格式错误，找不到media配置项！") # 检查站点配置 if Config().get_config('pt'): pt_client = Config().get_config('pt').get('pt_client') print("下载软件设置为：%s" % pt_client) rmt_mode = Config().get_config('pt').get('rmt_mode', 'copy') if rmt_mode == "link": print("默认文件转移模式为：硬链接") elif rmt_mode == "softlink": print("默认文件转移模式为：软链接") elif rmt_mode == "move": print("默认文件转移模式为：移动") elif rmt_mode == "rclone": print("默认文件转移模式为：rclone移动") elif rmt_mode == "rclonecopy": print("默认文件转移模式为：rclone复制") else: print("默认文件转移模式为：复制") search_indexer = Config().get_config('pt').get('search_indexer') if search_indexer: print("索引器设置为：%s" % search_indexer) search_auto = Config().get_config('pt').get('search_auto') if search_auto: print("微信等移动端渠道搜索已开启自动择优下载") ptsignin_cron = Config().get_config('pt').get('ptsignin_cron') if not ptsignin_cron: print("站点自动签到时间未配置，站点签到功能已关闭") pt_check_interval = Config().get_config('pt').get('pt_check_interval') if not pt_check_interval: print("RSS订阅周期未配置，RSS订阅功能已关闭") pt_monitor = Config().get_config('pt').get('pt_monitor') if not pt_monitor: print("下载软件监控未开启，下载器监控功能已关闭") else: print("配置文件格式错误，找不到pt配置项！") # 检查Douban配置 if not Config().get_config('douban'): print("豆瓣未配置") else: if not Config().get_config('douban').get('users') \ or not Config().get_config('douban').get('types') \ or not Config().get_config('douban').get('days'): print("豆瓣配置不完整") def update_config(): """ 升级配置文件 """ _config = Config().get_config() _dbhelper = DbHelper() overwrite_cofig = False # 密码初始化 login_password = _config.get("app", {}).get("login_password") or "password" if login_password and not login_password.startswith("[hash]"): _config['app']['login_password'] = "[hash]%s" % generate_password_hash( login_password) overwrite_cofig = True # 实验室配置初始化 if not _config.get("laboratory"): _config['laboratory'] = { 'search_keyword': False, 'tmdb_cache_expire': True, 'use_douban_titles': False, 'search_en_title': True, 'chrome_browser': False } overwrite_cofig = True # 安全配置初始化 if not _config.get("security"): _config['security'] = { 'media_server_webhook_allow_ip': { 'ipv4': '0.0.0.0/0', 'ipv6': '::/0' }, 'telegram_webhook_allow_ip': { 'ipv4': '127.0.0.1', 'ipv6': '::/0' } } overwrite_cofig = True # Synology Chat安全配置初始化 if not _config.get("security", {}).get("synology_webhook_allow_ip"): _config['security']['synology_webhook_allow_ip'] = { 'ipv4': '127.0.0.1', 'ipv6': '::/0' } overwrite_cofig = True # API密钥初始化 if not _config.get("security", {}).get("api_key"): _config['security']['api_key'] = _config.get("security", {}).get("subscribe_token") \ or StringUtils.generate_random_str() if _config.get('security', {}).get('subscribe_token'): _config['security'].pop('subscribe_token') overwrite_cofig = True # 刮削NFO配置初始化 if not _config.get("scraper_nfo"): _config['scraper_nfo'] = { "movie": { "basic": True, "credits": True, "credits_chinese": False}, "tv": { "basic": True, "credits": True, "credits_chinese": False, "season_basic": True, "episode_basic": True, "episode_credits": True} } overwrite_cofig = True # 刮削图片配置初始化 if not _config.get("scraper_pic"): _config['scraper_pic'] = { "movie": { "poster": True, "backdrop": True, "background": True, "logo": True, "disc": True, "banner": True, "thumb": True}, "tv": { "poster": True, "backdrop": True, "background": True, "logo": True, "clearart": True, "banner": True, "thumb": True, "season_poster": True, "season_banner": True, "season_thumb": True, "episode_thumb": False, "episode_thumb_ffmpeg": False} } overwrite_cofig = True # 下载目录配置初始化 if not _config.get('downloaddir'): dl_client = _config.get('pt', {}).get('pt_client') if dl_client and _config.get(dl_client): save_path = _config.get(dl_client).get('save_path') if not isinstance(save_path, dict): save_path = {"movie": save_path, "tv": save_path, "anime": save_path} container_path = _config.get(dl_client).get('save_containerpath') if not isinstance(container_path, dict): container_path = {"movie": container_path, "tv": container_path, "anime": container_path} downloaddir = [] type_dict = {"movie": "电影", "tv": "电视剧", "anime": "动漫"} for mtype, path in save_path.items(): if not path: continue save_dir = path.split('|')[0] save_label = None if len(path.split('|')) > 1: save_label = path.split('|')[1] container_dir = container_path.get(mtype) if save_dir: downloaddir.append({"save_path": save_dir, "type": type_dict.get(mtype), "category": "", "container_path": container_dir, "label": save_label}) _config['downloaddir'] = downloaddir if _config.get('qbittorrent', {}).get('save_path'): _config['qbittorrent'].pop('save_path') if _config.get('qbittorrent', {}).get('save_containerpath'): _config['qbittorrent'].pop('save_containerpath') if _config.get('transmission', {}).get('save_path'): _config['transmission'].pop('save_path') if _config.get('transmission', {}).get('save_containerpath'): _config['transmission'].pop('save_containerpath') overwrite_cofig = True elif isinstance(_config.get('downloaddir'), dict): downloaddir_list = [] for path, attr in _config.get('downloaddir').items(): downloaddir_list.append({"save_path": path, "type": attr.get("type"), "category": attr.get("category"), "container_path": attr.get("path"), "label": attr.get("label")}) _config['downloaddir'] = downloaddir_list overwrite_cofig = True # 自定义识别词兼容旧配置 try: ignored_words = Config().get_config('laboratory').get("ignored_words") if ignored_words: ignored_words = ignored_words.split("||") for ignored_word in ignored_words: if not _dbhelper.is_custom_words_existed(replaced=ignored_word): _dbhelper.insert_custom_word(replaced=ignored_word, replace="", front="", back="", offset=0, wtype=1, gid=-1, season=-2, enabled=1, regex=1, whelp="") _config['laboratory'].pop('ignored_words') overwrite_cofig = True replaced_words = Config().get_config('laboratory').get("replaced_words") if replaced_words: replaced_words = replaced_words.split("||") for replaced_word in replaced_words: replaced_word = replaced_word.split("@") if not _dbhelper.is_custom_words_existed(replaced=replaced_word[0]): _dbhelper.insert_custom_word(replaced=replaced_word[0], replace=replaced_word[1], front="", back="", offset=0, wtype=2, gid=-1, season=-2, enabled=1, regex=1, whelp="") _config['laboratory'].pop('replaced_words') overwrite_cofig = True offset_words = Config().get_config('laboratory').get("offset_words") if offset_words: offset_words = offset_words.split("||") for offset_word in offset_words: offset_word = offset_word.split("@") if not _dbhelper.is_custom_words_existed(front=offset_word[0], back=offset_word[1]): _dbhelper.insert_custom_word(replaced="", replace="", front=offset_word[0], back=offset_word[1], offset=offset_word[2], wtype=4, gid=-1, season=-2, enabled=1, regex=1, whelp="") _config['laboratory'].pop('offset_words') overwrite_cofig = True except Exception as e: ExceptionUtils.exception_traceback(e) # 目录同步兼容旧配置 try: sync_paths = Config().get_config('sync').get('sync_path') rmt_mode = Config().get_config('pt').get('sync_mod') if sync_paths: if isinstance(sync_paths, list): for sync_items in sync_paths: SyncPath = {'from': "", 'to': "", 'unknown': "", 'syncmod': rmt_mode, 'rename': 1, 'enabled': 1} # 是否启用 if sync_items.startswith("#"): SyncPath['enabled'] = 0 sync_items = sync_items[1:-1] # 是否重命名 if sync_items.startswith("["): SyncPath['rename'] = 0 sync_items = sync_items[1:-1] # 转移方式 config_items = sync_items.split("@") if not config_items: continue if len(config_items) > 1: SyncPath['syncmod'] = config_items[-1] else: SyncPath['syncmod'] = rmt_mode if not SyncPath['syncmod']: continue # 源目录|目的目录|未知目录 paths = config_items[0].split("|") if not paths: continue if len(paths) > 0: if not paths[0]: continue SyncPath['from'] = os.path.normpath(paths[0]) if len(paths) > 1: SyncPath['to'] = os.path.normpath(paths[1]) if len(paths) > 2: SyncPath['unknown'] = os.path.normpath(paths[2]) # 相同from的同步目录不能同时开启 if SyncPath['enabled'] == 1: _dbhelper.check_config_sync_paths(source=SyncPath['from'], enabled=0) _dbhelper.insert_config_sync_path(source=SyncPath['from'], dest=SyncPath['to'], unknown=SyncPath['unknown'], mode=SyncPath['syncmod'], rename=SyncPath['rename'], enabled=SyncPath['enabled']) else: _dbhelper.insert_config_sync_path(source=sync_paths, dest="", unknown="", mode=rmt_mode, rename=1, enabled=0) _config['sync'].pop('sync_path') overwrite_cofig = True except Exception as e: ExceptionUtils.exception_traceback(e) # 消息服务兼容旧配置 try: message = Config().get_config('message') or {} msg_channel = message.get('msg_channel') if msg_channel: switchs = [] switch = message.get('switch') if switch: if switch.get("download_start"): switchs.append("download_start") if switch.get("download_fail"): switchs.append("download_fail") if switch.get("transfer_finished"): switchs.append("transfer_finished") if switch.get("transfer_fail"): switchs.append("transfer_fail") if switch.get("rss_added"): switchs.append("rss_added") if switch.get("rss_finished"): switchs.append("rss_finished") if switch.get("site_signin"): switchs.append("site_signin") switchs.append('site_message') switchs.append('brushtask_added') switchs.append('brushtask_remove') switchs.append('mediaserver_message') if message.get('telegram'): token = message.get('telegram', {}).get('telegram_token') chat_id = message.get('telegram', {}).get('telegram_chat_id') user_ids = message.get('telegram', {}).get('telegram_user_ids') webhook = message.get('telegram', {}).get('webhook') if token and chat_id: name = "Telegram" ctype = 'telegram' enabled = 1 if msg_channel == ctype else 0 interactive = 1 if enabled else 0 client_config = json.dumps({ 'token': token, 'chat_id': chat_id, 'user_ids': user_ids, 'webhook': webhook }) _dbhelper.insert_message_client(name=name, ctype=ctype, config=client_config, switchs=switchs, interactive=interactive, enabled=enabled) if message.get('wechat'): corpid = message.get('wechat', {}).get('corpid') corpsecret = message.get('wechat', {}).get('corpsecret') agent_id = message.get('wechat', {}).get('agentid') default_proxy = message.get('wechat', {}).get('default_proxy') token = message.get('wechat', {}).get('Token') encodingAESkey = message.get( 'wechat', {}).get('EncodingAESKey') if corpid and corpsecret and agent_id: name = "WeChat" ctype = 'wechat' enabled = 1 if msg_channel == ctype else 0 interactive = 1 if enabled else 0 client_config = json.dumps({ 'corpid': corpid, 'corpsecret': corpsecret, 'agentid': agent_id, 'default_proxy': default_proxy, 'token': token, 'encodingAESKey': encodingAESkey }) _dbhelper.insert_message_client(name=name, ctype=ctype, config=client_config, switchs=switchs, interactive=interactive, enabled=enabled) if message.get('serverchan'): sckey = message.get('serverchan', {}).get('sckey') if sckey: name = "ServerChan" ctype = 'serverchan' interactive = 0 enabled = 1 if msg_channel == ctype else 0 client_config = json.dumps({ 'sckey': sckey }) _dbhelper.insert_message_client(name=name, ctype=ctype, config=client_config, switchs=switchs, interactive=interactive, enabled=enabled) if message.get('bark'): server = message.get('bark', {}).get('server') apikey = message.get('bark', {}).get('apikey') if server and apikey: name = "Bark" ctype = 'bark' interactive = 0 enabled = 1 if msg_channel == ctype else 0 client_config = json.dumps({ 'server': server, 'apikey': apikey }) _dbhelper.insert_message_client(name=name, ctype=ctype, config=client_config, switchs=switchs, interactive=interactive, enabled=enabled) if message.get('pushplus'): token = message.get('pushplus', {}).get('push_token') topic = message.get('pushplus', {}).get('push_topic') channel = message.get('pushplus', {}).get('push_channel') webhook = message.get('pushplus', {}).get('push_webhook') if token and channel: name = "PushPlus" ctype = 'pushplus' interactive = 0 enabled = 1 if msg_channel == ctype else 0 client_config = json.dumps({ 'token': token, 'topic': topic, 'channel': channel, 'webhook': webhook }) _dbhelper.insert_message_client(name=name, ctype=ctype, config=client_config, switchs=switchs, interactive=interactive, enabled=enabled) if message.get('iyuu'): token = message.get('iyuu', {}).get('iyuu_token') if token: name = "IyuuMsg" ctype = 'iyuu' interactive = 0 enabled = 1 if msg_channel == ctype else 0 client_config = json.dumps({ 'token': token }) _dbhelper.insert_message_client(name=name, ctype=ctype, config=client_config, switchs=switchs, interactive=interactive, enabled=enabled) # 删除旧配置 if _config.get('message', {}).get('msg_channel'): _config['message'].pop('msg_channel') if _config.get('message', {}).get('switch'): _config['message'].pop('switch') if _config.get('message', {}).get('wechat'): _config['message'].pop('wechat') if _config.get('message', {}).get('telegram'): _config['message'].pop('telegram') if _config.get('message', {}).get('serverchan'): _config['message'].pop('serverchan') if _config.get('message', {}).get('bark'): _config['message'].pop('bark') if _config.get('message', {}).get('pushplus'): _config['message'].pop('pushplus') if _config.get('message', {}).get('iyuu'): _config['message'].pop('iyuu') overwrite_cofig = True except Exception as e: ExceptionUtils.exception_traceback(e) # 站点兼容旧配置 try: sites = _dbhelper.get_config_site() for site in sites: if not site.NOTE or str(site.NOTE).find('{') != -1: continue # 是否解析种子详情为|分隔的第1位 site_parse = str(site.NOTE).split("|")[0] or "Y" # 站点过滤规则为|分隔的第2位 rule_groupid = str(site.NOTE).split("|")[1] if site.NOTE and len( str(site.NOTE).split("|")) > 1 else "" # 站点未读消息为|分隔的第3位 site_unread_msg_notify = str(site.NOTE).split("|")[2] if site.NOTE and len( str(site.NOTE).split("|")) > 2 else "Y" # 自定义UA为|分隔的第4位 ua = str(site.NOTE).split("|")[3] if site.NOTE and len( str(site.NOTE).split("|")) > 3 else "" # 是否开启浏览器仿真为|分隔的第5位 chrome = str(site.NOTE).split("|")[4] if site.NOTE and len( str(site.NOTE).split("|")) > 4 else "N" # 是否使用代理为|分隔的第6位 proxy = str(site.NOTE).split("|")[5] if site.NOTE and len( str(site.NOTE).split("|")) > 5 else "N" _dbhelper.update_config_site_note(tid=site.ID, note=json.dumps({ "parse": site_parse, "rule": rule_groupid, "message": site_unread_msg_notify, "ua": ua, "chrome": chrome, "proxy": proxy })) except Exception as e: ExceptionUtils.exception_traceback(e) # 订阅兼容旧配置 try: def __parse_rss_desc(desc): rss_sites = [] search_sites = [] over_edition = False restype = None pix = None team = None rule = None total = None current = None notes = str(desc).split('#') # 订阅站点 if len(notes) > 0: if notes[0]: rss_sites = [s for s in str(notes[0]).split( '|') if s and len(s) < 20] # 搜索站点 if len(notes) > 1: if notes[1]: search_sites = [s for s in str(notes[1]).split('|') if s] # 洗版 if len(notes) > 2: over_edition = notes[2] # 过滤条件 if len(notes) > 3: if notes[3]: filters = notes[3].split('@') if len(filters) > 0: restype = filters[0] if len(filters) > 1: pix = filters[1] if len(filters) > 2: rule = int( filters[2]) if filters[2].isdigit() else None if len(filters) > 3: team = filters[3] # 总集数及当前集数 if len(notes) > 4: if notes[4]: ep_info = notes[4].split('@') if len(ep_info) > 0: total = int(ep_info[0]) if ep_info[0] else None if len(ep_info) > 1: current = int(ep_info[1]) if ep_info[1] else None return { "rss_sites": rss_sites, "search_sites": search_sites, "over_edition": over_edition, "restype": restype, "pix": pix, "team": team, "rule": rule, "total": total, "current": current } # 电影订阅 rss_movies = _dbhelper.get_rss_movies() for movie in rss_movies: if not movie.DESC or str(movie.DESC).find('#') == -1: continue # 更新到具体字段 _dbhelper.update_rss_movie_desc( rid=movie.ID, desc=json.dumps(__parse_rss_desc(movie.DESC)) ) # 电视剧订阅 rss_tvs = _dbhelper.get_rss_tvs() for tv in rss_tvs: if not tv.DESC or str(tv.DESC).find('#') == -1: continue # 更新到具体字段 _dbhelper.update_rss_tv_desc( rid=tv.ID, desc=json.dumps(__parse_rss_desc(tv.DESC)) ) except Exception as e: ExceptionUtils.exception_traceback(e) # 字幕兼容旧配置 try: subtitle = Config().get_config('subtitle') or {} if subtitle: if subtitle.get("server") == "opensubtitles": PluginManager().save_plugin_config(pid="OpenSubtitles", conf={ "enable": subtitle.get("opensubtitles", {}).get("enable") }) else: chinesesubfinder = subtitle.get("chinesesubfinder", {}) PluginManager().save_plugin_config(pid="ChineseSubFinder", conf={ "host": chinesesubfinder.get("host"), "api_key": chinesesubfinder.get("api_key"), "local_path": chinesesubfinder.get("local_path"), "remote_path": chinesesubfinder.get("remote_path") }) # 删除旧配置 _config.pop("subtitle") overwrite_cofig = True except Exception as e: ExceptionUtils.exception_traceback(e) # 重写配置文件 if overwrite_cofig: Config().save_config(_config)