90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
from .fields import Field
|
|
|
|
class ModelMeta(type):
|
|
def __new__(cls, name, bases, ns):
|
|
new = type.__new__(cls, name, bases, ns)
|
|
if '__name__' in ns:
|
|
new.__name__ = ns['__name__']
|
|
if '__namespace__' in ns:
|
|
new.__namespace__ = ns['__namespace__']
|
|
else:
|
|
new.__namespace__ = {}
|
|
|
|
return new
|
|
|
|
class ModelBase(object, metaclass=ModelMeta):
|
|
|
|
def __new__(cls, *args, **kwargs):
|
|
obj = super().__new__(cls, *args, **kwargs)
|
|
obj._xml_attributes = {}
|
|
obj._fields = {}
|
|
obj._text = ""
|
|
obj._namespace_prefix = None
|
|
|
|
# forzamos registros de campos al modelo
|
|
# al instanciar
|
|
for (key, v) in type(obj).__dict__.items():
|
|
if isinstance(v, fields.Attribute) or isinstance(v, fields.Many2One):
|
|
if hasattr(v, 'default') and v.default is not None:
|
|
setattr(obj, key, v.default)
|
|
|
|
|
|
return obj
|
|
|
|
def __setitem__(self, key, val):
|
|
self._xml_attributes[key] = val
|
|
|
|
def __getitem__(self, key):
|
|
return self._xml_attributes[key]
|
|
|
|
def __before_xml__(self):
|
|
pass
|
|
|
|
def __default_set__(self, value):
|
|
return value
|
|
|
|
def _set_content(self, value):
|
|
default = self.__default_set__(value)
|
|
if default is not None:
|
|
self._text = str(default)
|
|
|
|
def _hook_before_xml(self):
|
|
self.__before_xml__()
|
|
for field in self._fields.values():
|
|
if hasattr(field, '__before_xml__'):
|
|
field.__before_xml__()
|
|
|
|
def to_xml(self):
|
|
self._hook_before_xml()
|
|
|
|
tag = self.__name__
|
|
ns = ''
|
|
if self._namespace_prefix is not None:
|
|
ns = "%s:" % (self._namespace_prefix)
|
|
|
|
pair_attributes = ["%s=\"%s\"" % (k, v) for (k, v) in self._xml_attributes.values()]
|
|
|
|
for (prefix, url) in self.__namespace__.items():
|
|
pair_attributes.append("xmlns:%s=\"%s\"" % (prefix, url))
|
|
attributes = ""
|
|
if pair_attributes:
|
|
attributes = " " + " ".join(pair_attributes)
|
|
|
|
content = ""
|
|
|
|
for name, value in self._fields.items():
|
|
if hasattr(value, 'to_xml'):
|
|
content += value.to_xml()
|
|
elif isinstance(value, str):
|
|
content += value
|
|
content += self._text
|
|
|
|
if content == "":
|
|
return "<%s%s%s/>" % (ns, tag, attributes)
|
|
else:
|
|
return "<%s%s%s>%s</%s%s>" % (ns, tag, attributes, content, ns, tag)
|
|
|
|
class Model(ModelBase):
|
|
pass
|
|
|