38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from sqlalchemy import Column, Integer, String, text
|
|
from sqlalchemy.dialects.postgresql import JSON
|
|
from sqlalchemy_utils import ArrowType
|
|
|
|
from lemur.database import db
|
|
from lemur.plugins.base import plugins
|
|
from lemur.utils import Vault
|
|
|
|
|
|
class DnsProvider(db.Model):
|
|
__tablename__ = 'dns_providers'
|
|
id = Column(
|
|
Integer(),
|
|
primary_key=True,
|
|
)
|
|
name = Column(String(length=256), unique=True, nullable=True)
|
|
description = Column(String(length=1024), nullable=True)
|
|
provider_type = Column(String(length=256), nullable=True)
|
|
credentials = Column(Vault, nullable=True)
|
|
api_endpoint = Column(String(length=256), nullable=True)
|
|
date_created = Column(ArrowType(), server_default=text('now()'), nullable=False)
|
|
status = Column(String(length=128), nullable=True)
|
|
options = Column(JSON, nullable=True)
|
|
domains = Column(JSON, nullable=True)
|
|
|
|
def __init__(self, name, description, provider_type, credentials):
|
|
self.name = name
|
|
self.description = description
|
|
self.provider_type = provider_type
|
|
self.credentials = credentials
|
|
|
|
@property
|
|
def plugin(self):
|
|
return plugins.get(self.plugin_name)
|
|
|
|
def __repr__(self):
|
|
return "DnsProvider(name={name})".format(name=self.name)
|