142 lines
6.0 KiB
Python
142 lines
6.0 KiB
Python
#!/usr/bin/env python
|
||
#coding=utf-8
|
||
|
||
# 加载核心SDK
|
||
#from aliyunsdkcore import client
|
||
#from aliyunsdksts.request.v20150401 import AssumeRoleRequest
|
||
#import json
|
||
#import oss2
|
||
|
||
from aliyunsdkcore.client import AcsClient
|
||
from aliyunsdkcore.acs_exception.exceptions import ClientException
|
||
from aliyunsdkcore.acs_exception.exceptions import ServerException
|
||
|
||
# 加载获取 、 新增、 更新、 删除接口
|
||
from aliyunsdkalidns.request.v20150109 import DescribeSubDomainRecordsRequest, AddDomainRecordRequest, UpdateDomainRecordRequest, DeleteDomainRecordRequest
|
||
|
||
# 加载内置模块
|
||
import json,urllib
|
||
|
||
# AccessKey 和 Secret 建议使用 RAM 子账户的 KEY 和 SECRET 增加安全性
|
||
ID = ''
|
||
SECRET = ''
|
||
|
||
# 地区节点 可选地区取决于你的阿里云帐号等级，普通用户只有四个，分别是杭州、上海、深圳、河北，具体参考官网API
|
||
regionId = 'cn-hangzhou'
|
||
|
||
# 配置认证信息
|
||
client = AcsClient(ID, SECRET, regionId)
|
||
|
||
# 设置主域名
|
||
DomainName = ''
|
||
|
||
# 子域名列表 列表参数可根据实际需求增加或减少值
|
||
SubDomainList = ['@','www']
|
||
|
||
# 获取外网IP 三个地址返回的ip地址格式各不相同，3322 的是最纯净的格式， 备选1为 json格式 备选2 为curl方式获取 两个备选地址都需要对获取值作进一步处理才能使用
|
||
def getIp():
|
||
# 备选地址: 1. http://pv.sohu.com/cityjson?ie=utf-8 2. curl -L tool.lu/ip
|
||
with urllib.request.urlopen('http://www.3322.org/dyndns/getip') as response:
|
||
html = response.read()
|
||
ip = str(html, encoding='utf-8').replace("\n", "")
|
||
return ip
|
||
|
||
# 查询记录
|
||
def getDomainInfo(SubDomain):
|
||
request = DescribeSubDomainRecordsRequest.DescribeSubDomainRecordsRequest()
|
||
request.set_accept_format('json')
|
||
|
||
# 设置要查询的记录类型为 A记录 官网支持A / CNAME / MX / AAAA / TXT / NS / SRV / CAA / URL隐性(显性)转发 如果有需要可将该值配置为参数传入
|
||
request.set_Type("A")
|
||
#request.set_Type("www")
|
||
# 指定查记的域名 格式为 'test.binghe.com'
|
||
request.set_SubDomain(SubDomain)
|
||
|
||
response = client.do_action_with_exception(request)
|
||
response = str(response, encoding='utf-8')
|
||
|
||
# 将获取到的记录转换成json对象并返回
|
||
return json.loads(response)
|
||
|
||
# 新增记录 (默认都设置为A记录，通过配置set_Type可设置为其他记录)
|
||
def addDomainRecord(client,value,rr,domainname):
|
||
request = AddDomainRecordRequest.AddDomainRecordRequest()
|
||
request.set_accept_format('json')
|
||
|
||
# request.set_Priority('1') # MX 记录时的必选参数
|
||
request.set_TTL('600') # 可选值的范围取决于你的阿里云账户等级，免费版为 600 - 86400 单位为秒
|
||
request.set_Value(value) # 新增的 ip 地址
|
||
request.set_Type('A') # 记录类型
|
||
request.set_RR(rr) # 子域名名称
|
||
request.set_DomainName(domainname) #主域名
|
||
|
||
# 获取记录信息，返回信息中包含 TotalCount 字段，表示获取到的记录条数 0 表示没有记录， 其他数字为多少表示有多少条相同记录，正常有记录的值应该为1，如果值大于1则应该检查是不是重复添加了相同的记录
|
||
response = client.do_action_with_exception(request)
|
||
response = str(response, encoding='utf-8')
|
||
relsult = json.loads(response)
|
||
return relsult
|
||
|
||
# 更新记录
|
||
def updateDomainRecord(client,value,rr,record_id):
|
||
request = UpdateDomainRecordRequest.UpdateDomainRecordRequest()
|
||
request.set_accept_format('json')
|
||
|
||
# request.set_Priority('1')
|
||
request.set_TTL('600')
|
||
request.set_Value(value) # 新的ip地址
|
||
request.set_Type('A')
|
||
request.set_RR(rr)
|
||
request.set_RecordId(record_id) # 更新记录需要指定 record_id ，该字段为记录的唯一标识，可以在获取方法的返回信息中得到该字段的值
|
||
|
||
response = client.do_action_with_exception(request)
|
||
response = str(response, encoding='utf-8')
|
||
return response
|
||
|
||
# 删除记录
|
||
def delDomainRecord(client,subdomain):
|
||
info = getDomainInfo(subdomain)
|
||
if info['TotalCount'] == 0:
|
||
print('没有相关的记录信息，删除失败！')
|
||
elif info["TotalCount"] == 1:
|
||
print('准备删除记录')
|
||
request = DeleteDomainRecordRequest.DeleteDomainRecordRequest()
|
||
request.set_accept_format('json')
|
||
|
||
record_id = info["DomainRecords"]["Record"][0]["RecordId"]
|
||
request.set_RecordId(record_id) # 删除记录需要指定 record_id ，该字段为记录的唯一标识，可以在获取方法的返回信息中得到该字段的值
|
||
result = client.do_action_with_exception(request)
|
||
print('删除成功，返回信息：')
|
||
print(result)
|
||
else:
|
||
# 正常不应该有多条相同的记录，如果存在这种情况，应该手动去网站检查核实是否有操作失误
|
||
print("存在多个相同子域名解析记录值，请核查后再操作！")
|
||
|
||
# 有记录则更新，没有记录则新增
|
||
def setDomainRecord(client,value,rr,domainname):
|
||
info = getDomainInfo(rr + '.' + domainname)
|
||
if info['TotalCount'] == 0:
|
||
print('准备添加新记录')
|
||
add_result = addDomainRecord(client,value,rr,domainname)
|
||
print(add_result)
|
||
elif info["TotalCount"] == 1:
|
||
print('准备更新已有记录')
|
||
record_id = info["DomainRecords"]["Record"][0]["RecordId"]
|
||
cur_ip = getIp()
|
||
old_ip = info["DomainRecords"]["Record"][0]["Value"]
|
||
if cur_ip == old_ip:
|
||
print ("新ip与原ip相同,不更新!")
|
||
else:
|
||
update_result = updateDomainRecord(client,value,rr,record_id)
|
||
print('更新成功，返回信息：')
|
||
print(update_result)
|
||
else:
|
||
# 正常不应该有多条相同的记录，如果存在这种情况，应该手动去网站检查核实是否有操作失误
|
||
print("存在多个相同子域名解析记录值，请核查删除后再操作!")
|
||
|
||
|
||
IP = getIp()
|
||
|
||
# 循环子域名列表进行批量操作
|
||
for x in SubDomainList:
|
||
setDomainRecord(client,IP,x,DomainName)
|