diff --git a/db/migrations/0001_initial.py b/db/migrations/0001_initial.py new file mode 100644 index 000000000..c201e1842 --- /dev/null +++ b/db/migrations/0001_initial.py @@ -0,0 +1,52 @@ +# Generated by Django 6.0.4 on 2026-05-19 00:43 + +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(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..9b42f98dd 100644 --- a/db/models.py +++ b/db/models.py @@ -1 +1,56 @@ +# db/models.py + from django.db import models + + +class Race(models.Model): + name = models.CharField(max_length=255, unique=True) + description = models.TextField(blank=True) + + def __str__(self) -> str: + return self.name + + +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", + ) + + def __str__(self) -> str: + return self.name + + +class Guild(models.Model): + name = models.CharField(max_length=255, unique=True) + description = models.TextField(null=True) + + def __str__(self) -> str: + return self.name + + +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, + null=True, + related_name="players", + ) + + created_at = models.DateTimeField(auto_now_add=True) + + def __str__(self) -> str: + return self.nickname diff --git a/main.py b/main.py index e71f55d89..8c55ff014 100644 --- a/main.py +++ b/main.py @@ -1,10 +1,51 @@ -import init_django_orm # noqa: F401 +import json -from db.models import Race, Skill, Player, Guild +from db.models import Guild, Player, Race, Skill def main() -> None: - pass + with open("players.json", "r") as file: + players_data = json.load(file) + + for nickname, player_data in players_data.items(): + race_data = player_data["race"] + + race, _ = Race.objects.get_or_create( + name=race_data["name"], + defaults={ + "description": race_data.get("description", ""), + }, + ) + + for skill_data in race_data.get("skills", []): + Skill.objects.get_or_create( + name=skill_data["name"], + defaults={ + "bonus": skill_data["bonus"], + "race": race, + }, + ) + + guild = None + guild_data = player_data.get("guild") + + if guild_data: + guild, _ = Guild.objects.get_or_create( + name=guild_data["name"], + defaults={ + "description": guild_data.get("description"), + }, + ) + + Player.objects.get_or_create( + nickname=nickname, + defaults={ + "email": player_data["email"], + "bio": player_data["bio"], + "race": race, + "guild": guild, + }, + ) if __name__ == "__main__":