init

app/core/config.py

@@ -0,0 +1,182 @@
+import secrets
+from pathlib import Path
+
+from pydantic import BaseSettings
+
+
+class Settings(BaseSettings):
+    # 项目名称
+    PROJECT_NAME = "NASbot"
+    # API路径
+    API_V1_STR: str = "/api/v1"
+    # 密钥
+    SECRET_KEY: str = secrets.token_urlsafe(32)
+    # TOKEN过期时间
+    ACCESS_TOKEN_EXPIRE_MINUTES: int = 60 * 24 * 8
+    # 时区
+    TZ: str = "Asia/Shanghai"
+    # 监听地址，ipv6改为::
+    HOST: str = "0.0.0.0"
+    # 监听端口
+    PORT: int = 3001
+    # 配置文件目录
+    CONFIG_DIR: str = None
+    # 超级管理员
+    SUPERUSER: str = "admin"
+    # 超级管理员密码
+    SUPERUSER_PASSWORD: str = "password"
+    # API密钥，需要更换
+    API_TOKEN: str = "nasbot"
+    # 网络代理
+    PROXY_HOST: str = None
+    # 媒体信息搜索来源
+    SEARCH_SOURCE: str = "themoviedb"
+    # 刮削来源
+    SCRAP_SOURCE: str = "themoviedb"
+    # TMDB图片地址
+    TMDB_IMAGE_DOMAIN: str = "image.tmdb.org"
+    # TMDB API地址
+    TMDB_API_DOMAIN: str = "api.themoviedb.org"
+    # TMDB API Key
+    TMDB_API_KEY: str = "db55323b8d3e4154498498a75642b381"
+    # Fanart API Key
+    FANART_API_KEY: str = "d2d31f9ecabea050fc7d68aa3146015f"
+    # 支持的后缀格式
+    RMT_MEDIAEXT: list = ['.mp4', '.mkv', '.ts', '.iso',
+                          '.rmvb', '.avi', '.mov', '.mpeg',
+                          '.mpg', '.wmv', '.3gp', '.asf',
+                          '.m4v', '.flv', '.m2ts', '.strm',
+                          '.tp']
+    # 支持的字幕文件后缀格式
+    RMT_SUBEXT: list = ['.srt', '.ass', '.ssa']
+    # 支持的音轨文件后缀格式
+    RMT_AUDIO_TRACK_EXT: list = ['.mka']
+    # 索引器
+    INDEXER: str = "builtin"
+    # 消息通知渠道 telegram/wechat
+    MESSAGER: str = "telegram"
+    # WeChat企业ID
+    WECHAT_CORPID: str = None
+    # WeChat应用Secret
+    WECHAT_APP_SECRET: str = None
+    # WeChat应用ID
+    WECHAT_APP_ID: str = None
+    # WeChat代理服务器
+    WECHAT_PROXY: str = None
+    # WeChat Token
+    WECHAT_TOKEN: str = None
+    # WeChat EncodingAESKey
+    WECHAT_ENCODING_AESKEY: str = None
+    # WeChat 管理员
+    WECHAT_ADMINS: str = None
+    # Telegram Bot Token
+    TELEGRAM_TOKEN: str = None
+    # Telegram Chat ID
+    TELEGRAM_CHAT_ID: str = None
+    # Telegram 用户ID，使用,分隔
+    TELEGRAM_USERS: str = ""
+    # Telegram 管理员ID，使用,分隔
+    TELEGRAM_ADMINS: str = ""
+    # 下载器 qbittorrent/transmission
+    DOWNLOADER: str = "qbittorrent"
+    # Qbittorrent地址
+    QB_HOST: str = None
+    # Qbittorrent用户名
+    QB_USER: str = None
+    # Qbittorrent密码
+    QB_PASSWORD: str = None
+    # Transmission地址
+    TR_HOST: str = None
+    # Transmission用户名
+    TR_USER: str = None
+    # Transmission密码
+    TR_PASSWORD: str = None
+    # 下载保存目录，容器内映射路径需要一致
+    DOWNLOAD_PATH: str = "/downloads"
+    # 媒体服务器 emby/jellyfin/plex
+    MEDIASERVER: str = "emby"
+    # EMBY服务器地址
+    EMBY_HOST: str = None
+    # EMBY Api Key
+    EMBY_API_KEY: str = None
+    # Jellyfin服务器地址
+    JELLYFIN_HOST: str = None
+    # Jellyfin Api Key
+    JELLYFIN_API_KEY: str = None
+    # Plex服务器地址
+    PLEX_HOST: str = None
+    # Plex Token
+    PLEX_TOKEN: str = None
+    # 过滤规则
+    FILTER_RULE: str = ""
+    # 转移方式 link/copy/move/softlink
+    TRANSFER_TYPE: str = "copy"
+    # CookieCloud服务器地址
+    COOKIECLOUD_HOST: str = "https://nastool.org/cookiecloud"
+    # CookieCloud用户KEY
+    COOKIECLOUD_KEY: str = None
+    # CookieCloud端对端加密密码
+    COOKIECLOUD_PASSWORD: str = None
+    # CookieCloud同步间隔（分钟）
+    COOKIECLOUD_INTERVAL: int = 3600
+    # CookieCloud对应的浏览器UA
+    USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36 Edg/113.0.1774.57"
+    # 媒体库目录
+    LIBRARY_PATH: str = None
+    # 二级分类
+    LIBRARY_CATEGORY: bool = True
+    # 豆瓣用户ID，用于同步豆瓣数据，使用,分隔
+    DOUBAN_USER_IDS: str = ""
+
+    @property
+    def INNER_CONFIG_PATH(self):
+        return self.ROOT_PATH / "config"
+
+    @property
+    def CONFIG_PATH(self):
+        if self.CONFIG_DIR:
+            return Path(self.CONFIG_DIR)
+        return self.INNER_CONFIG_PATH
+
+    @property
+    def TEMP_PATH(self):
+        return self.CONFIG_PATH / "temp"
+
+    @property
+    def ROOT_PATH(self):
+        return Path(__file__).parents[2]
+
+    @property
+    def PLUGIN_DATA_PATH(self):
+        return self.CONFIG_PATH / "plugins"
+
+    @property
+    def LOG_PATH(self):
+        return self.CONFIG_PATH / "logs"
+
+    @property
+    def PROXY(self):
+        if self.PROXY_HOST:
+            return {
+                "http": self.PROXY_HOST,
+                "https": self.PROXY_HOST
+            }
+        return None
+
+    def __init__(self):
+        super().__init__()
+        with self.CONFIG_PATH as p:
+            if not p.exists():
+                p.mkdir(parents=True, exist_ok=True)
+        with self.TEMP_PATH as p:
+            if not p.exists():
+                p.mkdir(parents=True, exist_ok=True)
+        with self.LOG_PATH as p:
+            if not p.exists():
+                p.mkdir(parents=True, exist_ok=True)
+
+    class Config:
+        case_sensitive = True
+
+
+settings = Settings()