You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.8 KiB
71 lines
2.8 KiB
#!/usr/bin/python3
|
|
import requests, json
|
|
import xlrd,re,sys
|
|
sys.path.append("..")
|
|
from config import consul_token,consul_url
|
|
|
|
def importconsul(row,imptype):
|
|
try:
|
|
if imptype == 'blackbox':
|
|
module, company, project, env, name, instance = row
|
|
data = {
|
|
"id": f"{module}/{company}/{project}/{env}@{name}",
|
|
"name": 'blackbox_exporter',
|
|
"tags": [module],
|
|
"Meta": {'module': module, 'company': company, 'project': project,
|
|
'env': env, 'name': name,'instance': instance}
|
|
}
|
|
elif imptype == 'selfnode':
|
|
vendor,account,region,group,name,instance,os = row
|
|
print(row)
|
|
sid = f"{vendor}/{account}/{region}/{group}@{name}"
|
|
ip = instance.split(':')[0]
|
|
port = instance.split(':')[1]
|
|
data = {
|
|
"id": sid,
|
|
"name": 'selfnode_exporter',
|
|
'Address': ip,
|
|
'port': int(port),
|
|
"tags": [vendor,os],
|
|
"Meta": {'vendor':vendor,'account':account,'region':region,'group':group,
|
|
'name':name,'instance':instance,'os':os},
|
|
"check": {"tcp": instance,"interval": "60s"}
|
|
}
|
|
except Exception as e:
|
|
print("【import】导入失败",e,flush=True)
|
|
return {"code": 50000, "data": f"导入内容格式异常！{row}"}
|
|
headers = {'X-Consul-Token': consul_token}
|
|
|
|
reg = requests.put(f"{consul_url}/agent/service/register", headers=headers, data=json.dumps(data))
|
|
if reg.status_code == 200:
|
|
print({"code": 20000, "data": "增加成功！"},instance,flush=True)
|
|
return {"code": 20000, "data": "增加成功！"}
|
|
else:
|
|
print({"code": 50000, "data": f'{reg.status_code}:{reg.text}'},instance,flush=True)
|
|
return {"code": 50000, "data": f'{reg.status_code}:{reg.text}'}
|
|
|
|
def read_execl(file_contents,imptype):
|
|
data = xlrd.open_workbook(file_contents=file_contents, encoding_override="utf-8")
|
|
table = data.sheets()[0]
|
|
print("【import】开始读取导入文件",flush=True)
|
|
for rownum in range(table.nrows):
|
|
row = table.row_values(rownum)
|
|
if rownum == 0:
|
|
continue
|
|
nrow = []
|
|
for i in row:
|
|
try:
|
|
float(i)
|
|
if i % 1 == 0:
|
|
i = int(i)
|
|
nrow.append(str(i))
|
|
except:
|
|
j = i.strip()
|
|
if i != row[5]:
|
|
j = re.sub('[[ \]`~!\\\#$^/&*=|"{}\':;?]','_',j)
|
|
nrow.append(j)
|
|
imp = importconsul(nrow,imptype)
|
|
if imp['code'] == 50000:
|
|
return imp
|
|
return {"code": 20000, "data": f"导入成功！共导入 {rownum} 条数据。"}
|