Add get_related_field utility function

Add the get_related_field function to resolve a
model's related field reference by a path separate
by Django's default field separator '__'.

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-01-19 00:13:19 -04:00
parent 2d9aca55c5
commit 9ed93b54af

View File

@@ -87,6 +87,26 @@ def fs_cleanup(filename, file_descriptor=None, suppress_exceptions=True):
raise
def get_related_field(model, related_field_name):
try:
local_field_name, remaining_field_path = related_field_name.split(
LOOKUP_SEP, 1
)
except ValueError:
local_field_name = related_field_name
remaining_field_path = None
related_field = model._meta.get_field(local_field_name)
if remaining_field_path:
return get_related_field(
model=related_field.related_model,
related_field_name=remaining_field_path
)
return related_field
def get_descriptor(file_input, read=True):
try:
# Is it a file like object?
@@ -119,7 +139,6 @@ def get_storage_subclass(dotted_path):
def deconstruct(self):
return ('mayan.apps.common.classes.FakeStorageSubclass', (), {})
return StorageSubclass