mariano/mediaproc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed model names and generator

Browse Source
This commit is contained in:
buenosairesam
2026-02-06 08:51:35 -03:00
parent 65c3055de6
commit 68622bd6b1
3 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modelgen/generator/django.py
View File

@@ -217,13 +217,14 @@ class DjangoGenerator(BaseGenerator):
# Enum # Enum
if isinstance(base, type) and issubclass(base, Enum): if isinstance(base, type) and issubclass(base, Enum):
enum_name = base.__name__
extra = [] extra = []
if optional: if optional:
extra.append("null=True, blank=True") extra.append("null=True, blank=True")
if default is not dc.MISSING and isinstance(default, Enum): if default is not dc.MISSING and isinstance(default, Enum):
extra.append(f"default=Status.{default.name}") extra.append(f"default={enum_name}.{default.name}")
return DJANGO_TYPES["enum"].format( return DJANGO_TYPES["enum"].format(
opts=", " + ", ".join(extra) if extra else "" enum_name=enum_name, opts=", " + ", ".join(extra) if extra else ""
) )
# Text fields (based on name heuristics) # Text fields (based on name heuristics)

2
modelgen/types.py
View File

@@ -22,7 +22,7 @@ DJANGO_TYPES: dict[Any, str] = {
"list": "models.JSONField(default=list, blank=True)", "list": "models.JSONField(default=list, blank=True)",
"text": "models.TextField(blank=True, default='')", "text": "models.TextField(blank=True, default='')",
"bigint": "models.BigIntegerField({opts})", "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] = { DJANGO_SPECIAL: dict[str, str] = {

4
mpr/media_assets/models.py
View File

@@ -25,7 +25,7 @@ class MediaAsset(models.Model):
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
filename = models.CharField(max_length=500) filename = models.CharField(max_length=500)
file_path = models.CharField(max_length=1000) file_path = models.CharField(max_length=1000)
status = models.CharField(max_length=20, choices=Status.choices, default=Status.PENDING) status = models.CharField(max_length=20, choices=AssetStatus.choices, default=AssetStatus.PENDING)
error_message = models.TextField(blank=True, default='') error_message = models.TextField(blank=True, default='')
file_size = models.BigIntegerField(null=True, blank=True) file_size = models.BigIntegerField(null=True, blank=True)
duration = models.FloatField(null=True, blank=True, default=None) duration = models.FloatField(null=True, blank=True, default=None)
@@ -89,7 +89,7 @@ class TranscodeJob(models.Model):
output_filename = models.CharField(max_length=500) output_filename = models.CharField(max_length=500)
output_path = models.CharField(max_length=1000, null=True, blank=True) output_path = models.CharField(max_length=1000, null=True, blank=True)
output_asset_id = models.UUIDField(null=True, blank=True) output_asset_id = models.UUIDField(null=True, blank=True)
status = models.CharField(max_length=20, choices=Status.choices, default=Status.PENDING) status = models.CharField(max_length=20, choices=JobStatus.choices, default=JobStatus.PENDING)
progress = models.FloatField(default=0.0) progress = models.FloatField(default=0.0)
current_frame = models.IntegerField(null=True, blank=True, default=None) current_frame = models.IntegerField(null=True, blank=True, default=None)
current_time = models.FloatField(null=True, blank=True, default=None) current_time = models.FloatField(null=True, blank=True, default=None)