-
Notifications
You must be signed in to change notification settings - Fork 7
/
Main.py
208 lines (190 loc) · 8.76 KB
/
Main.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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
import os
import time
import traceback
from threading import Thread
import psutil
from autopt.AutoPT_BYR import AutoPT_BYR
from autopt.AutoPT_FRDS import AutoPT_FRDS
from autopt.AutoPT_MTEAM import AutoPT_MTEAM
from autopt.AutoPT_PTER import AutoPT_PTER
from autopt.AutoPT_PTHOME import AutoPT_PTHOME
from autopt.AutoPT_SOULVOICE import AutoPT_SOULVOICE
from autopt.AutoPT_TJU import AutoPT_TJU
from autopt.AutoPT_TTG import AutoPT_TTG
from autopt.AutoPT_LEMONHD import AutoPT_LEMONHD
from autopt.QBmanage_Reseed import Manager
from tools import Myconfig, Mylogger, BGIcon
from tools import globalvar as gl
from tools.iyuu import iyuu
def run():
logger = gl.get_value('logger').logger
try:
maxtime = 1
auto_byr = None
auto_tju = None
auto_pter = None
auto_mteam = None
auto_pthome = None
auto_frds = None
auto_ttg = None
auto_lemonhd = None
auto_soulvoice = None
Runqbittorrent()
refconfig = {
'name': 'reseed',
'ref': {
'byr': auto_byr,
'tju': auto_tju,
'mteam': auto_mteam,
'pter': auto_pter,
'pthome': auto_pthome,
'frds': auto_frds,
'ttg': auto_ttg,
'lemonhd': auto_lemonhd,
'soulvoice': auto_soulvoice
}
}
gl.set_value('allref', refconfig)
auto_byr = AutoPT_BYR()
refconfig['ref']['byr'] = auto_byr
auto_tju = AutoPT_TJU()
refconfig['ref']['tju'] = auto_tju
auto_pter = AutoPT_PTER()
refconfig['ref']['pter'] = auto_pter
auto_mteam = AutoPT_MTEAM()
refconfig['ref']['mteam'] = auto_mteam
auto_pthome = AutoPT_PTHOME()
refconfig['ref']['pthome'] = auto_pthome
auto_frds = AutoPT_FRDS()
refconfig['ref']['frds'] = auto_frds
auto_ttg = AutoPT_TTG()
refconfig['ref']['ttg'] = auto_ttg
auto_lemonhd = AutoPT_LEMONHD()
refconfig['ref']['lemonhd'] = auto_lemonhd
auto_soulvoice = AutoPT_SOULVOICE()
refconfig['ref']['soulvoice'] = auto_soulvoice
if gl.get_value('config').switch('byr'):
if maxtime % gl.get_value('config').intervaltime('byr') != 0:
maxtime *= gl.get_value('config').intervaltime('byr')
if gl.get_value('config').switch('tju'):
if maxtime % gl.get_value('config').intervaltime('tju') != 0:
maxtime *= gl.get_value('config').intervaltime('tju')
if gl.get_value('config').switch('pter'):
if maxtime % gl.get_value('config').intervaltime('pter') != 0:
maxtime *= gl.get_value('config').intervaltime('pter')
if gl.get_value('config').switch('mteam'):
if maxtime % gl.get_value('config').intervaltime('mteam') != 0:
maxtime *= gl.get_value('config').intervaltime('mteam')
if gl.get_value('config').switch('pthome'):
if maxtime % gl.get_value('config').intervaltime('pthome') != 0:
maxtime *= gl.get_value('config').intervaltime('pthome')
if gl.get_value('config').switch('frds'):
if maxtime % gl.get_value('config').intervaltime('frds') != 0:
maxtime *= gl.get_value('config').intervaltime('frds')
if gl.get_value('config').switch('ttg'):
if maxtime % gl.get_value('config').intervaltime('ttg') != 0:
maxtime *= gl.get_value('config').intervaltime('ttg')
if gl.get_value('config').switch('lemonhd'):
if maxtime % gl.get_value('config').intervaltime('lemonhd') != 0:
maxtime *= gl.get_value('config').intervaltime('lemonhd')
if gl.get_value('config').switch('soulvoice'):
if maxtime % gl.get_value('config').intervaltime('soulvoice') != 0:
maxtime *= gl.get_value('config').intervaltime('soulvoice')
manager = Manager()
if maxtime % (6 * 3600) != 0:
maxtime *= (6 * 3600)
counttime = 0
while gl.get_value('thread_flag'):
if gl.get_value('config').switch('reseed') and counttime % 120 == 0:
manager.checkalltorrentexist()
manager.recheck()
if gl.get_value('config').switch('reseed') and counttime % (3 * 3600) == 0:
manager.checkprttracker()
manager.recheckall()
manager.checkemptydir()
if gl.get_value('thread_flag') and gl.get_value('config').switch('soulvoice') and counttime % gl.get_value(
'config').intervaltime('soulvoice') == 0:
auto_soulvoice.start()
pass
if gl.get_value('thread_flag') and gl.get_value('config').switch('lemonhd') and counttime % gl.get_value(
'config').intervaltime('lemonhd') == 0:
auto_lemonhd.start()
pass
if gl.get_value('thread_flag') and gl.get_value('config').switch('ttg') and counttime % gl.get_value(
'config').intervaltime('ttg') == 0:
auto_ttg.start()
pass
if gl.get_value('thread_flag') and gl.get_value('config').switch('frds') and counttime % gl.get_value(
'config').intervaltime('frds') == 0:
auto_frds.start()
pass
if gl.get_value('thread_flag') and gl.get_value('config').switch('pthome') and counttime % gl.get_value(
'config').intervaltime('pthome') == 0:
auto_pthome.start()
if gl.get_value('thread_flag') and gl.get_value('config').switch('mteam') and counttime % gl.get_value(
'config').intervaltime('mteam') == 0:
auto_mteam.start()
if gl.get_value('thread_flag') and gl.get_value('config').switch('tju') and counttime % gl.get_value(
'config').intervaltime('tju') == 0:
auto_tju.start()
if gl.get_value('thread_flag') and gl.get_value('config').switch('pter') and counttime % gl.get_value(
'config').intervaltime('pter') == 0:
auto_pter.start()
if gl.get_value('thread_flag') and gl.get_value('config').switch('byr') and counttime % gl.get_value(
'config').intervaltime('byr') == 0:
auto_byr.start()
counttime = (1 + counttime) % maxtime
time.sleep(1)
except BaseException:
logger.exception(traceback.format_exc())
# TODO 测试qb退出连接不上时候不会自动报错
# gl.set_value('thread_flag', False)
# traceback.print_exc(file=open('treace.txt', 'w+'))
def CheckProgramStatus(name):
list = psutil.pids()
for i in range(0, len(list)):
try:
p = psutil.Process(list[i])
if 'qbittorrent.exe' in p.name():
return True
except BaseException as e:
# 当某些进程不存在了会有异常，无视即可
pass
return False
def Runqbittorrent():
logger = gl.get_value('logger').logger
if gl.get_value('config').qbtpath != '':
try:
if not CheckProgramStatus('qbittorrent.exe'):
logger.debug('未检测到QBitTorrent打开，正在尝试打开')
os.startfile(gl.get_value('config').qbtpath)
trytime = 60
while trytime > 0 and (not CheckProgramStatus('qbittorrent.exe')):
trytime -= 5
logger.info('正在等待QBT启动')
time.sleep(5)
if trytime <= 0:
logger.error('QBT启动失败,异常退出')
else:
logger.error('QBT启动成功')
else:
logger.debug('QBitTorrent已在运行')
except BaseException as e:
logger.exception(traceback.format_exc())
if __name__ == '__main__':
gl._init()
gl.set_value('thread_flag', True)
try:
gl.set_value('config', Myconfig.Config())
gl.set_value('logger', Mylogger.Mylogger())
gl.set_value('wechat', iyuu(gl.get_value('config').token))
gl.set_value('thread', Thread(target=run))
app = BGIcon.MyWindows()
gl.set_value('wxpython', app)
gl.get_value('logger').logger.info('程序启动')
gl.get_value('thread').start()
app.MainLoop()
except BaseException:
traceback.print_exc(file=open('treace.txt', 'w+'))
# gl.get_value('logger').logger.exception(traceback.format_exc())
gl.set_value('thread_flag', False)