feat(vote): Created a election status model

app/org_eleicoes/votepeloclima/candidature/migrations/0023_electionresult.py

@@ -0,0 +1,23 @@
+# Generated by Django 4.2 on 2024-10-07 22:04
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('candidature', '0022_alter_candidature_ballot_name_and_more'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ElectionResult',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('year', models.PositiveIntegerField(verbose_name='Ano da eleição')),
+                ('status', models.CharField(choices=[('eleita', 'Eleita/o'), ('segundo_turno', 'Segundo Turno'), ('nao_eleito', 'Não Eleita/o')], max_length=20, verbose_name='Status da eleição')),
+                ('candidature', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='election_results', to='candidature.candidature')),
+            ],
+        ),
+    ]

app/org_eleicoes/votepeloclima/candidature/models.py

@@ -45,15 +45,6 @@ class Candidature(models.Model):
     created_at = models.DateTimeField(verbose_name="Criado em", auto_now_add=True, null=True, blank=True)
     updated_at = models.DateTimeField(verbose_name="Atualizado em", auto_now=True, null=True, blank=True)
 
-    election_status = models.CharField(
-        max_length=20,
-        choices=ElectionStatus.choices,
-        blank=True, 
-        null=True,
-        verbose_name="Status da Eleição"
-    )
-    election_year = models.PositiveIntegerField(blank=True, null=True, verbose_name="Ano da Eleição")
-
     class Meta:
         verbose_name = "Candidatura"
         ordering = ["-updated_at"]
@@ -113,13 +104,23 @@ def get_proposes_items(self):
 
         return proposes_list
 
+    @property
+    def get_election_result(self, year):
+        return self.election_results.filter(year=year).first()
+
 
     def save(self, *args, **kwargs):
         if not self.slug:
             self.slug = f"{slugify(self.ballot_name)}-{self.number_id}"
         super().save(*args, **kwargs)
 
 
+class ElectionResult(models.Model):
+    candidature = models.ForeignKey('Candidature', on_delete=models.CASCADE, related_name='election_results')
+    year = models.PositiveIntegerField(verbose_name="Ano da eleição")
+    status = models.CharField(max_length=20, choices=ElectionStatus.choices, verbose_name="Status da eleição")
+
+
 class CandidatureFlow(models.Model):
     # Propriedades só podem ser editadas quando status for `draft`
     photo = models.ImageField(upload_to="candidatures/photos/", null=True, verbose_name="Foto")