deblan/PowerDNS-Admin
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add the adjustment from @CaptainQwark in PR #49

Browse source
This commit is contained in:
Khanh Ngo 2016-06-16 10:31:36 +07:00
parent a14ccec39e
commit af7402096e
1 changed files with 25 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
app/models.py
View file

@ -434,6 +434,7 @@ class Domain(db.Model):
"""
db_domain = Domain.query.all()
list_db_domain = [d.name for d in db_domain]
dict_db_domain = dict((x.name,x) for x in db_domain)
headers = {}
headers['X-API-Key'] = PDNS_API_KEY
@ -461,15 +462,25 @@ class Domain(db.Model):
# update/add new domain
for data in jdata:
d = Domain.query.filter(Domain.name == data['name']).first()
d = dict_db_domain.get(data['name'].rstrip('.'), None)
changed = False
if d:
#update exist domain
d.master = str(data['masters'])
d.type = data['kind']
d.serial = data['serial']
d.notified_serial = data['notified_serial']
d.last_check = 1 if data['last_check'] else 0
d.dnssec = data['dnssec']
# existing domain, only update if something actually has changed
if ( d.master != str(data['masters'])
or d.type != data['kind']
or d.serial != data['serial']
or d.notified_serial != data['notified_serial']
or d.last_check != ( 1 if data['last_check'] else 0 )
or d.dnssec != data['dnssec'] ):
d.master = str(data['masters'])
d.type = data['kind']
d.serial = data['serial']
d.notified_serial = data['notified_serial']
d.last_check = 1 if data['last_check'] else 0
d.dnssec = data['dnssec']
changed = True
else:
# add new domain
d = Domain()
@ -481,10 +492,12 @@ class Domain(db.Model):
d.last_check = data['last_check']
d.dnssec = 1 if data['dnssec'] else 0
db.session.add(d)
try:
db.session.commit()
except:
db.session.rollback()
changed = True
if changed:
try:
db.session.commit()
except:
db.session.rollback()
return {'status': 'ok', 'msg': 'Domain table has been updated successfully'}
except Exception, e:
logging.error('Can not update domain table.' + str(e))