Enums Choices en los talks

temii/talks/admin.py

@@ -1,21 +1,9 @@
 from django.contrib import admin
 
-from .models import Talk, TalkLanguage, TalkLevel
+from .models import Talk
 
 
 @admin.register(Talk)
 class TalkAdmin(admin.ModelAdmin):
     list_display = ["name"]
     search_fields = ["name"]
-
-
-@admin.register(TalkLanguage)
-class TalkLanguageAdmin(admin.ModelAdmin):
-    list_display = ["name"]
-    search_fields = ["name"]
-
-
-@admin.register(TalkLevel)
-class TalkLevelAdmin(admin.ModelAdmin):
-    list_display = ["name"]
-    search_fields = ["name"]

temii/talks/migrations/0001_initial.py

@@ -1,4 +1,4 @@
-# Generated by Django 4.1.8 on 2023-08-13 00:25
+# Generated by Django 4.1.8 on 2023-09-06 02:05
 
 from django.conf import settings
 from django.db import migrations, models
@@ -13,30 +13,24 @@ class Migration(migrations.Migration):
     ]
 
     operations = [
-        migrations.CreateModel(
-            name="TalkLanguage",
-            fields=[
-                ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
-                ("name", models.CharField(max_length=60)),
-            ],
-        ),
-        migrations.CreateModel(
-            name="TalkLevel",
-            fields=[
-                ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
-                ("name", models.CharField(max_length=20)),
-            ],
-        ),
         migrations.CreateModel(
             name="Talk",
             fields=[
                 ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
                 ("name", models.CharField(max_length=60)),
                 ("description", models.CharField(max_length=300)),
+                (
+                    "level",
+                    models.PositiveIntegerField(
+                        choices=[(1, "Beginner"), (2, "Intermediate"), (3, "Advanced")], default=1
+                    ),
+                ),
+                (
+                    "language",
+                    models.CharField(choices=[("es", "Spanish"), ("en", "English")], default="es", max_length=2),
+                ),
                 ("timezone", models.CharField(max_length=60)),
                 ("comments", models.CharField(max_length=300)),
-                ("language", models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, to="talks.talklanguage")),
-                ("level", models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, to="talks.talklevel")),
                 ("user", models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, to=settings.AUTH_USER_MODEL)),
             ],
         ),

temii/talks/models.py

@@ -1,27 +1,23 @@
 from django.db import models
+from django.utils.translation import gettext_lazy as _
 
 from temii.users.models import User
 
 
-class TalkLanguage(models.Model):
-    name = models.CharField(max_length=60)
-
-    def __str__(self):
-        return self.name
-
-
-class TalkLevel(models.Model):
-    name = models.CharField(max_length=20)
-
-    def __str__(self):
-        return self.name
+class Talk(models.Model):
+    class Level(models.IntegerChoices):
+        BEGINNER = 1, _("Beginner")
+        INTERMEDIATE = 2, _("Intermediate")
+        ADVANCED = 3, _("Advanced")
 
+    class Language(models.TextChoices):
+        ES = "es", _("Spanish")
+        EN = "en", _("English")
 
-class Talk(models.Model):
     user = models.ForeignKey(User, on_delete=models.PROTECT)
     name = models.CharField(max_length=60)
     description = models.CharField(max_length=300)
-    level = models.ForeignKey(TalkLevel, on_delete=models.PROTECT)
-    language = models.ForeignKey(TalkLanguage, on_delete=models.PROTECT)
+    level = models.PositiveIntegerField(choices=Level.choices, default=Level.BEGINNER)
+    language = models.CharField(max_length=2, choices=Language.choices, default=Language.ES)
     timezone = models.CharField(max_length=60)
     comments = models.CharField(max_length=300)