add TDAMM_TAG_CHOICES to collection_choice_fields

Author: Kirandawadi

sde_collections/admin.py

@@ -8,6 +8,7 @@
 from .models.collection import Collection, WorkflowHistory
 from .models.pattern import DivisionPattern, IncludePattern, TitlePattern
 from .tasks import import_candidate_urls_from_api
+from .models.collection_choice_fields import TDAMMTags
 
 
 @admin.action(description="Generate deployment message")
@@ -267,14 +268,14 @@ def exclude_and_delete_children(modeladmin, request, queryset):
 class CandidateURLForm(forms.ModelForm):
     # Define the fields as MultipleChoiceFields with checkboxes
     tdamm_tag_manual = forms.MultipleChoiceField(
-        choices=CandidateURL.TDAMM_TAG_CHOICES,
+        choices=TDAMMTags.choices,
         required=False,
         label="TDAMM Manual Tags",
         widget=forms.CheckboxSelectMultiple,
     )
 
     tdamm_tag_ml = forms.MultipleChoiceField(
-        choices=CandidateURL.TDAMM_TAG_CHOICES,
+        choices=TDAMMTags.choices,
         required=False,
         label="TDAMM ML Tags",
         widget=forms.CheckboxSelectMultiple,

sde_collections/models/candidate_url.py

@@ -7,7 +7,7 @@
 
 from ..utils.paired_field_descriptor import PairedFieldDescriptor
 from .collection import Collection
-from .collection_choice_fields import Divisions, DocumentTypes
+from .collection_choice_fields import Divisions, DocumentTypes, TDAMMTags
 from .pattern import ExcludePattern, TitlePattern
 
 
@@ -82,53 +82,9 @@ class CandidateURL(models.Model):
         help_text="Helps keep track if the Current URL is present in production or not",
     )
     is_tdamm = models.BooleanField("Is TDAMM?", default=False, help_text="Enable TDAMM tagging for this URL")
-
-    TDAMM_TAG_CHOICES = [
-        ("MMA_M_EM", "Messenger - EM Radiation"),
-        ("MMA_M_EM_G", "Messenger - EM Radiation - Gamma rays"),
-        ("MMA_M_EM_X", "Messenger - EM Radiation - X-rays"),
-        ("MMA_M_EM_U", "Messenger - EM Radiation - Ultraviolet"),
-        ("MMA_M_EM_O", "Messenger - EM Radiation - Optical"),
-        ("MMA_M_EM_I", "Messenger - EM Radiation - Infrared"),
-        ("MMA_M_EM_M", "Messenger - EM Radiation - Microwave"),
-        ("MMA_M_EM_R", "Messenger - EM Radiation - Radio"),
-        ("MMA_M_G", "Messenger - Gravitational Waves"),
-        ("MMA_M_G_CBI", "Messenger - Gravitational Waves - Compact Binary Inspiral"),
-        ("MMA_M_G_S", "Messenger - Gravitational Waves - Stochastic"),
-        ("MMA_M_G_CON", "Messenger - Gravitational Waves - Continuous"),
-        ("MMA_M_G_B", "Messenger - Gravitational Waves - Burst"),
-        ("MMA_M_C", "Messenger - Cosmic Rays"),
-        ("MMA_M_N", "Messenger - Neutrinos"),
-        ("MMA_O_BI", "Objects - Binaries"),
-        ("MMA_O_BI_BBH", "Objects - Binaries - Binary Black Holes"),
-        ("MMA_O_BI_BNS", "Objects - Binaries - Binary Neutron Stars"),
-        ("MMA_O_BI_C", "Objects - Binaries - Cataclysmic Variables"),
-        ("MMA_O_BI_N", "Objects - Binaries - Neutron Star-Black Hole"),
-        ("MMA_O_BI_B", "Objects - Binaries - Binary Pulsars"),
-        ("MMA_O_BI_W", "Objects - Binaries - White Dwarf Binaries"),
-        ("MMA_O_BH", "Objects - Black Holes"),
-        ("MMA_O_BH_AGN", "Objects - Black Holes - Active Galactic Nuclei"),
-        ("MMA_O_BH_IM", "Objects - Black Holes - Intermediate mass"),
-        ("MMA_O_BH_STM", "Objects - Black Holes - Stellar mass"),
-        ("MMA_O_BH_SUM", "Objects - Black Holes - Supermassive"),
-        ("MMA_O_E", "Objects - Exoplanets"),
-        ("MMA_O_N", "Objects - Neutron Stars"),
-        ("MMA_O_N_M", "Objects - Neutron Stars - Magnetars"),
-        ("MMA_O_N_P", "Objects - Neutron Stars - Pulsars"),
-        ("MMA_O_N_PWN", "Objects - Neutron Stars - Pulsar Wind Nebula"),
-        ("MMA_O_S", "Objects - Supernova Remnants"),
-        ("MMA_S_F", "Signals - Fast Radio Bursts"),
-        ("MMA_S_G", "Signals - Gamma-ray Bursts"),
-        ("MMA_S_K", "Signals - Kilonovae"),
-        ("MMA_S_N", "Signals - Novae"),
-        ("MMA_S_P", "Signals - Pevatrons"),
-        ("MMA_S_ST", "Signals - Stellar flares"),
-        ("MMA_S_SU", "Signals - Supernovae"),
-    ]
-
     tdamm_tag = PairedFieldDescriptor(
         field_name="tdamm_tag",
-        field_type=ArrayField(models.CharField(max_length=255, choices=TDAMM_TAG_CHOICES), blank=True, null=True),
+        field_type=ArrayField(models.CharField(max_length=255, choices=TDAMMTags.choices), blank=True, null=True),
         switch="is_tdamm",
         verbose_name="TDAMM Tags",
     )

