se instancia modelo en caso de no existir para Many2One

FossilOrigin-Name: 006f6a780ae0436649addd2abe89eb6a9bfc5ad573ee1a1835a8f65ab039fd26
This commit is contained in:
bit4bit 2021-06-24 01:51:05 +00:00
parent 58e7387292
commit 92bae58e51
4 changed files with 35 additions and 21 deletions

View File

@ -25,3 +25,17 @@ class Field:
if callable(call):
return call(*args)
def _create_model(self, inst, name=None, model=None):
try:
return inst._fields[self.name]
except KeyError:
if model is not None:
obj = model()
else:
obj = self.model()
if name is not None:
obj.__name__ = name
self._set_namespace(obj, self.namespace, inst.__namespace__)
inst._fields[self.name] = obj
return obj

View File

@ -13,15 +13,11 @@ class Many2One(Field):
if inst is None:
return self
assert self.name is not None
return inst._fields[self.name]
return self._create_model(inst)
def __set__(self, inst, value):
assert self.name is not None
class_model = self.model
inst_model = class_model()
self._set_namespace(inst_model, self.namespace, inst.__namespace__)
inst._fields[self.name] = inst_model
inst_model = self._create_model(inst, model=self.model)
# si hay setter manual se ejecuta
# de lo contrario se asigna como texto del elemento

View File

@ -10,20 +10,8 @@ class Model(Field):
if inst is None:
return self
assert self.name is not None
return self._create_model(inst)
return self._create_model(inst, name=self.field_name)
def __set__(self, inst, value):
obj = self._create_model(inst)
obj._set_content(value)
def _create_model(self, inst):
try:
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
obj = self._create_model(inst, name=self.field_name)
obj._set_content(value)

View File

@ -80,6 +80,22 @@ def test_many2one_with_custom_setter():
party.location = 99
assert '<Party><PhysicalLocation ID="99"/></Party>' == party.to_xml()
def test_many2one_auto_create():
class TaxAmount(facho.model.Model):
__name__ = 'TaxAmount'
currencyID = fields.Attribute('currencyID')
class TaxTotal(facho.model.Model):
__name__ = 'TaxTotal'
amount = fields.Many2One(TaxAmount)
tax_total = TaxTotal()
tax_total.amount.currencyID = 'COP'
tax_total.amount = 3333
assert '<TaxTotal><TaxAmount currencyID="COP">3333</TaxAmount></TaxTotal>' == tax_total.to_xml()
def test_field_model():
class ID(facho.model.Model):
__name__ = 'ID'