#!/usr/bin/env python
#coding=utf-8

# 加載核心SDK
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 = 'xxxxxxx'
SECRET = 'xxxxxx'

# 地區節點 可選地區取決于你的阿里云帳號等級，普通用戶只有四個，分別是杭州、上海、深圳、河北，具體參考官網API
regionId = 'cn-hangzhou'

# 配置認證信息
client = AcsClient(ID, SECRET, regionId)

# 設置主域名
DomainName = 'example.com'

# 子域名列表 列表參數可根據實際需求增加或減少值
SubDomainList = ['a', 'b', 'c']

# 獲取外網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")

  # 指定查記的域名 格式為 'test.example.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)

# 刪除記錄測試
# delDomainRecord(client,'b.jsoner.com')

# 新增或更新記錄測試
# setDomainRecord(client,'192.168.3.222','a',DomainName)

# 獲取記錄測試
# print (getDomainInfo(DomainName, 'y'))

# 批量獲取記錄測試
# for x in SubDomainList:
#   print (getDomainInfo(DomainName, x))

# 獲取外網ip地址測試
# print ('(' + getIp() + ')')