fields.Model se permite cambiar el nombre de la etiqueta
FossilOrigin-Name: 896b797629e426a5e366d5be76fc00c3cc272299d6749e40f8317893b1545a9e
This commit is contained in:
parent
49feee8809
commit
6cc4610b45
@ -1,9 +1,10 @@
|
||||
from .field import Field
|
||||
|
||||
class Model(Field):
|
||||
def __init__(self, model, namespace=None):
|
||||
def __init__(self, model, name=None, namespace=None):
|
||||
self.model = model
|
||||
self.namespace = namespace
|
||||
self.field_name = name
|
||||
|
||||
def __get__(self, inst, cls):
|
||||
if inst is None:
|
||||
@ -20,6 +21,8 @@ class Model(Field):
|
||||
return inst._fields[self.name]
|
||||
except KeyError:
|
||||
obj = self.model()
|
||||
if self.field_name is not None:
|
||||
obj.__name__ = self.field_name
|
||||
self._set_namespace(obj, self.namespace, inst.__namespace__)
|
||||
inst._fields[self.name] = obj
|
||||
return obj
|
||||
|
@ -108,6 +108,20 @@ def test_field_model_failed_initialization():
|
||||
person.id = 33
|
||||
assert "<Person><ID>33</ID></Person>" == person.to_xml()
|
||||
|
||||
def test_field_model_with_custom_name():
|
||||
class ID(facho.model.Model):
|
||||
__name__ = 'ID'
|
||||
|
||||
class Person(facho.model.Model):
|
||||
__name__ = 'Person'
|
||||
|
||||
id = fields.Model(ID, name='DID')
|
||||
|
||||
|
||||
person = Person()
|
||||
person.id = 33
|
||||
assert "<Person><DID>33</DID></Person>" == person.to_xml()
|
||||
|
||||
def test_field_model_default_initialization_with_attributes():
|
||||
class ID(facho.model.Model):
|
||||
__name__ = 'ID'
|
||||
|
Loading…
Reference in New Issue
Block a user