From a94ab3662cc04eb8f61d2a2f2439801bbbab289c Mon Sep 17 00:00:00 2001 From: Evgeny Date: Wed, 17 Jan 2024 15:08:52 -0500 Subject: [PATCH 1/2] Added workaround for aliased functions --- django_dbml/management/commands/dbml.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/django_dbml/management/commands/dbml.py b/django_dbml/management/commands/dbml.py index 93710ee..5333738 100644 --- a/django_dbml/management/commands/dbml.py +++ b/django_dbml/management/commands/dbml.py @@ -38,7 +38,10 @@ def get_field_notes(self, field): if name == "default": if callable(value): - value = "{}.{}()".format(inspect.getmodule(value).__name__, value.__name__) + if inspect.getmodule(value).__name__: + value = "{}.{}()".format(inspect.getmodule(value).__name__, value.__name__) + else: + value = "{}()".format(value.__name__) elif isinstance(value, str): value = "\"{}\"".format(value) attributes.append('default:`{}`'.format(value)) From 7898f58b0db2303280dacd5faaff85f0cf4d1536 Mon Sep 17 00:00:00 2001 From: Evgeny Date: Wed, 17 Jan 2024 15:09:45 -0500 Subject: [PATCH 2/2] needed to drop __name__ --- django_dbml/management/commands/dbml.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/django_dbml/management/commands/dbml.py b/django_dbml/management/commands/dbml.py index 5333738..31fa2c4 100644 --- a/django_dbml/management/commands/dbml.py +++ b/django_dbml/management/commands/dbml.py @@ -38,7 +38,7 @@ def get_field_notes(self, field): if name == "default": if callable(value): - if inspect.getmodule(value).__name__: + if inspect.getmodule(value): value = "{}.{}()".format(inspect.getmodule(value).__name__, value.__name__) else: value = "{}()".format(value.__name__)