sde_collections/models/collection_choice_fields.py

@@ -97,3 +97,56 @@ class WorkflowStatusChoices(models.IntegerChoices):
     PROD_MAJOR = 16, "Prod: Major Issues"
     MERGE_PENDING = 17, "Code Merge Pending"
     NEEDS_DELETE = 19, "Delete from Prod"
+
+
+class TDAMMTags(models.TextChoices):
+    """TDAMM (Tagged Data for Multi-Messenger Astronomy) tag choices."""
+
+    MMA_M_EM = "MMA_M_EM", "Messenger - EM Radiation"
+    MMA_M_EM_G = "MMA_M_EM_G", "Messenger - EM Radiation - Gamma rays"
+    MMA_M_EM_X = "MMA_M_EM_X", "Messenger - EM Radiation - X-rays"
+    MMA_M_EM_U = "MMA_M_EM_U", "Messenger - EM Radiation - Ultraviolet"
+    MMA_M_EM_O = "MMA_M_EM_O", "Messenger - EM Radiation - Optical"
+    MMA_M_EM_I = "MMA_M_EM_I", "Messenger - EM Radiation - Infrared"
+    MMA_M_EM_M = "MMA_M_EM_M", "Messenger - EM Radiation - Microwave"
+    MMA_M_EM_R = "MMA_M_EM_R", "Messenger - EM Radiation - Radio"
+    MMA_M_G = "MMA_M_G", "Messenger - Gravitational Waves"
+    MMA_M_G_CBI = "MMA_M_G_CBI", "Messenger - Gravitational Waves - Compact Binary Inspiral"
+    MMA_M_G_S = "MMA_M_G_S", "Messenger - Gravitational Waves - Stochastic"
+    MMA_M_G_CON = "MMA_M_G_CON", "Messenger - Gravitational Waves - Continuous"
+    MMA_M_G_B = "MMA_M_G_B", "Messenger - Gravitational Waves - Burst"
+    MMA_M_C = "MMA_M_C", "Messenger - Cosmic Rays"
+    MMA_M_N = "MMA_M_N", "Messenger - Neutrinos"
+    MMA_O_BI = "MMA_O_BI", "Objects - Binaries"
+    MMA_O_BI_BBH = "MMA_O_BI_BBH", "Objects - Binaries - Binary Black Holes"
+    MMA_O_BI_BNS = "MMA_O_BI_BNS", "Objects - Binaries - Binary Neutron Stars"
+    MMA_O_BI_C = "MMA_O_BI_C", "Objects - Binaries - Cataclysmic Variables"
+    MMA_O_BI_N = "MMA_O_BI_N", "Objects - Binaries - Neutron Star-Black Hole"
+    MMA_O_BI_B = "MMA_O_BI_B", "Objects - Binaries - Binary Pulsars"
+    MMA_O_BI_W = "MMA_O_BI_W", "Objects - Binaries - White Dwarf Binaries"
+    MMA_O_BH = "MMA_O_BH", "Objects - Black Holes"
+    MMA_O_BH_AGN = "MMA_O_BH_AGN", "Objects - Black Holes - Active Galactic Nuclei"
+    MMA_O_BH_IM = "MMA_O_BH_IM", "Objects - Black Holes - Intermediate mass"
+    MMA_O_BH_STM = "MMA_O_BH_STM", "Objects - Black Holes - Stellar mass"
+    MMA_O_BH_SUM = "MMA_O_BH_SUM", "Objects - Black Holes - Supermassive"
+    MMA_O_E = "MMA_O_E", "Objects - Exoplanets"
+    MMA_O_N = "MMA_O_N", "Objects - Neutron Stars"
+    MMA_O_N_M = "MMA_O_N_M", "Objects - Neutron Stars - Magnetars"
+    MMA_O_N_P = "MMA_O_N_P", "Objects - Neutron Stars - Pulsars"
+    MMA_O_N_PWN = "MMA_O_N_PWN", "Objects - Neutron Stars - Pulsar Wind Nebula"
+    MMA_O_S = "MMA_O_S", "Objects - Supernova Remnants"
+    MMA_S_F = "MMA_S_F", "Signals - Fast Radio Bursts"
+    MMA_S_G = "MMA_S_G", "Signals - Gamma-ray Bursts"
+    MMA_S_K = "MMA_S_K", "Signals - Kilonovae"
+    MMA_S_N = "MMA_S_N", "Signals - Novae"
+    MMA_S_P = "MMA_S_P", "Signals - Pevatrons"
+    MMA_S_ST = "MMA_S_ST", "Signals - Stellar flares"
+    MMA_S_SU = "MMA_S_SU", "Signals - Supernovae"
+
+    @classmethod
+    def lookup_by_text(cls, text: str) -> str | None:
+        """Look up a TDAMM tag by its display text."""
+        for choice in cls.choices:
+            if choice[1].lower() == text.lower():
+                return choice[0]
+        return None