diff --git a/db/migrations/0001_initial.py b/db/migrations/0001_initial.py new file mode 100644 index 00000000..5170442f --- /dev/null +++ b/db/migrations/0001_initial.py @@ -0,0 +1,52 @@ +# Generated by Django 4.0.2 on 2023-08-22 10:16 + +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.CharField(max_length=255)), + ('bio', models.CharField(max_length=255)), + ('created_at', models.DateTimeField(auto_now=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')), + ], + ), + ] diff --git a/db/migrations/0002_alter_player_email.py b/db/migrations/0002_alter_player_email.py new file mode 100644 index 00000000..f79d1c9b --- /dev/null +++ b/db/migrations/0002_alter_player_email.py @@ -0,0 +1,18 @@ +# Generated by Django 4.0.2 on 2023-08-22 12:33 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('db', '0001_initial'), + ] + + operations = [ + migrations.AlterField( + model_name='player', + name='email', + field=models.EmailField(max_length=255), + ), + ] diff --git a/db/models.py b/db/models.py index 137941ff..dddeb6e6 100644 --- a/db/models.py +++ b/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, null=True, on_delete=models.SET_NULL) + created_at = models.DateTimeField(auto_now=True) diff --git a/main.py b/main.py index e71f55d8..4e0e5c3b 100644 --- a/main.py +++ b/main.py @@ -1,10 +1,40 @@ import init_django_orm # noqa: F401 +import json from db.models import Race, Skill, Player, Guild def main() -> None: - pass + with open("players.json") as players: + players_data = json.load(players) + + for player_nickname, player_attributes in players_data.items(): + race, created = Race.objects.get_or_create( + name=player_attributes["race"]["name"], + description=player_attributes["race"]["description"] + ) + if created: + for skill in player_attributes["race"].get("skills", []): + Skill.objects.create( + name=skill["name"], + bonus=skill["bonus"], + race=race + ) + + guild = None + if player_attributes.get("guild"): + guild, created = Guild.objects.get_or_create( + name=player_attributes["guild"]["name"], + description=player_attributes["guild"]["description"] + ) + + player, created = Player.objects.get_or_create( + nickname=player_nickname, + email=player_attributes["email"], + bio=player_attributes["bio"], + race=race, + guild=guild + ) if __name__ == "__main__":