更新日誌： 1.解決一個域名下擁有大量記錄時的執行效率問題 2.優化代碼邏輯 3.增加批量和cli兩種方式 最新代碼： 批量：https://github.com/coocla/dnspod-batch Cli：https://github.com/coocla/dnspod-cli
做遊戲的哥們都知道，每天大量的開服都意味着每天有大量的域名需要解析，這下可苦了我們公司的妹子，每天解析域名點的手疼，浪費生命啊。理所應當的一個批量解析的需求就轟轟烈烈的誕生了。
參考文檔：
1.https://github.com/DNSPod/dnspod-python
2.https://www.dnspod.cn/docs/index.html
域名列表文本內容格式：
域名 電信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:[email protected] #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()