DNS Providers list returned

lemur/dns_providers/models.py

@@ -6,15 +6,13 @@ from lemur.database import db
 
 
 class DnsProviders(db.Model):
-    db.Table('dns_providers',
-             Column('id', Integer(), nullable=False),
-             Column('name', String(length=256), nullable=True),
-             Column('description', String(length=1024), nullable=True),
-             Column('provider_type', String(length=256), nullable=True),
-             Column('credentials', String(length=256), nullable=True),
-             Column('api_endpoint', String(length=256), nullable=True),
-             Column('date_created', ArrowType(), server_default=text('now()'), nullable=False),
-             Column('status', String(length=128), nullable=True),
-             Column('options', JSON),
-             PrimaryKeyConstraint('id'),
-             UniqueConstraint('name'))
+    __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(String(length=256), 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)

lemur/dns_providers/schemas.py

@@ -0,0 +1,20 @@
+from lemur.common.fields import ArrowDateTime
+from lemur.common.schema import LemurOutputSchema
+
+from marshmallow import fields
+
+
+class DnsProvidersNestedOutputSchema(LemurOutputSchema):
+    __envelope__ = False
+    id = fields.Integer()
+    name = fields.String()
+    description = fields.String()
+    provider_type = fields.String()
+    credentials = fields.String()
+    api_endpoint = fields.String()
+    date_created = ArrowDateTime()
+    status = fields.String()
+    options = fields.String()
+
+
+default_output_schema = DnsProvidersNestedOutputSchema()

lemur/dns_providers/service.py

@@ -7,4 +7,10 @@ def get_all_dns_providers(status="active"):
 
     :return:
     """
-    return DnsProviders.query.all(status=status)
+    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