Splitting models on many files

Author: ht-albert

connect/models/__init__.py

@@ -1,5 +1,8 @@
 from .activation_response import ActivationTemplateResponse, ActivationTileResponse
-from .schema import BaseScheme, FulfillmentScheme, Param, ParamsScheme, ServerErrorScheme
+from .base import BaseScheme
+from .fulfillment import FulfillmentScheme
+from .parameters import Param, ParamsScheme
+from .server_error import ServerErrorScheme
 
 __all__ = [
     'ActivationTemplateResponse',

connect/models/asset.py

@@ -0,0 +1,28 @@
+from marshmallow import fields, post_load
+
+from .base import BaseObject, BaseScheme
+from .connection import ConnectionScheme
+from .parameters import ParamsScheme
+from .product import ItemScheme, ProductScheme
+from .tiers import TiersSchemeMixin
+
+
+class Asset(BaseObject):
+    pass
+
+
+class AssetScheme(BaseScheme):
+    status = fields.Str()
+    external_id = fields.Str()
+    external_uid = fields.UUID()
+    product = fields.Nested(ProductScheme, only=('id', 'name'))
+    connection = fields.Nested(
+        ConnectionScheme, only=('id', 'type', 'provider', 'vendor'),
+    )
+    items = fields.List(fields.Nested(ItemScheme))
+    params = fields.List(fields.Nested(ParamsScheme))
+    tiers = fields.Nested(TiersSchemeMixin)
+
+    @post_load
+    def make_object(self, data):
+        return Asset(**data)

connect/models/base.py

@@ -0,0 +1,17 @@
+from marshmallow import Schema, fields, post_load
+
+
+class BaseObject:
+    def __init__(self, *args, **kwargs):
+        self.id = kwargs.get('id')
+        if kwargs:
+            for attr, val in kwargs.items():
+                setattr(self, attr, val)
+
+
+class BaseScheme(Schema):
+    id = fields.Str()
+
+    @post_load
+    def make_object(self, data):
+        return BaseObject(**data)

connect/models/company.py

@@ -0,0 +1,15 @@
+from marshmallow import fields, post_load
+
+from .base import BaseObject, BaseScheme
+
+
+class Company(BaseObject):
+    pass
+
+
+class CompanyScheme(BaseScheme):
+    name = fields.Str()
+
+    @post_load
+    def make_object(self, data):
+        return Company(**data)

connect/models/connection.py

@@ -0,0 +1,22 @@
+from marshmallow import fields, post_load
+
+from .base import BaseObject, BaseScheme
+from .company import CompanyScheme
+from .hub import HubScheme
+from .product import ProductScheme
+
+
+class Connection(BaseObject):
+    pass
+
+
+class ConnectionScheme(BaseScheme):
+    type = fields.Str()
+    provider = fields.Nested(CompanyScheme, only=('id', 'name'))
+    vendor = fields.Nested(CompanyScheme, only=('id', 'name'))
+    product = fields.Nested(ProductScheme)
+    hub = fields.Nested(HubScheme)
+
+    @post_load
+    def make_object(self, data):
+        return Connection(**data)

connect/models/exception.py

@@ -1,4 +1,4 @@
-from .schema import ServerError
+from .server_error import ServerError
 
 
 class Message(Exception):

connect/models/fulfillment.py

@@ -0,0 +1,26 @@
+from marshmallow import fields, post_load
+
+from .asset import AssetScheme
+from .base import BaseObject, BaseScheme
+from .marketplace import ContractScheme, MarketplaceScheme
+
+
+class Fulfillment(BaseObject):
+    pass
+
+
+class FulfillmentScheme(BaseScheme):
+    activation_key = fields.Str()
+    asset = fields.Nested(AssetScheme)
+    status = fields.Str()
+    type = fields.Str()
+    updated = fields.DateTime()
+    created = fields.DateTime()
+    reason = fields.Str()
+    params_form_url = fields.Str()
+    contract = fields.Nested(ContractScheme, only=('id', 'name'))
+    marketplace = fields.Nested(MarketplaceScheme, only=('id', 'name'))
+
+    @post_load
+    def make_object(self, data):
+        return Fulfillment(**data)

connect/models/hub.py

@@ -0,0 +1,20 @@
+from marshmallow import Schema, fields, post_load
+
+from .base import BaseObject, BaseScheme
+
+
+class Hub(BaseObject):
+    pass
+
+
+class HubScheme(BaseScheme):
+    name = fields.Str()
+
+    @post_load
+    def make_object(self, data):
+        return Hub(**data)
+
+
+class HubsSchemeMixin(Schema):
+    hub = fields.Nested(HubScheme, only=('id', 'name'))
+    external_id = fields.Str()

connect/models/marketplace.py

@@ -0,0 +1,71 @@
+from marshmallow import fields, post_load
+
+from .base import BaseObject, BaseScheme
+from .company import CompanyScheme
+from .hub import HubsSchemeMixin
+
+
+class Marketplace(BaseObject):
+    pass
+
+
+class MarketplaceScheme(BaseScheme):
+    name = fields.Str()
+    zone = fields.Str()
+    description = fields.Str()
+    active_contract = fields.Int()
+    icon = fields.Str()
+    owner = fields.Nested(CompanyScheme, only=('id', 'name'))
+    hubs = fields.List(fields.Nested(HubsSchemeMixin, only=('id', 'name')))
+
+    @post_load
+    def make_object(self, data):
+        return Marketplace(**data)
+
+
+class Agreement(BaseObject):
+    pass
+
+
+class AgreementScheme(BaseScheme):
+    type = fields.Str()
+    title = fields.Str()
+    description = fields.Str()
+    created = fields.DateTime()
+    updated = fields.DateTime()
+    owner = fields.Nested(CompanyScheme, only=('id', 'name'))
+    stats = fields.Dict()
+    active = fields.Bool()
+    version = fields.Int()
+    link = fields.Str()
+    version_created = fields.DateTime()
+    version_contracts = fields.Int()
+
+    @post_load
+    def make_object(self, data):
+        return Agreement(**data)
+
+
+class Contract(BaseObject):
+    pass
+
+
+class ContractScheme(BaseScheme):
+    name = fields.Str()
+    status = fields.Str()
+    version = fields.Int()
+    type = fields.Str()
+    agreement = fields.Nested(AgreementScheme, only=('id', 'name'))
+    marketplace = fields.Nested(MarketplaceScheme, only=('id', 'name'))
+    owner = fields.Nested(CompanyScheme, only=('id', 'name'))
+    creater = fields.Nested(CompanyScheme, only=('id', 'name'))
+    created = fields.DateTime()
+    updated = fields.DateTime()
+    enrolled = fields.Str()
+    version_created = fields.DateTime()
+    activation = fields.Dict()
+    signee = fields.Nested(CompanyScheme, only=('id', 'name'))
+
+    @post_load
+    def make_object(self, data):
+        return Contract(**data)

connect/models/parameters.py

@@ -0,0 +1,32 @@
+from marshmallow import Schema, fields, post_load
+
+from .base import BaseObject, BaseScheme
+
+
+class ValueChoice(BaseObject):
+    pass
+
+
+class ValueChoiceScheme(Schema):
+    value = fields.Str()
+    label = fields.Str()
+
+    @post_load
+    def make_object(self, data):
+        return ValueChoice(**data)
+
+
+class Param(BaseObject):
+    pass
+
+
+class ParamsScheme(BaseScheme):
+    name = fields.Str()
+    type = fields.Str()
+    value = fields.Str()
+    value_choices = fields.List(fields.Nested(ValueChoiceScheme))
+    value_error = fields.Str()
+
+    @post_load
+    def make_object(self, data):
+        return Param(**data)