-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
126 lines (98 loc) · 4.46 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
import time
import threading
import json
import requests
from aliyunsdkcore.client import AcsClient
from aliyunsdkalidns.request.v20150109.DescribeDomainRecordsRequest import DescribeDomainRecordsRequest
from aliyunsdkalidns.request.v20150109.UpdateDomainRecordRequest import UpdateDomainRecordRequest
class AliDDNS(object):
def __init__(self,
domain, record,
id, secret,
sleep = 600,
ipv4 = True, ipv6 = True,
v4api = "https://ipv4.gh.ink",
v6api = "https://ipv6.gh.ink"):
assert domain.count(".") == 1 and \
not domain.startswith(".") and \
not domain.endswith(".")
self.domain = domain
self.record = record
self.sleep = sleep
self.ipv4 = ipv4
self.ipv6 = ipv6
self.v4api = v4api
self.v6api = v6api
self.__client = AcsClient(id, secret, 'cn-hangzhou')
daemon = threading.Thread(target=self.__daemon, name="Daemon Thread")
daemon.daemon = True
daemon.start()
def __daemon(self):
while True:
self.refresh()
time.sleep(self.sleep)
def refresh(self):
# Get local public ipv4 address
if self.ipv4:
try:
self.__v4 = requests.get(self.v4api).text
except Exception as e:
print("Failed to get local public ipv4 address,", e)
else:
print("Your ipv4 address is:", self.__v4)
describe_v4 = DescribeDomainRecordsRequest()
describe_v4.set_accept_format('json')
describe_v4.set_DomainName(self.domain)
describe_v4.set_RRKeyWord(self.record)
describe_v4.set_TypeKeyWord("A")
describeback_v4 = self.__client.do_action_with_exception(describe_v4)
recordlist_v4 = json.loads(describeback_v4)
recordid_v4 = recordlist_v4['DomainRecords']['Record'][0]['RecordId']
value_v4 = recordlist_v4['DomainRecords']['Record'][0]['Value']
if value_v4 != self.__v4:
print('Your ipv4 address has been changed, update now.')
update_v4 = UpdateDomainRecordRequest()
update_v4.set_accept_format('json')
update_v4.set_RecordId(recordid_v4)
update_v4.set_Value(self.__v4)
update_v4.set_Type("A")
update_v4.set_RR(self.record)
self.__client.do_action_with_exception(update_v4)
print('Your ipv4 record has been updated successfully.')
# Get local public ipv6 address
if self.ipv6:
try:
self.__v6 = requests.get(self.v6api).text
except Exception as e:
print("Failed to get local public ipv6 address,", e)
else:
print("Your ipv6 address is:", self.__v6)
describe_v6 = DescribeDomainRecordsRequest()
describe_v6.set_accept_format('json')
describe_v6.set_DomainName(self.domain)
describe_v6.set_RRKeyWord(self.record)
describe_v6.set_TypeKeyWord("AAAA")
describeback_v6 = self.__client.do_action_with_exception(describe_v6)
recordlist_v6 = json.loads(describeback_v6)
recordid_v6 = recordlist_v6['DomainRecords']['Record'][0]['RecordId']
value_v6 = recordlist_v6['DomainRecords']['Record'][0]['Value']
if value_v6 != self.__v6:
print('Your ipv6 address has been changed, update now.')
update_v6 = UpdateDomainRecordRequest()
update_v6.set_accept_format('json')
update_v6.set_RecordId(recordid_v6)
update_v6.set_Value(self.__v6)
update_v6.set_Type("AAAA")
update_v6.set_RR(self.record)
self.__client.do_action_with_exception(update_v6)
print('Your ipv6 record has been updated successfully.')
if __name__ == "__main__":
dns = AliDDNS("MainDomain.com", "SubDomain", "AccessKey ID", "AccessKey Secret")
while True:
cmd = input()
if cmd == "exit":
break
elif cmd == "refresh":
dns.refresh()
else:
print("Unknown command.")