-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_check.py
112 lines (101 loc) · 5.51 KB
/
run_check.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from qb_api import *
from config import main_config
import logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
def check_qb(main_url,login_param,max_num,max_addtime,rss_item,rule):
# 登录qb
session = login(main_url, login_param)
# 获取qb所有种子信息
torrents_info = get_torrents_info(session, main_url)
# logger.info(torrents_info)
if len(torrents_info) >= max_num:
logger.warning("【节点：已满种子最大值】")
# 检查种子添加时间是否超过最大时间，执行删除操作
check_del_torrent = check_torrents_addtime(session, main_url, max_addtime)
if check_del_torrent == []:
logger.warning("【节点：暂无可删除的超时种子】")
# 刷新并获取所有rss项目数据
get_rss_items(session, main_url, rss_item)
# 批量已读订阅源
mark_Read(session, main_url, rss_item)
# send_msg(token, '暂无可删除的超时种子')
elif len(check_del_torrent) >= 1:
logger.warning("【节点：存在超时添加种子，开始判断是否删除部分种子】")
# 刷新并获取所有rss项目数据
rss_data = get_rss_items(session, main_url, rss_item)
# 根据规则过滤有效的种子
valid_rss = filter_rss(rss_data, rss_item, rule)
if valid_rss == []:
logger.warning("【节点：无可用的有效种子，暂不删除过期种子】")
# 批量已读订阅源
mark_Read(session, main_url, rss_item)
elif len(valid_rss) > 0:
logger.warning("【节点：找到可用的有效种子，开始删除过期种子并新增种子】")
# 对比删除种子数量和可用的有效种子数量，取其中的较小值操作
min_num = min(len(check_del_torrent),len(valid_rss))
# 按种子的添加时间戳升序排列，便于先删除旧种
sort_check_del_torrent = sorted(check_del_torrent, key=lambda k: k['added_on'])
for count in range(0, min_num):
del_torrent = sort_check_del_torrent[count]
# add_time = del_torrents['added_on']
del_hash = del_torrent['hash']
del_name = del_torrent['name']
del_torrents(session, main_url, del_hash, del_name)
# 打印已移除的种子
logger.info("已移除种子：{0}".format(del_torrent['name']))
# 获取第x个有效种子
new_torrent = valid_rss[count]
torrent_url = new_torrent['torrentURL']
torrent_title = new_torrent['title']
add_torrents(session, main_url, torrent_url, torrent_title)
logger.info("新增种子：{0}".format(torrent_title))
msg = '删除种子：' + del_name + '，新增种子：' + torrent_title
send_msg(token, '删除旧种，并新增推送种子', msg)
# send_msg(token, '删除种子',del_torrents['name'])
mark_Read(session, main_url, rss_item)
else:
logger.warning("【节点：未达到种子数做种最大值，开始拉取rss尝试添加】")
# 计算与最大值的差值
missing_num = max_num - len(torrents_info)
# 刷新并获取所有rss项目数据
rss_data = get_rss_items(session, main_url, rss_item)
# 根据规则过滤有效的种子
valid_rss = filter_rss(rss_data, rss_item, rule)
if valid_rss == []:
logger.warning("【节点：暂无符合规则的种子添加填充】")
# 批量已读订阅源
mark_Read(session, main_url, rss_item)
# logger.info("暂无符合规则的种子添加。。。")
# send_msg(token, '无可用种子新增')
else:
logger.warning("【节点：找到可用种子，执行新增】")
# 有效种子的数量
valid_num = len(valid_rss)
# 批量添加种子
for count in range(0,min(valid_num,missing_num)): # 有效种子数和差值比较，取小的值，防止发送过多种子或数组越界报错
torrent = valid_rss[count]
torrent_url = torrent['torrentURL']
torrent_title = torrent['title']
add_torrents(session, main_url, torrent_url,torrent_title)
send_msg(token, '新增推送种子',torrent_title + torrent_url)
mark_Read(session, main_url, rss_item)
if __name__=="__main__":
# 获取qb客户端的WEB UI地址+端口
main_url = main_config.main_url
# 获取登录qb的账号密码
login_param = main_config.param
# 获取允许同时运行种子的最大值
max_num = main_config.max_num
# 获取用于check_torrents_addtime()方法的种子最大添加时间，超过则删除
max_addtime = main_config.max_addtime
# 获取订阅源组名
rss_item = main_config.rss_item
# 获取rss并过滤整理
rule = main_config.rule
# 获取爱语飞飞 token
token = main_config.token
# 主函数
check_qb(main_url, login_param, max_num, max_addtime, rss_item, rule)