Add country code field to driver and team

Author: jolpica

.pre-commit-config.yaml

@@ -1,7 +1,7 @@
 repos:
 - repo: https://github.com/astral-sh/ruff-pre-commit
   # Ruff version.
-  rev: v0.8.3
+  rev: v0.9.9
   hooks:
     # Run the linter.
     - id: ruff

docs/database.svg

@@ -107,6 +107,7 @@ class TeamObject(F1ObjectSchema):
     reference: str | None = None
     name: str | None = None
     nationality: str | None = None
+    country_code: str | None = Field(None, max_length=3)
     wikipedia: HttpUrl | None = None
 
 
@@ -126,6 +127,7 @@ class DriverObject(F1ObjectSchema):
     surname: str | None = None
     abbreviation: str | None = None
     nationality: str | None = None
+    country_code: str | None = Field(None, max_length=3)
     permanent_car_number: PositiveInt | None = None
     date_of_birth: date | None = None
     wikipedia: HttpUrl | None = None

jolpica/formula_one/importer/json_models.py

@@ -0,0 +1,44 @@
+# Generated by Django 5.1.6 on 2025-03-02 20:06
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("formula_one", "0003_lap_is_deleted"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="driver",
+            name="country_code",
+            field=models.CharField(
+                blank=True,
+                max_length=3,
+                null=True,
+                verbose_name="3 Letter country code (ISO 3166-1 alpha-3)",
+            ),
+        ),
+        migrations.AddField(
+            model_name="team",
+            name="country_code",
+            field=models.CharField(
+                blank=True,
+                max_length=3,
+                null=True,
+                verbose_name="3 Letter country code (ISO 3166-1 alpha-3)",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="session",
+            name="point_system",
+            field=models.ForeignKey(
+                default=1,
+                on_delete=django.db.models.deletion.PROTECT,
+                related_name="sessions",
+                to="formula_one.pointsystem",
+            ),
+        ),
+    ]

jolpica/formula_one/migrations/0004_driver_country_code_team_country_code_and_more.py

@@ -27,6 +27,7 @@ class Driver(models.Model):
     surname = models.CharField(max_length=255)
     abbreviation = models.CharField(max_length=10, null=True, blank=True)
     nationality = models.CharField(max_length=255, null=True, blank=True)
+    country_code = models.CharField("3 Letter country code (ISO 3166-1 alpha-3)", max_length=3, null=True, blank=True)
     permanent_car_number = models.PositiveSmallIntegerField(null=True, blank=True)
     date_of_birth = models.DateField(null=True, blank=True)
     wikipedia = models.URLField(max_length=255, null=True, blank=True)

jolpica/formula_one/models/driver.py

@@ -254,6 +254,7 @@ May track reserve drivers with no sessions, or junior drivers only taking part i
   surname char [not null]
   abbreviation char [null]
   nationality char [null]
+  country_code char [null]
   permanent_car_number positive_small_integer [null]
   date_of_birth date [null]
   wikipedia url [null]
@@ -528,6 +529,7 @@ Information about a Constructor / Team
   reference char [unique, null]
   name char [not null]
   nationality char [null]
+  country_code char [null]
   wikipedia url [null]
 
   indexes {

jolpica/formula_one/models/models.dbml

@@ -37,6 +37,7 @@ class Team(models.Model):
     reference = models.CharField(max_length=32, unique=True, null=True, blank=True)
     name = models.CharField(max_length=255)
     nationality = models.CharField(max_length=255, null=True, blank=True)
+    country_code = models.CharField("3 Letter country code (ISO 3166-1 alpha-3)", max_length=3, null=True, blank=True)
     wikipedia = models.URLField(max_length=255, null=True, blank=True)
 
     def __str__(self) -> str:

jolpica/formula_one/models/team.py

@@ -260,9 +260,9 @@ def test_factory_deserialisers_require_at_least_one_object():
         data = {"object_type": object_type, "foreign_keys": {}, "objects": []}
         result = deserialiser.deserialise(data)
 
-        assert "too_short" in [
-            error["type"] for error in result.errors if "objects" in error["loc"]
-        ], "Not all deserialisers requite at least 1 object"
+        assert "too_short" in [error["type"] for error in result.errors if "objects" in error["loc"]], (
+            "Not all deserialisers requite at least 1 object"
+        )
 
 
 @pytest.fixture