Rss_Rule_Set.py

# -*- coding: UTF-8 -*-
import json
import os
import posixpath
import xml.etree.ElementTree as ET
import sys
import qbittorrentapi
import requests


def load_config(conf_path: str) -> dict:
    if os.path.exists(conf_path):
        with open(conf_path, "r", encoding="utf-8") as f:
            conf_dict = json.load(f)
    else:
        create_config(conf_path)
        with open(conf_path, "r", encoding="utf-8") as f:
            conf_dict = json.load(f)
    return conf_dict


def create_config(conf_path: str):
    conf_dict = {
        "qb_adress": "127.0.0.1",
        "qb_port": 8080,
        "qb_user_name": "admin",
        "qb_user_pwd": "adminadmin",
        "qb_save_path": "/home/downloads/",
    }
    with open(conf_path, "w", encoding="utf-8") as f:
        json.dump(conf_dict, f, ensure_ascii=False)


def get_feed_title(feed_url: str, proxy: bool = False):
    proxies = {"http": "http://127.0.0.1:10800", "https": "http://127.0.0.1:10800"}
    if proxy:
        r = requests.get(feed_url, proxies=proxies)
    else:
        r = requests.get(feed_url)
    root = ET.fromstring(r.text)
    rss_title = root[0][0].text
    return rss_title


def get_qb_client(conf_dict: dict):
    qbt_client = qbittorrentapi.Client(
        host=conf_dict["qb_adress"],
        port=conf_dict["qb_port"],
        username=conf_dict["qb_user_name"],
        password=conf_dict["qb_user_pwd"],
    )
    print(f"正在登录...请稍后...")
    print(f"webui地址：{conf_dict['qb_adress']}")
    print(f"用户名：{conf_dict['qb_user_name']}")
    try:
        qbt_client.auth_log_in()
        print(f"登录成功！")
        print(f"qBittorrent: {qbt_client.app.version}")
        print(f"qBittorrent Web API: {qbt_client.app.web_api_version}")
    except qbittorrentapi.LoginFailed as e:
        print(e)
        return
    return qbt_client


def rss_rule_set(
    qbt_client: qbittorrentapi.Client,
    conf_dict: dict,
    bangumi_name: str,
    feed_url: str,
    must_not_contain: str,
    romaji: str,
    season_number: int,
    proxy: bool,
):
    # 是否使用罗马音作为文件夹名称
    if romaji == "not use":
        folder_name = bangumi_name
    else:
        folder_name = romaji
    if season_number == 1:
        # 需要使用qbt webui的时候，大概率是在linux上运行的qbt，
        # 但是本软件的运行环境未必是linux，如果直接使用os.path.join，并且运行环境是windows时，
        # 会导致拼接路径的符号变成“\”，从而导致qbt创建文件夹时候出现错误，
        # 该错误会导致“\”后不会被认为是子文件夹，而是文件夹名称的一部分，
        # 从而会使qbt创建出奇怪的文件夹，并且会在其中下载文件。
        # 使用posixpath来规避该问题。
        save_path = posixpath.join(conf_dict["qb_save_path"], folder_name)
    else:
        save_path = posixpath.join(
            conf_dict["qb_save_path"], folder_name, f"S0{season_number}"
        )
    rule = {
        "enable": True,
        "mustContain": bangumi_name,
        "mustNotContain": must_not_contain,
        "useRegx": True,
        "episodeFilter": "",
        "smartFilter": False,
        "previouslyMatchedEpisodes": [],
        "affectedFeeds": [feed_url],
        "ignoreDays": 0,
        "lastMatch": "",
        "addPaused": False,
        "assignedCategory": bangumi_name,
        "savePath": save_path,
    }
    print("番剧名称(必须包含)：")
    print(bangumi_name)
    print("订阅地址：")
    print(feed_url)
    print("忽略下载的文件所包含的字符：")
    print(must_not_contain)
    print("番剧存储文件夹：")
    print(save_path)
    input("请确认订阅信息!按下Enter确定!")
    try:
        qbt_client.torrents_create_category(bangumi_name)
    except qbittorrentapi.exceptions.Conflict409Error:
        print("无法在QB中创建对应番剧名称的分类，可能已经存在该分类")
    qbt_client.rss_add_feed(feed_url, get_feed_title(feed_url, proxy))
    qbt_client.rss_set_rule(rule_name=bangumi_name, rule_def=rule)


if __name__ == "__main__":
    conf_dict = load_config("rss_config.json")
    qbt_client = get_qb_client(conf_dict)
    if qbt_client is None:
        input("登录失败！请检查配置文件及网络环境！")
        sys.exit()
    # =======================================================================
    # 番剧名称、同时也是RSS自动下载中“必须包含”的字
    bangumi_name = "怪兽 8 号"
    # 番剧罗马音，下载的番剧存储的文件夹名称，同时方便刮削元数据
    romaji = "not use"
    # 订阅网址
    feed_url = "https://mikanani.me/RSS/Bangumi?bangumiId=3301&subgroupid=583"
    # 如果是gfw外网址则替换回国内网址
    feed_url = feed_url.replace("mikanani.me", "mikanime.tv")
    # RSS自动下载中“不可包含”的字
    # 720|CHT|繁体|B-Global|BIG5|港澳台|bilibili
    must_not_contain = ""
    # 如果季度不为1则启用季度子文件夹
    season_number = 1
    # 获取订阅标题时使用代理
    use_proxy = False
    # =======================================================================

    rss_rule_set(
        qbt_client,  # qbt实例
        conf_dict,  # config文件dict
        bangumi_name,  # 番剧名称
        feed_url,  # 订阅url
        must_not_contain,  # 不可包含
        romaji,  # 罗马音
        season_number,  # 季度
        use_proxy,  # 是否代理
    )