fields.Model se permite cambiar el nombre de la etiqueta

FossilOrigin-Name: 896b797629e426a5e366d5be76fc00c3cc272299d6749e40f8317893b1545a9e
This commit is contained in:
bit4bit
2021-06-24 01:18:27 +00:00
parent 49feee8809
commit 6cc4610b45
2 changed files with 18 additions and 1 deletions

View File

@@ -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