add more account attributes

Author: dmitry-sinina

yeti_switch_api/orm/__init__.py

@@ -3,6 +3,7 @@
 from .contact import Contact  # noqa: F401
 from .account import Account  # noqa: F401
 from .invoice import Invoice  # noqa: F401
+from .invoice_template import InvoiceTemplate  # noqa: F401
 from .invoice_originated_destination import InvoiceOriginatedDestination  # noqa: F401
 from .invoice_originated_network import InvoiceOriginatedNetwork  # noqa: F401
 from .invoice_terminated_destination import InvoiceTerminatedDestination  # noqa: F401

yeti_switch_api/orm/account.py

@@ -9,10 +9,58 @@ class Meta:
     contractor = RelationField("contractor")
     timezone = RelationField("timezone")
 
+    external-id = AttributeField("external-id")
+    uuid = AttributeField("uuid")
+
     name = AttributeField("name")
     balance = AttributeField("balance")
     min_balance = AttributeField("min-balance")
     max_balance = AttributeField("max-balance")
+    vat = AttributeField("vat")
+    balance_low_threshold = AttributeField("balance-low-threshold")
+    balance_high_threshold = AttributeField("balance-high-threshold")
+    send_balance_notifications_to = AttributeField("send-balance-notifications-to")
+
+    destination_rate_limit = AttributeField("destination-rate-limit")
+    max_call_duration = AttributeField("max-call-duration")
+    origination_capacity = AttributeField("origination-capacity")
+    termination_capacity = AttributeField("termination-capacity")
+    total_capacity = AttributeField("total-capacity")
+
+    invoice_template = RelationField("invoice-template")
+    invoice_period = AttributeField("invoice-period-id")
+    send_invoices_to = AttributeField("send-invoices-to")
+
+    INVOICE_PERIOD_DAILY = 1
+    INVOICE_PERIOD_WEEKLY = 2
+    INVOICE_PERIOD_BIWEEKLY = 3
+    INVOICE_PERIOD_MONTHLY = 4
+    INVOICE_PERIOD_BIWEEKLY_SPLIT = 5
+    INVOICE_PERIOD_WEEKLY_SPLIT = 6
 
     def creatable_fields(self):
-        return ["name", "contractor", "timezone", "min-balance", "max-balance"]
+        return [
+           "external-id",
+           "uuid",
+           "name",
+           "contractor",
+           "timezone",
+           "invoice-template",
+           "min-balance",
+           "max-balance",
+           "vat",
+           "balance-low-threshold",
+           "balance-high-threshold",
+           "send-balance-notifications-to"
+           "destination-rate-limit",
+           "max-call-duration",
+           "origination-capacity",
+           "termination-capacity",
+           "total-capacity",
+           "invoice-period-id",
+           "send-invoices-to"
+        ]
+
+    def updatable_fields(self):
+        return self.creatable_fields()
+

yeti_switch_api/orm/invoice_template.py

@@ -0,0 +1,10 @@
+from .base_model import BaseModel, AttributeField, RelationField  # noqa: F401
+
+
+class InvoiceTemplate(BaseModel):
+    class Meta:
+        path = "invoice-template"
+        type = "invoice-template"
+
+    name = AttributeField("name")
+    filename = AttributeField("filename")

yeti_switch_api/orm/orm_client.py

@@ -6,6 +6,7 @@
 from .contact import Contact
 from .account import Account
 from .invoice import Invoice
+from .invoice_template import InvoiceTemplate
 from .invoice_originated_destination import InvoiceOriginatedDestination
 from .invoice_originated_network import InvoiceOriginatedNetwork
 from .invoice_terminated_destination import InvoiceTerminatedDestination
@@ -49,6 +50,7 @@ def __register_models(cls):
         cls.__register_model(Contact)
         cls.__register_model(Account)
         cls.__register_model(Invoice)
+        cls.__register_model(InvoiceTemplate)
         cls.__register_model(InvoiceOriginatedDestination)
         cls.__register_model(InvoiceOriginatedNetwork)
         cls.__register_model(InvoiceTerminatedDestination)

yeti_switch_api/orm/package_counter.py

@@ -8,7 +8,8 @@ class Meta:
     duration = AttributeField("duration")
     exclude = AttributeField("exclude")
     prefix = AttributeField("prefix")
-    service_id = AttributeField("service-id")
 
     account = RelationField("account")
     service = RelationField("service")
+
+