from dns.cf_api import CfApi
from phxweb.settings import CF_URL
#获取当前目录
current_dir = os.path.abspath(os.path.dirname(__file__))
#print basedir
cursor = django.db.connection.cursor()
with open('%s/api_info.txt' % current_dir, 'r') as f:
contents = f.readlines()
for info in contents:
if not '#' in info and info.replace(' ', '') != '\n':
cf_acc = cf_account.objects.filter(name=info.split()[3]).first()
api = CfApi(CF_URL, cf_acc.email, cf_acc.key)
zone_id = api.GetZoneId('.'.join(
info.split()[2].split('.')[-2:]))['zone_id']
record_id = api.GetDnsRecordId(zone_id, info.split()[2])
try:
#update = domain_info.objects.filter(product=info.split()[0], client=info.split()[1], domain=info.split()[2], route=info.split()[4]).first()
#update = domain_info.objects.filter(product=info.split()[0], client=info.split()[1], domain=info.split()[2], content=info.split()[5]).first()
#update.zone_id = zone_id
#update.route = info.split()[4]
#update.record_id = record_id
#update.status = info.split()[6]
#update.route_status = info.split()[7]
#update.save()
#print info.split()[0], info.split()[2], info.split()[4], 'update success.'
insert = domain_info(product=info.split()[0],
client=info.split()[1],
def receive(self, text=None, bytes=None, **kwargs):
"""
Called when a message is received with either text or bytes
filled out.
"""
#self.close()
data = json.loads(self.message['text'])
self.clientip = '127.0.0.1'
self.username = self.message.user.username
try:
self.role = self.message.user.userprofile.role
except:
self.role = 'none'
### step one ##
info = {}
info['result'] = []
info['step'] = 'one'
self.message.reply_channel.send({'text': json.dumps(info)})
#time.sleep(2)
### two step ###
info['step'] = 'two'
cdn_d = {}
cdns = cdn_t.objects.all()
for cdn in cdns:
cdn_d[cdn.get_name_display() + "_" + cdn.account] = {
'name': cdn.get_name_display(),
'domain': [],
'secretid': str(cdn.secretid),
'secretkey': str(cdn.secretkey),
'failed': [],
'success': [],
}
cf_d = {}
cfs = cf_t.objects.all()
for cf in cfs:
cf_d[cf.name] = {
'name': cf.name,
'domain': [],
'email': str(cf.email),
'key': str(cf.key),
'failed': [],
'success': [],
}
domain_l = domains.objects.filter(id__in=data['domain']).all()
logger.info(domain_l)
for domain in domain_l:
for cdn in domain.cdn.all():
cdn_d[cdn.get_name_display() + "_" +
cdn.account]['domain'].append(
urlparse.urlsplit(domain.name).scheme + "://" +
urlparse.urlsplit(domain.name).netloc)
for cf in domain.cf.all():
cf_d[cf.name]['domain'].append(
urlparse.urlsplit(domain.name).scheme + "://" +
urlparse.urlsplit(domain.name).netloc)
#logger.info(cdn_d)
for cdn in cdn_d:
info['cdn'] = cdn
if cdn_d[cdn]['domain']:
#开始清缓存，判断CDN接口是否存在
if cdn_d[cdn]['name'] == "tencent":
req = tcApi(cdn_d[cdn]['secretid'],
cdn_d[cdn]['secretkey'])
elif cdn_d[cdn]['name'] == "wangsu":
req = wsApi(cdn_d[cdn]['secretid'],
cdn_d[cdn]['secretkey'])
else:
info['result'] = ["CDN 接口不存在！"]
cdn_d[cdn]['failed'].append("%s: 接口不存在！" % cdn)
self.message.reply_channel.send({'text': json.dumps(info)})
continue
while len(cdn_d[cdn]['domain']) != 0:
domains_c = cdn_d[cdn]['domain'][:10]
cdn_d[cdn]['domain'] = cdn_d[cdn]['domain'][10:]
for uri in data['uri']:
result, status = req.purge(domains_c, uri)
if status:
info['result'] = [
cdn + ": " + domain + uri + ": 清缓存成功。"
for domain in domains_c
]
cdn_d[cdn]['success'] += [
domain + uri for domain in domains_c
]
else:
info['result'] = [
cdn + ": " + domain + uri + ": 清缓存失败！"
for domain in domains_c
]
cdn_d[cdn]['failed'] += [
domain + uri for domain in domains_c
]
self.message.reply_channel.send(
{'text': json.dumps(info)})
for cf in cf_d:
info['cf'] = cf
if cf_d[cf]['domain']:
#开始清CF缓存
req = CfApi(CF_URL, cf_d[cf]['email'], cf_d[cf]['key'])
for domain in cf_d[cf]['domain']:
zone = ".".join(domain.split(".")[-2:])
zone_id = req.GetZoneId(zone)['zone_id']
if not zone_id:
info['result'] = [
"CloudFlare_" + cf + ": " + domain + ": 清缓存失败！"
]
cf_d[cf]['failed'] += [domain]
result = req.purge(zone_id)
if result['success']:
info['result'] = [
"CloudFlare_" + cf + ": " + domain + ": 清缓存成功。"
]
cf_d[cf]['success'] += [domain]
else:
info['result'] = [
"CloudFlare_" + cf + ": " + domain + ": 清缓存失败！"
]
cf_d[cf]['failed'] += [domain]
self.message.reply_channel.send({'text': json.dumps(info)})
info['step'] = 'final'
self.message.reply_channel.send({'text': json.dumps(info)})
for cdn in cdn_d:
if cdn_d[cdn]['failed']:
message["text"] = cdn_d[cdn]['failed']
message['caption'] = cdn + ': 域名缓存清理失败!'
sendTelegramRe(message)
if cdn_d[cdn]['success']:
message["text"] = cdn_d[cdn]['success']
message['caption'] = cdn + ': 域名缓存清理成功。'
sendTelegramRe(message)
for cf in cf_d:
if cf_d[cf]['failed']:
message["text"] = cf_d[cf]['failed']
message['caption'] = "CloudFlare_" + cf + ': 域名缓存清理失败!'
sendTelegramRe(message)
if cf_d[cf]['success']:
message["text"] = cf_d[cf]['success']
message['caption'] = "CloudFlare_" + cf + ': 域名缓存清理成功。'
sendTelegramRe(message)
self.close()