Solution-py-game-models v1.1

mate-academy · Nov 3, 2024 · 9191281 · 9191281
1 parent 8e678b2
commit 9191281
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/db/models.py b/db/models.py
@@ -2,7 +2,13 @@
 
 
 class Race(models.Model):
-    name = models.CharField(max_length=255, unique=True)
+    RACE_TYPE = (
+        ("Elf", "elf race"),
+        ("Dwarf", "dwarf race"),
+        ("Human", "human race"),
+        ("Ork", "ork race")
+    )
+    name = models.CharField(max_length=255, unique=True, choices=RACE_TYPE)
     description = models.TextField(blank=True)
 
 

diff --git a/main.py b/main.py
@@ -47,7 +47,7 @@ def create_db(players: dict) -> None:
         else:
             guild = None
 
-        Player.objects.create(
+        Player.objects.get_or_create(
             nickname=player,
             email=email_info,
             bio=bio_info,