import inspect

import pecan

from pecan import request

from pecan.rest import RestController

from wsme.types import Base, UserType, Unset

from billingstack.openstack.common import log

LOG = log.getLogger(__name__)

class Property(UserType):

"""

A Property that just passes the value around...

"""

def tonativetype(self, value):

return value

def fromnativetype(self, value):

return value

property_type = Property()

class RestBase(RestController):

__resource__ = None

__id__ = None

def __init__(self, parent=None, id_=None):

self.parent = parent

if self.__id__:

request.context[self.__id__ + '_id'] = id_

self.id_ = id_

@pecan.expose()

def _lookup(self, *url_data):

"""

A fun approach to _lookup - checks self.__resource__ for the "id"

"""

id_ = None

if len(url_data) >= 1:

id_ = url_data[0]

parts = url_data[1:] if len(url_data) > 1 else ()

LOG.debug("Lookup: id '%s' parts '%s'", id_, parts)

resource = self.__resource__

if inspect.isclass(resource) and issubclass(resource, RestBase):

return resource(parent=self, id_=id_), parts

def __getattr__(self, name):

"""

Overload this to look in self.__resource__ if name is defined as a

Controller

"""

if name in self.__dict__:

return self.__dict__[name]

elif isinstance(self.__resource__, dict) and name in self.__resource__:

return self.__resource__[name](parent=self)

else:

raise AttributeError

class ModelBase(Base):

def as_dict(self):

"""

Return this model as a dict

"""

data = {}

for attr in self._wsme_attributes:

value = attr.__get__(self, self.__class__)

if value is not Unset:

if isinstance(value, Base) and hasattr(value, "as_dict"):

value = value.as_dict()

data[attr.name] = value

return data

def to_db(self):

"""

Returns this Model object as it's DB form

Example

'currency' vs 'currency_name'

"""

return self.as_dict()

@classmethod

def from_db(cls, values):

"""

Return a class of this object from values in the from_db

"""

return cls(**values)