利用Dnspod api批量更新添加DNS解析【python脚本】
16lz
2021-01-22
需求:每天有大量的域名需要解析,变更,负责解析域名的妹子实属伤不起,批量更新添加的需求诞生了。
参考文档:https://github.com/DNSPod/dnspod-python
https://www.dnspod.cn/docs/index.html
注1:依据个人笨拙的编程思想,码了这么一堆代码,总算实现的添加更新功能,潜在bug尚不清楚。欢迎有兴趣的朋友一起讨论。
注2:以下代码适用于python2.6以下,如果python2.6以上的需要将代码中的:except Exception,e修改为:except Exception as e
注3:在dnsapi.apicn源代码中Ddns类中,需要把#record_type=record_type,这一行注释掉进行安装,已安装的需要把pyc文件删除,重新import即可
域名列表格式要求如下:
域名 电信IP,联通IP
www.baidu.com 1.1.1.1,2.2.2.2
www.sohu.com ,3.3.3.3
music.baidu.com 4.4.4.4,
代码如下:
#!/usr/bin/env python#-*- coding:utf-8 -*-
#who:czlinux@163.com
#when:2013-08-14
import os,sys
from dnspod.apicn import *
domain_list=[]
domain_dic={}
illegal_list=[]
login_email="XXXXX"
password="XXXXX"
domain_file="/root/dns"
def domain_index(domain_file):
all_domain_list=[]
for line in file(domain_file).readlines():
if line == '\n':
continue
if line[-1] == '\n':
line=line[:-1]
all_domain_list.append('.'.join(line.split()[0].split('.')[-2:]))
domain_list=list(set(all_domain_list))
api=DomainList(email=login_email,password=password)
all_msg=api().get("domains")
exists_list=[]
for i in range(0,len(all_msg)):
exists_list.append(all_msg[i].get("name"))
for dom in domain_list:
if dom not in exists_list:
api_create=DomainCreate(dom,email=login_email,password=password)
try:
domain_create=api_create()
all_msg=api().get("domains")
for n in range(0,len(all_msg)):
if dom == all_msg[n].get("name"):
dom_id=all_msg[n].get("id")
domain_dic[dom]=dom_id
except Exception,e:
illegal_list.append(dom)
#print "[Error:] %s" % e
print """\033[31m[Error]: Create domain [ %s ] Fail
.....You may not have permission to create it ! \033[0m""" % dom
else:
all_msg=api().get("domains")
for n in range(0,len(all_msg)):
if dom == all_msg[n].get("name"):
dom_id=all_msg[n].get("id")
domain_dic[dom]=dom_id
def add_record(domain,ip):
if len(ip.split(',')) == 2:
tel_ip=ip.split(',')[0]
unic_ip=ip.split(',')[1]
else:
print "\033[31m[Error]: IP format error...\033[0m"
#sys.exit()
sub_dom='.'.join(domain.split('.')[:-2])
dom_name=['.'.join(domain.split('.')[-2:])]
for k,v in domain_dic.iteritems():
if k in dom_name:
domain_id=domain_dic[k]
#print sub_dom #debug
api_subdom=RecordList(domain_id,email=login_email,password=password)
try:
rec_list=api_subdom().get("records")
tmp_sub=[]
try:
rec_list=api_subdom().get("records")
tmp_sub=[]
for i in range(0,len(rec_list)):
tmp_sub.append(rec_list[i].get("name"))
if sub_dom in tmp_sub:
#print "[Error]: Sub_domain [ %s.%s ] exists..." % (sub_dom,'.'.join(domain.split('.')[-2:]))
update_record(rec_list,sub_dom,domain_id,'.'.join(domain.split('.')[-2:]),tel_ip,unic_ip)
else:
tel_record=RecordCreate(sub_dom,"A",u'默认'.encode("utf8"),tel_ip,600,domain_id=domain_id,email=login_email,password=password)
unic_record=RecordCreate(sub_dom,"A",u'联通'.encode("utf8"),unic_ip,600,domain_id=domain_id,email=login_email,password=password)
tel_record()
print "\033[32m[Notice]: Add record [ %s.%s ] -- [ %s ] successful!\033[0m" % (sub_dom,'.'.join(domain.split('.')[-2:]),tel_ip)
unic_record()
print "\033[32m[Notice]: Add record [ %s.%s ] -- [ %s ] successful!\033[0m" % (sub_dom,'.'.join(domain.split('.')[-2:]),unic_ip)
except:
print "\033[31m[Error]: Add sub_domain record some error...\033[0m"
#sys.exit()
def update_record(rec_list,sub_dom,domain_id,dom,tel_ip="",unic_ip=""):
for nu in range(0,len(rec_list)):
if sub_dom == rec_list[nu].get("name"):
record_id=rec_list[nu].get("id")
if rec_list[nu].get("line") == u"默认" and tel_ip:
try:
tel_update_record=RecordDdns(record_id,sub_dom,"默认",domain_id=domain_id,ttl=600,value=tel_ip,email=login_email,password=password)
tel_update_record()
print "\033[32m[Notice]: Update [ %s.%s ] --- [ %s ] Successful!\033[0m" % (sub_dom,dom,tel_ip)
except Exception,e:
print "[Error]: %s" % e
print "\033[31m.......... Update [ %s.%s ] --- [ %s ] Fail!\033[0m" % (sub_dom,dom,tel_ip)
elif rec_list[nu].get("line") == u"联通" and unic_ip:
try:
unic_update_record=RecordDdns(record_id,sub_dom,"联通",domain_id=domain_id,ttl=600,value=unic_ip,email=login_email,password=password)
unic_update_record()
print "\033[32m[Notice]: Update [ %s.%s ] --- [ %s ] Successful!\033[0m" % (sub_dom,dom,unic_ip)
except Exception,e:
print "[Error]: %s" % e
print "\033[31m.......... Update [ %s.%s ] --- [ %s ] Fail!\033[0m" % (sub_dom,dom,unic_ip)
if __name__ == '__main__':
if os.path.isfile(domain_file):
domain_index(domain_file)
for line in file(domain_file).readlines():
if line == '\n':
continue
if line[-1] == '\n':
line=line[:-1]
if illegal_list:
for g in illegal_list:
try:
line.split()[0].index(g)
except:
add_record(line.split()[0],line.split()[1])
else:
add_record(line.split()[0],line.split()[1])
else:
print "\033[31m[Error]: Domain file [ %s ]not found...\033[0m" % domain_file
sys.exit()
本文出自 “My---Dream.*” 博客,请务必保留此出处http://grass51.blog.51cto.com/4356355/1273375
更多相关文章
- 怎样写贪吃蛇小游戏?用100行python代码轻松解决!
- 建模分析之机器学习算法(附python&R代码)
- 读取python中的unicode文件,该文件以与python源代码相同的方式声
- 自动完成在VS代码和Python中的自动化对象
- 在生产中是否应该减少服务器代码?
- 支持c和python之间的跨语言(c)标记的代码编辑器
- 【小白自学笔记】【机器学习实战】【Python代码逐行理解】CH02
- Python——域名解析成IP地址
- 三个猜数字游戏代码(Python)