Associate hub with subfields

src/hub/migrations/0023_hub_subfield.py

@@ -0,0 +1,26 @@
+# Generated by Django 4.2.13 on 2024-06-17 22:06
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("topic", "0006_alter_topic_keywords"),
+        ("hub", "0022_alter_hub_concept"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="hub",
+            name="subfield",
+            field=models.OneToOneField(
+                blank=True,
+                null=True,
+                on_delete=django.db.models.deletion.SET_NULL,
+                related_name="hub",
+                to="topic.subfield",
+            ),
+        ),
+    ]

src/hub/migrations/0024_hub_is_used_for_rep.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.13 on 2024-06-17 22:07
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("hub", "0023_hub_subfield"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="hub",
+            name="is_used_for_rep",
+            field=models.BooleanField(default=False),
+        ),
+    ]

src/hub/models.py

@@ -75,6 +75,16 @@ class Hub(models.Model):
         blank=True,
     )
 
+    subfield = models.OneToOneField(
+        "topic.subfield",
+        related_name="hub",
+        on_delete=models.SET_NULL,
+        null=True,
+        blank=True,
+    )
+
+    is_used_for_rep = models.BooleanField(default=False)
+
     def __str__(self):
         return "{}, locked: {}".format(self.name, self.is_locked)
 

src/topic/management/commands/associate_subfields_to_hubs.py

@@ -0,0 +1,38 @@
+"""
+The purpose of this script is to associate hubs with subfields.
+This is a prerequisite for calculating reputation since only specific
+hubs that are tagged with subfields are used for reputation calculation.
+"""
+
+from django.core.management.base import BaseCommand
+
+from hub.models import Hub
+from topic.models import Subfield
+
+
+class Command(BaseCommand):
+    help = "Associate hubs with subfields (Used for rep)"
+
+    def handle(self, *args, **kwargs):
+        subfields = Subfield.objects.all()
+
+        for subfield in subfields:
+            try:
+                hub = Hub.objects.all()
+                continue
+            except Hub.DoesNotExist:
+                hub, created = Hub.objects.get_or_create(
+                    name=subfield.display_name.lower(),
+                    defaults={"subfield": subfield, "is_used_for_rep": True},
+                )
+                if created:
+                    print(
+                        f"Created new hub {hub.name} and associated with subfield {subfield.display_name}."
+                    )
+                else:
+                    hub.subfield = subfield
+                    hub.is_used_for_rep = True
+                    hub.save()
+                    print(
+                        f"Updated hub {hub.name} with subfield {subfield.display_name}"
+                    )