fields.Function manejan default

FossilOrigin-Name: ec954ac9253429b99095fbcce443a2691f516f603282aec1ee59c4b7cbbd6c4a
This commit is contained in:
bit4bit
2021-06-25 01:56:49 +00:00
parent a9dde83e81
commit b6219bd171
4 changed files with 51 additions and 13 deletions

View File

@@ -1,8 +1,8 @@
from .field import Field
class Attribute(Field):
def __init__(self, tag, default=None):
self.tag = tag
def __init__(self, name, default=None):
self.attribute = name
self.value = default
self.default = default
@@ -16,4 +16,4 @@ class Attribute(Field):
def __set__(self, inst, value):
assert self.name is not None
self.value = value
inst._xml_attributes[self.name] = (self.tag, value)
inst._set_attribute(self.name, self.attribute, value)

View File

@@ -2,9 +2,14 @@ from .field import Field
from .model import Model
class Function(Field):
def __init__(self, field, getter=None):
"""
Permite modificar el modelo cuando se intenta,
obtener el valor de este campo.
"""
def __init__(self, field, getter=None, default=None):
self.field = field
self.getter = getter
self.default = default
def __get__(self, inst, cls):
if inst is None:
@@ -14,8 +19,7 @@ class Function(Field):
# si se indica `field` se adiciona
# como campo del modelo, esto es
# que se serializa a xml
self.field.name = self.name
inst._fields[self.name] = self.field
inst._set_field(self.name, self.field)
if self.getter is not None:
value = self._call(inst, self.getter, self.name, self.field)
@@ -24,3 +28,7 @@ class Function(Field):
self.field.__set__(inst, value)
return self.field
def __set__(self, inst, value):
inst._set_field(self.name, self.field)
self.field.__set__(inst, value)