31 lines
904 B
Python
31 lines
904 B
Python
from .field import Field
|
|
from .model import Model
|
|
|
|
class Function(Field):
|
|
def __init__(self, getter=None, setter=None, field=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
|
|
|
|
if self.getter is None and self.field is not None:
|
|
return Model(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)
|
|
|
|
def __set__(self, inst, value):
|
|
if self.setter is None:
|
|
return super().__set__(self.name, value)
|
|
self._call(inst, self.setter, self.name, value)
|