fixed model names and generator

2026-02-06 08:51:35 -03:00
parent 65c3055de6
commit 68622bd6b1
3 changed files with 6 additions and 5 deletions
--- a/modelgen/types.py
+++ b/modelgen/types.py
@@ -22,7 +22,7 @@ DJANGO_TYPES: dict[Any, str] = {
    "list": "models.JSONField(default=list, blank=True)",
    "text": "models.TextField(blank=True, default='')",
    "bigint": "models.BigIntegerField({opts})",
-    "enum": "models.CharField(max_length=20, choices=Status.choices{opts})",
+    "enum": "models.CharField(max_length=20, choices={enum_name}.choices{opts})",
 }

 DJANGO_SPECIAL: dict[str, str] = {