More stuff. Will prioritize this more next week

This commit is contained in:
Curtis Castrapel
2018-04-20 14:49:54 -07:00
parent fbce1ef7c7
commit 44e3b33aaa
22 changed files with 496 additions and 80 deletions

View File

@ -15,4 +15,5 @@ class DnsProviders(db.Model):
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)
options = Column(JSON, nullable=True)
domains = Column(JSON, nullable=True)

View File

@ -8,13 +8,11 @@ class DnsProvidersNestedOutputSchema(LemurOutputSchema):
__envelope__ = False
id = fields.Integer()
name = fields.String()
description = fields.String()
provider_type = fields.String()
description = fields.String()
credentials = fields.String()
api_endpoint = fields.String()
date_created = ArrowDateTime()
status = fields.String()
options = fields.String()
default_output_schema = DnsProvidersNestedOutputSchema()
dns_provider_schema = DnsProvidersNestedOutputSchema()

View File

@ -1,16 +1,33 @@
from lemur import database
from lemur.dns_providers.models import DnsProviders
def get_all_dns_providers(status="active"):
def render(args):
"""
Retrieves all certificates within Lemur.
Helper that helps us render the REST Api responses.
:param args:
:return:
"""
all_dns_providers = DnsProviders.query.all()
dns_provider_result = []
for provider in all_dns_providers:
print(provider)
if provider.status == status:
dns_provider_result.append(provider.__dict__)
return dns_provider_result
query = database.session_query(DnsProviders)
return database.sort_and_page(query, DnsProviders, args)
def get(dns_provider_id):
"""
Retrieves a dns provider by its lemur assigned ID.
:param dns_provider_id: Lemur assigned ID
:rtype : DnsProvider
:return:
"""
return database.get(DnsProviders, dns_provider_id)
def delete(dns_provider_id):
"""
Deletes a DNS provider.
:param dns_provider_id: Lemur assigned ID
"""
database.delete(get(dns_provider_id))

View File

@ -5,13 +5,15 @@
:license: Apache, see LICENSE for more details.
.. moduleauthor:: Curtis Castrapel <ccastrapel@netflix.com>
"""
from flask import Blueprint
from flask import Blueprint, g
from flask_restful import reqparse, Api
from lemur.auth.permissions import admin_permission
from lemur.auth.service import AuthenticatedResource
from lemur.common.schema import validate_schema
from lemur.common.utils import paginated_parser
from lemur.dns_providers import service
from lemur.dns_providers.schemas import dns_provider_schema
mod = Blueprint('dns_providers', __name__)
api = Api(mod)
@ -23,6 +25,7 @@ class DnsProvidersList(AuthenticatedResource):
self.reqparse = reqparse.RequestParser()
super(DnsProvidersList, self).__init__()
@validate_schema(None, dns_provider_schema)
def get(self):
"""
.. http:get:: /dns_providers
@ -66,7 +69,19 @@ class DnsProvidersList(AuthenticatedResource):
:statuscode 403: unauthenticated
"""
return service.get_all_dns_providers()
parser = paginated_parser.copy()
parser.add_argument('id', type=int, location='args')
parser.add_argument('name', type=str, location='args')
parser.add_argument('type', type=str, location='args')
args = parser.parse_args()
args['user'] = g.user
return service.render(args)
@admin_permission.require(http_exception=403)
def delete(self, dns_provider_id):
service.delete(dns_provider_id)
return {'result': True}
api.add_resource(DnsProvidersList, '/dns_providers', endpoint='dns_providers')