# -*- coding: UTF-8 -*-
import qbittorrentapi
import json
import argparse
import os

f = open("config.json")
server_info = json.load(f)

host_ip = "http://" + server_info['host_ip']
user_name = server_info['username']
password = server_info['password']
save_path = server_info['savepath']

parser = argparse.ArgumentParser(description='Set RSS download rule.')
parser.add_argument('--name', default='',
                    help='Bangumi Name')


def rule_set():
    qbt_client = qbittorrentapi.Client(host=host_ip, username=user_name, password=password)
    try:
        qbt_client.auth_log_in()
    except qbittorrentapi.LoginFailed as e:
        print(e)
    args = parser.parse_args()
    bangumi_name = args.name
    rule = {'enable': True,
            'mustContain': bangumi_name,
            'mustNotContain': '720',
            'useRegx': True,
            'episodeFilter': '',
            'smartFilter': False,
            'previouslyMatchedEpisodes': [],
            'affectedFeeds': [],
            'ignoreDays': 0,
            'lastMatch': '',
            'addPaused': False,
            'assignedCategory': 'Bangumi',
            'savePath': os.path.join(save_path, bangumi_name)
            }
    qbt_client.rss_set_rule(rule_name=bangumi_name, rule_def=rule)


if __name__ == "__main__":
    rule_set()