rewrite get_name_of_model function

mrtedn21 · Nov 14, 2023 · f12c9ab · f12c9ab
1 parent 65457d1
commit f12c9ab
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/martin_eden/utils.py b/martin_eden/utils.py
@@ -9,11 +9,11 @@
 
 
 def get_name_of_model(model_class: 'BaseModel') -> str:
-    model_name = model_class.__name__
-    # remove "Orm" postfix from model name
-    model_name = model_name[:-3]
-    model_name = model_name.lower()
-    return model_name
+    return (
+        getattr(model_class, '__entity_name__', '') or
+        getattr(model_class, '__tablename__', '') or
+        getattr(model_class, '__name__', '')
+    )
 
 
 def get_argument_names(foo: Callable) -> tuple[str]: