mirror of
https://jihulab.com/spiritlhl/qinglong_auto_tools.git
synced 2026-06-05 21:07:21 +08:00
287 lines
9.1 KiB
Python
287 lines
9.1 KiB
Python
# -*- coding:utf-8 -*-
|
||
#作者仓库:https://jihulab.com/spiritlhl/qinglong_auto_tools.git
|
||
#觉得不错麻烦点个star谢谢
|
||
# 频道：https://t.me/qinglong_auto_tools
|
||
'''
|
||
cron: 1
|
||
new Env('任务 二叉树同步任务启用禁用');
|
||
'''
|
||
# 同步任务是否启用禁用，不存在于分容器的不会同步新增
|
||
|
||
# 在脚本管理里修改这个文件的配置，然后保存，然后禁用 二叉树同步任务启用禁用 这个任务，有需要再点运行
|
||
|
||
|
||
# 主青龙，需要修改任务的容器，事先需要在容器里创建应用，给所有权限，然后重启容器，应用设置才会生效，
|
||
# 副青龙，被同步的任务容器，事先需要在容器里创建应用，给所有权限，然后重启容器，应用设置才会生效，
|
||
# 按照格式有几个写几个，没有的空的删除
|
||
'''
|
||
# ec_config.txt中填写如下设置
|
||
|
||
# 二叉树同步任务启用禁用
|
||
### 主青龙
|
||
tasks_sync_able_cilent_id1="xxxxxxxx"
|
||
tasks_sync_able_cilent_secret1="xxxxxxxxxx"
|
||
tasks_sync_able_url1="http://xxxxxxxx:xxxx/"
|
||
|
||
### 副青龙
|
||
tasks_sync_able_client_ids=["",""]
|
||
tasks_sync_able_client_secrets=["",""]
|
||
tasks_sync_able_urllist=["http://xxxxxxxxx:xxxx/",""]
|
||
|
||
'''
|
||
#client_id1=""
|
||
#client_secret1=""
|
||
#url1 = "http://ip:端口/"
|
||
#client_ids=['','','','']
|
||
#client_secrets=['','','','']
|
||
#urllist = ["http://xxxx:xxxx/","","",""]
|
||
|
||
import re
|
||
import time
|
||
import json
|
||
try:
|
||
import requests
|
||
except Exception as e:
|
||
print(e, "\n缺少requests 模块，请执行命令安装：python3 -m pip install requests")
|
||
|
||
try:
|
||
with open("ec_config.txt", "r", encoding="utf-8") as fp:
|
||
t = fp.readlines()
|
||
try:
|
||
for i in t:
|
||
try:
|
||
temp = re.findall(r"tasks_sync_able_cilent_id1=\"(.*?)\"", i)[0]
|
||
client_id1 = temp
|
||
if client_id1 == "":
|
||
print("tasks_sync_able_cilent_id1 未填写")
|
||
except:
|
||
pass
|
||
except:
|
||
print("tasks_sync_able_cilent_id1 未创建")
|
||
exit(3)
|
||
|
||
try:
|
||
for i in t:
|
||
try:
|
||
temp = re.findall(r"tasks_sync_able_cilent_secret1=\"(.*?)\"", i)[0]
|
||
client_secret1 = temp
|
||
if client_secret1 == "":
|
||
print("tasks_sync_able_cilent_secret1 未填写")
|
||
except:
|
||
pass
|
||
except:
|
||
print("tasks_sync_able_cilent_secret1 未创建")
|
||
exit(3)
|
||
|
||
try:
|
||
for i in t:
|
||
try:
|
||
temp = re.findall(r"tasks_sync_able_url1=\"(.*?)\"", i)[0]
|
||
url1 = temp
|
||
if url1 == "":
|
||
print("tasks_sync_able_url1 未填写")
|
||
except:
|
||
pass
|
||
except:
|
||
print("tasks_sync_able_url1 未创建")
|
||
exit(3)
|
||
except:
|
||
print("找不到配置文件或配置文件有错误, 请填写ec_config.txt")
|
||
|
||
|
||
try:
|
||
try:
|
||
for i in t:
|
||
try:
|
||
temp = "["+re.findall(r"tasks_sync_able_client_ids=\[(.*?)\]", i)[0]+"]"
|
||
try:
|
||
client_ids = json.loads(temp)
|
||
except:
|
||
print("tasks_sync_able_client_ids 填写有误")
|
||
if client_ids == []:
|
||
print("tasks_sync_able_client_ids 未填写")
|
||
except:
|
||
pass
|
||
except:
|
||
print("tasks_sync_able_client_ids 未创建")
|
||
exit(3)
|
||
|
||
try:
|
||
for i in t:
|
||
try:
|
||
temp = "["+re.findall(r"tasks_sync_able_client_secrets=\[(.*?)\]", i)[0]+"]"
|
||
try:
|
||
client_secrets = json.loads(temp)
|
||
except:
|
||
print("tasks_sync_able_client_secrets 填写有误")
|
||
if client_secrets == []:
|
||
print("tasks_sync_able_client_secrets 未填写")
|
||
except:
|
||
pass
|
||
except:
|
||
print("tasks_sync_able_client_secrets 未创建")
|
||
exit(3)
|
||
|
||
try:
|
||
for i in t:
|
||
try:
|
||
temp = "["+re.findall(r"tasks_sync_able_urllist=\[(.*?)\]", i)[0]+"]"
|
||
try:
|
||
urllist = json.loads(temp)
|
||
except:
|
||
print("tasks_sync_able_urllist 填写有误")
|
||
if urllist == []:
|
||
print("tasks_sync_able_urllist 未填写")
|
||
except:
|
||
pass
|
||
except:
|
||
print("tasks_sync_able_urllist 未创建")
|
||
exit(3)
|
||
except:
|
||
print("找不到配置文件或配置文件有错误, 请填写ec_config.txt")
|
||
|
||
|
||
requests.packages.urllib3.disable_warnings()
|
||
|
||
def gettimestamp():
|
||
return str(int(time.time() * 1000))
|
||
|
||
def gettoken(self,url_token):
|
||
r = requests.get(url_token).text
|
||
res = json.loads(r)["data"]["token"]
|
||
self.headers.update({"Authorization": "Bearer "+res})
|
||
|
||
def login(self, baseurl, client_id_temp, client_secret_temp):
|
||
url_token = baseurl+'open/auth/token?client_id='+client_id_temp+'&client_secret='+client_secret_temp
|
||
gettoken(self, url_token)
|
||
|
||
def getitem(self, baseurl, typ):
|
||
url = baseurl + typ + "/crons?t=%s" % gettimestamp()
|
||
r = self.get(url)
|
||
item = json.loads(r.text)["data"]
|
||
return item
|
||
|
||
def synchronous_tasks_enable(self, baseurl, typ, data):
|
||
url = baseurl + typ + "/crons/enable?t=%s" % gettimestamp()
|
||
self.headers.update({"Content-Type": "application/json;charset=UTF-8", 'Connection': 'close'})
|
||
r = self.put(url, data=json.dumps(data))
|
||
if json.loads(r.text)["code"] == 200:
|
||
return True
|
||
else:
|
||
return r.text
|
||
|
||
|
||
def synchronous_tasks_disable(self, baseurl, typ, data):
|
||
url = baseurl + typ + "/crons/disable?t=%s" % gettimestamp()
|
||
self.headers.update({"Content-Type": "application/json;charset=UTF-8", 'Connection': 'close'})
|
||
r = self.put(url, data=json.dumps(data))
|
||
if json.loads(r.text)["code"] == 200:
|
||
return True
|
||
else:
|
||
return r.text
|
||
|
||
|
||
if __name__ == '__main__':
|
||
#主容器
|
||
print("=========== 主容器任务状态获取中 =============")
|
||
s = requests.session()
|
||
login(s, url1, client_id1, client_secret1)
|
||
ztasks = getitem(s, url1, "open")
|
||
enable_list = []
|
||
disable_list = []
|
||
for i in ztasks:
|
||
if i['isDisabled'] == 0:
|
||
enable_list.append(i)
|
||
else:
|
||
disable_list.append(i)
|
||
enable_tlid = []
|
||
enable_tname = []
|
||
enable_tcommand = []
|
||
for j in enable_list:
|
||
try:
|
||
enable_tlid.append(j['_id'])
|
||
except:
|
||
enable_tlid.append(j['id'])
|
||
enable_tname.append(j['name'])
|
||
enable_tcommand.append(j['command'])
|
||
disable_tlid = []
|
||
disable_tname = []
|
||
disable_tcommand = []
|
||
for j in disable_list:
|
||
try:
|
||
disable_tlid.append(j['_id'])
|
||
except:
|
||
disable_tlid.append(j['id'])
|
||
disable_tname.append(j['name'])
|
||
disable_tcommand.append(j['command'])
|
||
print("=======主容器任务数量：{}，启用任务{}，禁用任务{}===========".format(len(ztasks), len(enable_tname), len(disable_tname)))
|
||
|
||
print()
|
||
|
||
#分容器
|
||
count = 0
|
||
for cc in urllist:
|
||
print("======== 分容器{}同步状态中 ============".format(count+1))
|
||
a = requests.session()
|
||
login(a, urllist[count], client_ids[count], client_secrets[count])
|
||
ftasks = getitem(a, urllist[count], "open")
|
||
tlid = []
|
||
tname = []
|
||
tcommand = []
|
||
ct = 0
|
||
for i in ftasks:
|
||
try:
|
||
tlid.append(i['_id'])
|
||
except:
|
||
tlid.append(i['id'])
|
||
tname.append(i['name'])
|
||
tcommand.append(i['command'])
|
||
|
||
# 启用
|
||
enable = []
|
||
e_name = []
|
||
over_e = []
|
||
for k in enable_tcommand:
|
||
try:
|
||
p = tcommand.index(k)
|
||
enable.append(tlid[p])
|
||
e_name.append(tname[p])
|
||
except:
|
||
p = enable_tcommand.index(k)
|
||
over_e.append(enable_tname[p])
|
||
|
||
# 禁用
|
||
disable = []
|
||
d_name = []
|
||
over_d = []
|
||
for k in disable_tcommand:
|
||
try:
|
||
p = tcommand.index(k)
|
||
disable.append(tlid[p])
|
||
d_name.append(tname[p])
|
||
except:
|
||
p = disable_tcommand.index(k)
|
||
over_d.append(disable_tname[p])
|
||
|
||
#同步
|
||
synchronous_tasks_enable(a, urllist[count], 'open', enable)
|
||
synchronous_tasks_disable(a, urllist[count], 'open', disable)
|
||
|
||
#通知
|
||
print('已同步状态的任务数量：{}'.format(len(e_name) + len(d_name)))
|
||
print('分容器不存在的任务无法被同步，无法同步任务数量为：{}'.format(len(over_e) + len(over_d)))
|
||
print("无法同步任务名如下：")
|
||
for t in over_e:
|
||
print(t)
|
||
for t in over_d:
|
||
print(t)
|
||
count += 1
|
||
print('========= 分容器{}同步状态完毕 ============='.format(count))
|
||
print()
|
||
|
||
print('同步完毕')
|
||
|
||
|
||
|
||
|