1+ # -*- coding:utf-8 -*-
2+ 3+ import optparse ,os ,json
4+ from subprocess import *
5+ 6+ class DomainHandler (object ):
7+ def __init__ (self ):
8+ pass
9+ 10+ def exec_cmd (self ,cmd ):
11+ res = Popen (cmd , shell = True , stdout = PIPE )
12+ ret = res .communicate ()[0 ].decode ('utf-8' )
13+ return ret .strip ()
14+ 15+ def domain_info (self ):
16+ cmd = 'curl -s https://dnsapi.cn/Domain.List -d "login_token=391845,92f408bb5343e&format=json"'
17+ data = json .loads (self .exec_cmd (cmd ))
18+ print (data )
19+ for item in data ['domains' ]:
20+ print ('%s:%s' % (item ['name' ], item ['id' ]))
21+ 22+ def add_Arecord (self ,domain_id ,sub_domain ,record_type ,address ):
23+ print (domain_id ,sub_domain ,record_type ,address )
24+ cmd2 = "curl -s -X POST https://dnsapi.cn/Record.Create -d 'login_token=391845,92f408bb5343e&format=json&domain_id={0}&sub_domain={1}&record_type={2}&record_line_id=0&value={3}'" .format (
25+ domain_id , sub_domain , record_type , address )
26+ r = json .loads (self .exec_cmd (cmd2 ))
27+ print (r ['status' ]['message' ])
28+ 29+ def add (self ):
30+ self .domain_info ()
31+ while tag :
32+ self .domain_id = input ('033円 [1;42m输入域名ID:033円 [0m' ).strip ()
33+ if self .domain_id == 'q' :
34+ break
35+ if not self .domain_id or not self .domain_id .isdigit ():
36+ print ('033円 [31merror id033円 [0m' )
37+ continue
38+ self .sub_domain = input ('033円 [1;42m子域名[@或*等]:033円 [0m' ).strip ()
39+ self .record_type = input ('033円 [1;42m类型[A或CNAME]:033円 [0m' ).strip ()
40+ self .address = input ('033円 [1;42m记录值(ip或域名):033円 [0m' ).strip ()
41+ 42+ if not self .sub_domain or not self .record_type or not self .address :
43+ print ('033円 [31m参数不能为空033円 [0m' )
44+ continue
45+ self .add_Arecord (self .domain_id ,self .sub_domain ,self .record_type ,self .address )
46+ if self .domain_id == 'q' or self .record_type == 'q' or self .address == 'q' :
47+ self .tag = False
48+ break
49+ 50+ def get_records (self ):
51+ self .domain_info ()
52+ flag = True
53+ while tag :
54+ if not flag :
55+ break
56+ self .domain_id = input ('033円 [1;42m输入域名ID:033円 [0m' ).strip ()
57+ if self .domain_id == 'q' :
58+ break
59+ if not self .domain_id or not self .domain_id .isdigit ():
60+ print ('033円 [31merror id033円 [0m' )
61+ continue
62+ self .sub_domain = input ('033円 [1;42m子域名[@或*等]:033円 [0m' ).strip ()
63+ self .record_type = input ('033円 [1;42m类型[A或CNAME]:033円 [0m' ).strip ()
64+ cmd3 = "curl -s -X POST https://dnsapi.cn/Record.List -d 'login_token=391845,92f408bb5343e&format=json&domain_id={0}&sub_domain={1}&record_type={2}&offset=0&length=3'" .format (
65+ self .domain_id , self .sub_domain , self .record_type )
66+ records = json .loads (self .exec_cmd (cmd3 ))
67+ try :
68+ print ('033円 [33m共%s条%s记录033円 [0m' % (len (records ['records' ]), self .record_type ))
69+ except Exception as e :
70+ print ('033円 [31m查无此记录033円 [0m' )
71+ continue
72+ for record in records ['records' ]:
73+ print ('033円 [35mID{0}: {1}{split}{2}{split}{3}033円 [0m' .format (record ['id' ], record ['name' ], record ['type' ],record ['value' ], split = ' ' * 10 ))
74+ return records
75+ 76+ def mod (self ):
77+ records = self .get_records ()
78+ while tag :
79+ record_id = input ('033円 [1;42m输入record ID:033円 [0m' ).strip ()
80+ if record_id == 'q' :
81+ break
82+ value = input ("033円 [1;42m输入新的record value:033円 [0m" ).strip ()
83+ if value == 'q' :
84+ break
85+ cmd4 = "curl -s -X POST https://dnsapi.cn/Record.Modify -d 'login_token=391845,92f408bb5343e&format=json&domain_id={0}&record_id={1}&sub_domain={2}&value={3}&record_type={4}&record_line_id=0'" .format (self .domain_id ,record_id ,self .sub_domain ,value ,self .record_type )
86+ r = json .loads (self .exec_cmd (cmd4 ))
87+ print (r ['status' ]['message' ])
88+ flag = False
89+ break
90+ def delete (self ):
91+ records = self .get_records ()
92+ while tag :
93+ record_id = input ('033円 [1;42m输入record ID:033円 [0m' ).strip ()
94+ if record_id == 'q' :
95+ break
96+ cmd5 = "curl -s -X POST https://dnsapi.cn/Record.Remove -d 'login_token=391845,92f408bb5343e&format=json&domain_id={0}&record_id={1}'" .format (self .domain_id ,record_id )
97+ r = json .loads (self .exec_cmd (cmd5 ))
98+ print (r ['status' ]['message' ])
99+ flag = False
100+ break
101+ 102+ dic = {
103+ '1' :DomainHandler ().add ,
104+ '2' :DomainHandler ().mod ,
105+ '3' :DomainHandler ().delete
106+ }
107+ 108+ tag = True
109+ while tag :
110+ print ('''
111+ 1.增加
112+ 2.修改
113+ 3.删除
114+ q.退出
115+ ''' )
116+ choice = input ('033円 [1;42m输入选项:033円 [0m' ).strip ()
117+ if not choice :
118+ continue
119+ if choice == 'q' :
120+ break
121+ if choice in dic :
122+ dic [choice ]()
123+ 124+ else :
125+ print ('033円 [31m选项不存在033円 [0m' )
0 commit comments