diff --git a/db/migrations/0001_initial.py b/db/migrations/0001_initial.py new file mode 100644 index 000000000..84ff25fc4 --- /dev/null +++ b/db/migrations/0001_initial.py @@ -0,0 +1,106 @@ +# Generated by Django 5.2.13 on 2026-05-03 16:52 + +import django.db.models.deletion +from django.db import migrations, models + + +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="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( + blank=True, + null=True, + on_delete=django.db.models.deletion.SET_NULL, + related_name="players", + to="db.guild", + ), + ), + ( + "race", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, + related_name="players", + to="db.race", + ), + ), + ], + ), + 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, + related_name="skills", + to="db.race", + ), + ), + ], + ), + ] diff --git a/db/models.py b/db/models.py index 137941ffa..1f07bdb38 100644 --- a/db/models.py +++ b/db/models.py @@ -1 +1,40 @@ 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, + related_name="skills" + ) + + +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, + related_name="players" + ) + guild = models.ForeignKey( + Guild, + on_delete=models.SET_NULL, + related_name="players", + null=True, + blank=True + ) + created_at = models.DateTimeField(auto_now_add=True) diff --git a/main.py b/main.py index e71f55d89..c626d9578 100644 --- a/main.py +++ b/main.py @@ -1,11 +1,51 @@ +import json + import init_django_orm # noqa: F401 from db.models import Race, Skill, Player, Guild def main() -> None: - pass + with open("players.json", "r") as f: + data = json.load(f) + + for name in data: + user_data = data[name] + race_data = user_data["race"] + guild_data = user_data.get("guild") + + race, _ = Race.objects.get_or_create( + name=race_data["name"], + defaults={ + "description": race_data["description"] + } + ) + + for skill_data in race_data["skills"]: + skill, _ = Skill.objects.get_or_create( + name=skill_data["name"], + defaults={ + "bonus": skill_data["bonus"], + "race": race + } + ) + if guild_data: + guild, _ = Guild.objects.get_or_create( + name=guild_data["name"], + defaults={ + "description": guild_data.get("description") + } + ) + else: + guild = None -if __name__ == "__main__": - main() + Player.objects.get_or_create( + nickname=name, + defaults={ + "email": user_data["email"], + "bio": user_data["bio"], + "race": race, + "guild": guild + } + )