Solution-py-game-models

db/migrations/0001_initial.py

@@ -0,0 +1,52 @@
+# Generated by Django 4.0.2 on 2024-11-03 01:48
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Guild',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=255, unique=True)),
+                ('description', models.TextField(null=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Race',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=255, unique=True)),
+                ('description', models.TextField(blank=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Skill',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=255, unique=True)),
+                ('bonus', models.CharField(max_length=255)),
+                ('race', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='db.race')),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Player',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('nickname', models.CharField(max_length=255, unique=True)),
+                ('email', models.EmailField(max_length=255)),
+                ('bio', models.CharField(max_length=255)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('guild', models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to='db.guild')),
+                ('race', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='db.race')),
+            ],
+        ),
+    ]

db/models.py

@@ -1 +1,26 @@
 from django.db import models
+
+
+class Race(models.Model):
+    name = models.CharField(max_length=255, unique=True)
+    description = models.TextField(blank=True)
+
+
+class Skill(models.Model):
+    name = models.CharField(max_length=255, unique=True)
+    bonus = models.CharField(max_length=255)
+    race = models.ForeignKey(Race, on_delete=models.CASCADE)
+
+
+class Guild(models.Model):
+    name = models.CharField(max_length=255, unique=True)
+    description = models.TextField(null=True)
+
+
+class Player(models.Model):
+    nickname = models.CharField(max_length=255, unique=True)
+    email = models.EmailField(max_length=255)
+    bio = models.CharField(max_length=255)
+    race = models.ForeignKey(Race, on_delete=models.CASCADE)
+    guild = models.ForeignKey(Guild, on_delete=models.SET_NULL, null=True)
+    created_at = models.DateTimeField(auto_now_add=True)

main.py

@@ -1,10 +1,67 @@
+import os
+import json
+
 import init_django_orm  # noqa: F401
 
 from db.models import Race, Skill, Player, Guild
 
 
 def main() -> None:
-    pass
+    players = file_data("players.json")
+    create_db(players)
+
+
+def create_db(players: dict) -> None:
+    for player, info in players.items():
+        email_info = info.get("email")
+        bio_info = info.get("bio")
+
+        if race_info := info.get("race"):
+            race_name = race_info.get("name")
+            race_description = race_info.get("description")
+
+            race, race_created = Race.objects.get_or_create(
+                name=race_name,
+                description=race_description
+            )
+
+            if skills_info := race_info.get("skills"):
+                for skill in skills_info:
+                    skill_name = skill.get("name")
+                    skill_bonus = skill.get("bonus")
+
+                    Skill.objects.get_or_create(
+                        name=skill_name, bonus=skill_bonus, race=race
+                    )
+        else:
+            race = None
+
+        if guild_info := info.get("guild"):
+            guild_name = guild_info.get("name")
+            guild_description = guild_info.get("description")
+
+            guild, guild_created = Guild.objects.get_or_create(
+                name=guild_name,
+                description=guild_description
+            )
+        else:
+            guild = None
+
+        Player.objects.create(
+            nickname=player,
+            email=email_info,
+            bio=bio_info,
+            race=race,
+            guild=guild
+        )
+
+
+def file_data(path: str) -> dict:
+    file_path = os.path.join(path)
+    with open(file_path) as f:
+        players = json.load(f)
+
+    return players
 
 
 if __name__ == "__main__":