#!/usr/bin/python
#-*- coding: utf-8 -*-
import os,requests,json
import re,xlsxwriter,time
import xlrd
#curl -X POST https://dnsapi.cn/Info.Version -d ‘login_token=LOGIN_TOKEN&format=json‘
#时间
date=time.strftime("%Y-%m-%dT%H-%M-%S", time.localtime())
#文件名
name=‘域名筛选-{0}.xlsx‘.format(date)
url=‘https://dnsapi.cn/Info.Version‘
token=‘‘
domail_name=‘daddymami.net‘
data={‘login_token‘:‘{0}‘.format(token),‘format‘:‘json‘}
r = requests.post(url,data=data)
if r.json()[‘status‘][‘code‘] != ‘1‘:
print(‘登录失败,检查网站,token,id‘)
exit(0)
else:
print(‘登录成功,正在执行脚本‘)
url=‘https://dnsapi.cn/Domain.List‘
data={‘login_token‘:‘{0}‘.format(token),‘format‘:‘json‘,‘type‘:‘all‘}
r = requests.post(url,data=data)
json_data=r.json()
#域名名字和id
id_name=dict()
domain_name_all=json_data[‘domains‘]
for i in domain_name_all:
id_name[str(i[‘name‘])]=i[‘id‘]
print(id_name[‘daddymami.net‘])
url=‘https://dnsapi.cn/Record.Create‘
ERROR=dict()
#指定文件位置
execl_file=‘‘
#指定页
table_name=‘‘
bk = xlrd.open_workbook(execl_file)
shxrange = range(bk.nsheets)
try:
table = bk.sheet_by_name(table_name)
except:
print ("no sheet in %s named %s"%(execl_file,table_name))
nrows = table.nrows
num=1
for row in range(1,nrows):
host_value = table.cell_value(row,1)
Record_type = table.cell_value(row,2)
line_value = table.cell_value(row,3)
Record_value = table.cell_value(row,4)
MX = table.cell_value(row,5)
if host_value != ‘‘:
if MX == ‘-‘:
data={‘login_token‘:‘{0}‘.format(token),‘format‘:‘json‘,‘domain_id‘:id_name[domail_name],‘sub_domain‘:host_value,‘record_line‘:line_value,‘record_type‘:Record_type,‘value‘:Record_value}
r = requests.post(url,data=data)
output = open(‘data%s.json‘%date, ‘w‘)
ERROR[row]={"code":int(r.json()[‘status‘][‘code‘]),"host":str(host_value)}
output.write(json.dumps(ERROR))
num +=1
else:
data={‘login_token‘:‘{0}‘.format(token),‘format‘:‘json‘,‘domain_id‘:id_name[domail_name],‘sub_domain‘:host_value,‘record_line‘:line_value,‘record_type‘:Record_type,‘value‘:Record_value,‘mx‘:MX}
r = requests.post(url,data=data)
output = open(‘data%s.json‘%date, ‘w‘)
ERROR[row]={"code":int(r.json()[‘status‘][‘code‘]),"host":str(host_value)}
output.write(json.dumps(ERROR))
num +=1
output.close( )
添加完成后,在做对比，看哪些没添加成功
#!/usr/bin/python
#-*- coding: utf-8 -*-
import os,requests,json
import re,xlsxwriter,time
import xlrd
#curl -X POST https://dnsapi.cn/Info.Version -d ‘login_token=LOGIN_TOKEN&format=json‘
#文件名
date=time.strftime("%Y-%m-%dT%H-%M-%S", time.localtime())
name=‘{0}.xlsx‘.format(date)
def start_data():
#登录测试
url=‘https://dnsapi.cn/Info.Version‘
token=‘‘
domail_name=‘daddymami.net‘
data={‘login_token‘:‘{0}‘.format(token),‘format‘:‘json‘}
r = requests.post(url,data=data)
if r.json()[‘status‘][‘code‘] != ‘1‘:
print(‘登录失败,检查网站,token,id‘)
exit(0)
else:
print(‘登录成功,正在执行脚本‘)
#获取域名ID
url=‘https://dnsapi.cn/Domain.List‘
data={‘login_token‘:‘{0}‘.format(token),‘format‘:‘json‘,‘type‘:‘all‘}
r = requests.post(url,data=data)
json_data=r.json()
#域名名字和id
id_name=dict()
domain_name_all=json_data[‘domains‘]
for i in domain_name_all:
id_name[str(i[‘name‘])]=i[‘id‘]
print(id_name[‘daddymami.net‘])
#获取域名记录
url=‘https://dnsapi.cn/Record.List‘
ERROR=dict()
#对比dnspod与文档的记录是否一致
#指定文件位置
execl_file=‘‘
#指定页
table_name=‘‘
bk = xlrd.open_workbook(execl_file)
shxrange = range(bk.nsheets)
try:
table = bk.sheet_by_name(table_name)
except:
print ("no sheet in %s named %s"%(execl_file,table_name))
nrows = table.nrows
num=1
domail_data=dict()
dict_num=0
for row in range(1,nrows):
host_value = str(table.cell_value(row,1))
Record_type = str(table.cell_value(row,2))
line_value = str(table.cell_value(row,3))
Record_value = str(table.cell_value(row,4))
MX = str(table.cell_value(row,5))
TTL = str(table.cell_value(row,6))
if host_value != ‘‘:
data={‘login_token‘:‘{0}‘.format(token),‘format‘:‘json‘,‘domain_id‘:id_name[domail_name],‘keyword‘:host_value}
r = requests.post(url,data=data)
num +=1
print(host_value,int(r.json()[‘status‘][‘code‘]))
if int(r.json()[‘status‘][‘code‘]) == 1:
num=0
for ii in r.json()[‘records‘]:
#由于len从1计数所以这里循环开始就加1
num+=1
if len(ii) > 0:
if str(ii[‘status‘]) == ‘enabled‘ and str(ii[‘name‘]) == host_value and str(ii[‘value‘]) == Record_value and str(ii[‘line‘]) == line_value and str(ii[‘type‘]) == Record_type:
break
else:
#如果循环的次数等于或等于json里的数据数量,那么意味着这条记录在dnspod不存在,加入到domail_data做记录
if num >= len(r.json()[‘records‘]) :
dict_num+=1
domail_data[dict_num]={‘host_value‘:host_value,‘Record_type‘:Record_type,‘line_value‘:line_value,‘Record_value‘:Record_value,‘MX‘:MX,‘TTL‘:TTL}
num=1
else:
dict_num+=1
domail_data[dict_num]={‘host_value‘:host_value,‘Record_type‘:Record_type,‘line_value‘:line_value,‘Record_value‘:Record_value,‘MX‘:MX,‘TTL‘:TTL}
output = open(‘data%s.json‘%date, ‘w‘)
output.write(json.dumps(domail_data))
output.close( )
print(‘xlsx原始数据%sdata%s.json‘%(os.path.abspath(os.curdir)+"\\",date))
#为了避免xlsxwriter出现故障导致又要重新查询一次dnspod,这里直接写入到json文件做存储,然后在取出,后面出现故障可直接修改open文件名
start_data()
output = open(‘data%s.json‘%date)
domail_data = json.load(output)
#设置文件名
workbook=xlsxwriter.Workbook(name)
#设置宽度
worksheet=workbook.add_worksheet()
worksheet.set_column(‘A:A‘,20)
worksheet.set_column(‘B:B‘,22)
worksheet.set_column(‘C:C‘,20)
worksheet.set_column(‘D:D‘,15)
worksheet.set_column(‘E:E‘,15)
worksheet.set_column(‘F:F‘,15)
#生成报表
title = [u‘主机记录‘,u‘记录类型‘,u‘线路类型‘,u‘记录值‘,u‘MX优先‘,‘TTL‘]
worksheet.write_row(‘A1‘,title)
num = 2
for i in domail_data.keys():
tmp=[domail_data[i][‘host_value‘],domail_data[i][‘Record_type‘],domail_data[i][‘line_value‘],domail_data[i][‘Record_value‘],domail_data[i][‘MX‘],domail_data[i][‘TTL‘]]
worksheet.write_row(‘A{0}‘.format(num),tmp)
num+=1
workbook.close()
#
print(‘执行完毕,文件路径‘+os.path.abspath(os.curdir)+"\\"+name)
原文地址：https://www.cnblogs.com/cainiaoit/p/8582680.html
时间： 2024-08-03 11:24:22