162 lines
4.1 KiB
Python
162 lines
4.1 KiB
Python
from django.utils.html import format_html
|
|
|
|
|
|
class OrchestraModel:
|
|
""" Base class from which all orchestra models will inherit. """
|
|
api_name = None
|
|
verbose_name = None
|
|
fields = ()
|
|
param_defaults = {}
|
|
|
|
def __init__(self, **kwargs):
|
|
if self.verbose_name is None:
|
|
self.verbose_name = self.api_name
|
|
|
|
for (param, default) in self.param_defaults.items():
|
|
setattr(self, param, kwargs.get(param, default))
|
|
|
|
# def get(self, key):
|
|
# # retrieve attr of the object and if undefined get raw data
|
|
# return getattr(self, key, self.data.get(key))
|
|
|
|
@classmethod
|
|
def new_from_json(cls, data, **kwargs):
|
|
""" Create a new instance based on a JSON dict. Any kwargs should be
|
|
supplied by the inherited, calling class.
|
|
Args:
|
|
data: A JSON dict, as converted from the JSON in the orchestra API.
|
|
"""
|
|
|
|
json_data = data.copy()
|
|
if kwargs:
|
|
for key, val in kwargs.items():
|
|
json_data[key] = val
|
|
|
|
c = cls(**json_data)
|
|
c._json = data
|
|
# TODO(@slamora) remove/replace by private variable to ovoid name collisions
|
|
c.data = data
|
|
return c
|
|
|
|
|
|
class BillingContact(OrchestraModel):
|
|
param_defaults = {
|
|
'name': None,
|
|
'address': None,
|
|
'city': None,
|
|
'zipcode': None,
|
|
'country': None,
|
|
'vat': None,
|
|
}
|
|
|
|
|
|
class PaymentSource(OrchestraModel):
|
|
api_name = 'payment-source'
|
|
param_defaults = {
|
|
"method": None,
|
|
"data": [],
|
|
"is_active": False,
|
|
}
|
|
|
|
|
|
class UserAccount(OrchestraModel):
|
|
api_name = 'accounts'
|
|
param_defaults = {
|
|
'username': None,
|
|
'type': None,
|
|
'language': None,
|
|
'short_name': None,
|
|
'full_name': None,
|
|
'billing': {},
|
|
}
|
|
|
|
@classmethod
|
|
def new_from_json(cls, data, **kwargs):
|
|
billing = None
|
|
|
|
if 'billcontact' in data:
|
|
billing = BillingContact.new_from_json(data['billcontact'])
|
|
return super().new_from_json(data=data, billing=billing)
|
|
|
|
|
|
class DatabaseUser(OrchestraModel):
|
|
api_name = 'databaseusers'
|
|
fields = ('username',)
|
|
param_defaults = {
|
|
'username': None,
|
|
}
|
|
|
|
|
|
class DatabaseService(OrchestraModel):
|
|
api_name = 'database'
|
|
fields = ('name', 'type', 'users')
|
|
param_defaults = {
|
|
"id": None,
|
|
"name": None,
|
|
"type": None,
|
|
"users": None,
|
|
}
|
|
|
|
@classmethod
|
|
def new_from_json(cls, data, **kwargs):
|
|
users = None
|
|
if 'users' in data:
|
|
users = [DatabaseUser.new_from_json(user_data) for user_data in data['users']]
|
|
return super().new_from_json(data=data, users=users)
|
|
|
|
|
|
class MailService(OrchestraModel):
|
|
api_name = 'address'
|
|
verbose_name = 'Mail'
|
|
fields = ('mail_address', 'aliases', 'type', 'type_detail')
|
|
|
|
FORWARD = 'forward'
|
|
MAILBOX = 'mailbox'
|
|
|
|
@property
|
|
def aliases(self):
|
|
return [
|
|
name + '@' + self.data['domain']['name'] for name in self.data['names'][1:]
|
|
]
|
|
|
|
@property
|
|
def mail_address(self):
|
|
return self.data['names'][0] + '@' + self.data['domain']['name']
|
|
|
|
@property
|
|
def type(self):
|
|
if self.data['forward']:
|
|
return self.FORWARD
|
|
return self.MAILBOX
|
|
|
|
@property
|
|
def type_detail(self):
|
|
if self.type == self.FORWARD:
|
|
return self.data['forward']
|
|
# TODO(@slamora) retrieve mailbox usage
|
|
return {'usage': 0, 'total': 213}
|
|
|
|
|
|
class MailinglistService(OrchestraModel):
|
|
api_name = 'mailinglist'
|
|
verbose_name = 'Mailing list'
|
|
fields = ('name', 'status', 'address_name', 'admin_email', 'configure')
|
|
param_defaults = {
|
|
'name': None,
|
|
'admin_email': None,
|
|
}
|
|
|
|
@property
|
|
def status(self):
|
|
# TODO(@slamora): where retrieve if the list is active?
|
|
return 'active'
|
|
|
|
@property
|
|
def address_name(self):
|
|
return "{}@{}".format(self.data['address_name'], self.data['address_domain']['name'])
|
|
|
|
@property
|
|
def configure(self):
|
|
# TODO(@slamora): build mailtran absolute URL
|
|
return format_html('<a href="#TODO">Mailtrain</a>')
|