se separa responsabilidad de fields.Function

FossilOrigin-Name: 4d5daa47a75a0e283e86bf992126bf60f3a8a14287e9acc437d5f2f3eca43150
This commit is contained in:
bit4bit
2021-06-24 23:38:28 +00:00
parent f630a544c2
commit 3eacb29afa
5 changed files with 36 additions and 44 deletions

View File

@@ -2,29 +2,24 @@ from .field import Field
from .model import Model
class Function(Field):
def __init__(self, getter=None, setter=None, field=None):
def __init__(self, field, getter=None):
self.field = field
self.getter = getter
self.setter = setter
def __get__(self, inst, cls):
if inst is None:
return self
assert self.name is not None
if self.getter is None and self.field is None:
return None
# si se indica `field` se adiciona
# como campo del modelo, esto es
# que se serializa a xml
inst._fields[self.name] = self.field
if self.getter is None and self.field is not None:
return Model(self.field)
if self.getter is not None:
value = self._call(inst, self.getter, self.name, self.field)
if self.field is None:
return self._call(inst, self.getter, self.name)
else:
obj = Model(self.field)
return self._call(inst, self.getter, self.name, obj)
if value is not None:
self.field.__set__(inst, value)
def __set__(self, inst, value):
if self.setter is None:
return super().__set__(self.name, value)
self._call(inst, self.setter, self.name, value)
return self.field