import os import shutil import sys from threading import Lock import ruamel.yaml # 种子名/文件名要素分隔字符 SPLIT_CHARS = r"\.|\s+|\(|\)|\[|]|-|\+|【|】|/|～|;|&|\||#|_|「|」|~" # 默认User-Agent DEFAULT_UA = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/98.0.4758.102 Safari/537.36" # 收藏了的媒体的目录名，名字可以改，在Emby中点击红星则会自动将电影转移到此分类下，需要在Emby Webhook中配置用户行为通知 RMT_FAVTYPE = '精选' # 支持的媒体文件后缀格式 RMT_MEDIAEXT = ['.mp4', '.mkv', '.ts', '.iso', '.rmvb', '.avi', '.mov', '.mpeg', '.mpg', '.wmv', '.3gp', '.asf', '.m4v', '.flv', '.m2ts', '.strm'] # 支持的字幕文件后缀格式 RMT_SUBEXT = ['.srt', '.ass', '.ssa'] # 电视剧动漫的分类genre_ids ANIME_GENREIDS = ['16'] # 默认过滤的文件大小，150M RMT_MIN_FILESIZE = 150 * 1024 * 1024 # 删种检查时间间隔 AUTO_REMOVE_TORRENTS_INTERVAL = 1800 # 下载文件转移检查时间间隔， PT_TRANSFER_INTERVAL = 300 # TMDB信息缓存定时保存时间 METAINFO_SAVE_INTERVAL = 600 # SYNC目录同步聚合转移时间 SYNC_TRANSFER_INTERVAL = 60 # RSS队列中处理时间间隔 RSS_CHECK_INTERVAL = 300 # 站点流量数据刷新时间间隔（小时） REFRESH_PT_DATA_INTERVAL = 6 # 刷新订阅TMDB数据的时间间隔（小时） RSS_REFRESH_TMDB_INTERVAL = 6 # 刷流删除的检查时间间隔 BRUSH_REMOVE_TORRENTS_INTERVAL = 300 # 定时清除未识别的缓存时间间隔（小时） META_DELETE_UNKNOWN_INTERVAL = 12 # 定时刷新壁纸的间隔（小时） REFRESH_WALLPAPER_INTERVAL = 1 # fanart的api，用于拉取封面图片 FANART_MOVIE_API_URL = 'https://webservice.fanart.tv/v3/movies/%s?api_key=d2d31f9ecabea050fc7d68aa3146015f' FANART_TV_API_URL = 'https://webservice.fanart.tv/v3/tv/%s?api_key=d2d31f9ecabea050fc7d68aa3146015f' # 默认背景图地址 DEFAULT_TMDB_IMAGE = 'https://s3.bmp.ovh/imgs/2022/07/10/77ef9500c851935b.webp' # 默认微信消息代理服务器地址 DEFAULT_WECHAT_PROXY = 'https://wechat.nastool.cn' # 默认OCR识别服务地址 DEFAULT_OCR_SERVER = 'https://nastool.cn' # 默认TMDB代理服务地址 DEFAULT_TMDB_PROXY = 'https://tmdb.nastool.cn' # 默认CookieCloud服务地址 DEFAULT_COOKIECLOUD_SERVER = 'http://nastool.cn:8088' # TMDB图片地址 TMDB_IMAGE_W500_URL = 'https://image.tmdb.org/t/p/w500%s' TMDB_IMAGE_ORIGINAL_URL = 'https://image.tmdb.org/t/p/original%s' TMDB_IMAGE_FACE_URL = 'https://image.tmdb.org/t/p/h632%s' TMDB_PEOPLE_PROFILE_URL = 'https://www.themoviedb.org/person/%s' # 添加下载时增加的标签，开始只监控NASTool添加的下载时有效 PT_TAG = "NASTOOL" # 电影默认命名格式 DEFAULT_MOVIE_FORMAT = '{title} ({year})/{title} ({year})-{part} - {videoFormat}' # 电视剧默认命名格式 DEFAULT_TV_FORMAT = '{title} ({year})/Season {season}/{title} - {season_episode}-{part} - 第 {episode} 集' # 辅助识别参数 KEYWORD_SEARCH_WEIGHT_1 = [10, 3, 2, 0.5, 0.5] KEYWORD_SEARCH_WEIGHT_2 = [10, 2, 1] KEYWORD_SEARCH_WEIGHT_3 = [10, 2] KEYWORD_STR_SIMILARITY_THRESHOLD = 0.2 KEYWORD_DIFF_SCORE_THRESHOLD = 30 KEYWORD_BLACKLIST = ['中字', '韩语', '双字', '中英', '日语', '双语', '国粤', 'HD', 'BD', '中日', '粤语', '完全版', '法语', '西班牙语', 'HRHDTVAC3264', '未删减版', '未删减', '国语', '字幕组', '人人影视', 'www66ystv', '人人影视制作', '英语', 'www6vhaotv', '无删减版', '完成版', '德意'] # WebDriver路径 WEBDRIVER_PATH = { "Docker": "/usr/lib/chromium/chromedriver", "Synology": "/var/packages/NASTool/target/bin/chromedriver" } # Xvfb虚拟显示路程 XVFB_PATH = [ "/usr/bin/Xvfb", "/usr/local/bin/Xvfb" ] # 线程锁 lock = Lock() # 全局实例 _CONFIG = None def singleconfig(cls): def _singleconfig(*args, **kwargs): global _CONFIG if not _CONFIG: with lock: _CONFIG = cls(*args, **kwargs) return _CONFIG return _singleconfig @singleconfig class Config(object): _config = {} _config_path = None def __init__(self): self._config_path = os.environ.get('NASTOOL_CONFIG') if not os.environ.get('TZ'): os.environ['TZ'] = 'Asia/Shanghai' self.init_syspath() self.init_config() def init_config(self): try: if not self._config_path: print("【Config】NASTOOL_CONFIG 环境变量未设置，程序无法工作，正在退出...") quit() if not os.path.exists(self._config_path): cfg_tp_path = os.path.join(self.get_inner_config_path(), "config.yaml") cfg_tp_path = cfg_tp_path.replace("\\", "/") shutil.copy(cfg_tp_path, self._config_path) print("【Config】config.yaml 配置文件不存在，已将配置文件模板复制到配置目录...") with open(self._config_path, mode='r', encoding='utf-8') as cf: try: # 读取配置 print("正在加载配置：%s" % self._config_path) self._config = ruamel.yaml.YAML().load(cf) except Exception as e: print("【Config】配置文件 config.yaml 格式出现严重错误！请检查：%s" % str(e)) self._config = {} except Exception as err: print("【Config】加载 config.yaml 配置出错：%s" % str(err)) return False def init_syspath(self): with open(os.path.join(self.get_root_path(), "third_party.txt"), "r") as f: for third_party_lib in f.readlines(): module_path = os.path.join(self.get_root_path(), "third_party", third_party_lib.strip()).replace("\\", "/") if module_path not in sys.path: sys.path.append(module_path) def get_proxies(self): return self.get_config('app').get("proxies") def get_ua(self): return self.get_config('app').get("user_agent") or DEFAULT_UA def get_config(self, node=None): if not node: return self._config return self._config.get(node, {}) def save_config(self, new_cfg): self._config = new_cfg with open(self._config_path, mode='w', encoding='utf-8') as sf: yaml = ruamel.yaml.YAML() return yaml.dump(new_cfg, sf) def get_config_path(self): return os.path.dirname(self._config_path) def get_temp_path(self): return os.path.join(self.get_config_path(), "temp") @staticmethod def get_root_path(): return os.path.dirname(os.path.realpath(__file__)) def get_inner_config_path(self): return os.path.join(self.get_root_path(), "config") def get_script_path(self): return os.path.join(self.get_inner_config_path(), "scripts") def get_domain(self): domain = (self.get_config('app') or {}).get('domain') if domain and not domain.startswith('http'): domain = "http://" + domain return domain @staticmethod def get_timezone(): return os.environ.get('TZ')