66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
from flask import current_app
|
|
from .base import db
|
|
|
|
|
|
class DomainTemplate(db.Model):
|
|
__tablename__ = "domain_template"
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(255), index=True, unique=True)
|
|
description = db.Column(db.String(255))
|
|
records = db.relationship('DomainTemplateRecord',
|
|
back_populates='template',
|
|
cascade="all, delete-orphan")
|
|
|
|
def __repr__(self):
|
|
return '<DomainTemplate {0}>'.format(self.name)
|
|
|
|
def __init__(self, name=None, description=None):
|
|
self.id = None
|
|
self.name = name
|
|
self.description = description
|
|
|
|
def replace_records(self, records):
|
|
try:
|
|
self.records = []
|
|
for record in records:
|
|
self.records.append(record)
|
|
db.session.commit()
|
|
return {
|
|
'status': 'ok',
|
|
'msg': 'Template records have been modified'
|
|
}
|
|
except Exception as e:
|
|
current_app.logger.error(
|
|
'Cannot create template records Error: {0}'.format(e))
|
|
db.session.rollback()
|
|
return {
|
|
'status': 'error',
|
|
'msg': 'Can not create template records'
|
|
}
|
|
|
|
def create(self):
|
|
try:
|
|
db.session.add(self)
|
|
db.session.commit()
|
|
return {'status': 'ok', 'msg': 'Template has been created'}
|
|
except Exception as e:
|
|
current_app.logger.error(
|
|
'Can not update zone template table. Error: {0}'.format(e))
|
|
db.session.rollback()
|
|
return {
|
|
'status': 'error',
|
|
'msg': 'Can not update zone template table'
|
|
}
|
|
|
|
def delete_template(self):
|
|
try:
|
|
self.records = []
|
|
db.session.delete(self)
|
|
db.session.commit()
|
|
return {'status': 'ok', 'msg': 'Template has been deleted'}
|
|
except Exception as e:
|
|
current_app.logger.error(
|
|
'Can not delete zone template. Error: {0}'.format(e))
|
|
db.session.rollback()
|
|
return {'status': 'error', 'msg': 'Can not delete zone template'}
|