Addressing comments. Updating copyrights. Added function to determine authorative name server

This commit is contained in:
Curtis Castrapel
2018-05-29 10:18:16 -07:00
parent ae26e44cc2
commit 544a02ca3f
123 changed files with 251 additions and 150 deletions

View File

@ -1,7 +1,7 @@
"""
.. module: lemur.authorizations.models
:platform: unix
:copyright: (c) 2015 by Netflix Inc., see AUTHORS for more
:copyright: (c) 2018 by Netflix Inc., see AUTHORS for more
:license: Apache, see LICENSE for more details.
.. moduleauthor:: Netflix Secops <secops@netflix.com>
"""
@ -12,7 +12,7 @@ from lemur.database import db
from lemur.plugins.base import plugins
class Authorizations(db.Model):
class Authorization(db.Model):
__tablename__ = 'pending_dns_authorizations'
id = Column(Integer, primary_key=True, autoincrement=True)
account_number = Column(String(128))
@ -25,7 +25,7 @@ class Authorizations(db.Model):
return plugins.get(self.plugin_name)
def __repr__(self):
return "Authorizations(id={id})".format(label=self.id)
return "Authorization(id={id})".format(label=self.id)
def __init__(self, account_number, domains, dns_provider_type, options=None):
self.account_number = account_number

View File

@ -1,18 +1,18 @@
"""
.. module: lemur.pending_certificates.service
Copyright (c) 2017 and onwards Instart Logic, Inc. All rights reserved.
Copyright (c) 2018 and onwards Netflix, Inc. All rights reserved.
.. moduleauthor:: Secops <secops@netflix.com>
"""
from lemur import database
from lemur.authorizations.models import Authorizations
from lemur.authorizations.models import Authorization
def get(authorization_id):
"""
Retrieve dns authorization by ID
"""
return database.get(Authorizations, authorization_id)
return database.get(Authorization, authorization_id)
def create(account_number, domains, dns_provider_type, options=None):
@ -20,5 +20,5 @@ def create(account_number, domains, dns_provider_type, options=None):
Creates a new dns authorization.
"""
authorization = Authorizations(account_number, domains, dns_provider_type, options)
authorization = Authorization(account_number, domains, dns_provider_type, options)
return database.create(authorization)