Update import to include area owners

Author: prophile

scripts/import.py

@@ -30,6 +30,21 @@
     'Website',
 )
 
+AREA_OWNERS = (
+    'committee',
+    'health-safety',
+    'media-press',
+    'teams',
+    'mentoring',
+    'kit',
+    'event-logistics',
+    'livestream',
+    'production',
+    'game',
+    'volunteer-coordination',
+    'simulator',
+)
+
 
 def process(element_name: str, *, year: str, handle_dep: Callable[[str], int]) -> Ticket:
     """
@@ -64,6 +79,10 @@ def process(element_name: str, *, year: str, handle_dep: Callable[[str], int]) -
     if component not in COMPONENTS:
         raise RuntimeError(f"{path} has an unknown component: {component}")
 
+    area_owner = raw_elements.get('area-owner')
+    if area_owner not in AREA_OWNERS:
+        raise RuntimeError(f"{path} has an unknown area owner: {area_owner}")
+
     milestone = raw_elements.get('milestone')
     dependencies = raw_elements.get('dependencies', ())
 
@@ -75,6 +94,7 @@ def process(element_name: str, *, year: str, handle_dep: Callable[[str], int]) -
         priority=priority,
         milestone=milestone,
         original_name=element_name,
+        area_owner=area_owner,
         description=description,
         dependencies=computed_dependencies,
     )

scripts/import_backends.py

@@ -151,6 +151,20 @@ class GitHubBackend:
         'blocker': 'I: Must Have',
     }
 
+    AREA_OWNER_MAPPING: Dict[str, str] = {
+        'committee': 'AO: Committee',
+        'health-safety': 'AO: Health & Safety',
+        'media-press': 'AO: Media & Press',
+        'teams': 'AO: Teams',
+        'mentoring': 'AO: Mentoring',
+        'kit': 'AO: Kit',
+        'event-logistics': 'AO: Event Logistics',
+        'game': 'AO: Game',
+        'production': 'AO: Production',
+        'simulator': 'AO: Simulator',
+        'livestream': 'AO: Livestream',
+    }
+
     def __init__(self, repo_name: str) -> None:
         self.github = github.Github(get_github_credential())
         self.repo = self.github.get_repo(repo_name)
@@ -170,6 +184,9 @@ def __init__(self, repo_name: str) -> None:
             self._component_priority_mapping = {
                 k: labels[v] for k, v in self.COMPONENT_PRIORITY_MAPPING.items()
             }
+            self._area_owner_mapping = {
+                k: labels[v] for k, v in self.AREA_OWNER_MAPPING.items()
+            }
         except KeyError as e:
             raise ValueError(
                 f"Label {e} does not exist within {repo_name!r}",
@@ -213,6 +230,7 @@ def get_or_create_milestone(self, title: str) -> github.Milestone.Milestone:
     def labels(self, ticket: Ticket) -> List[github.Label.Label]:
         labels = [self._component_priority_mapping[ticket.priority]]
         labels += self._component_label_mapping[ticket.component]
+        labels.append(self._area_owner_mapping[ticket.area_owner])
         return labels
 
     def submit(self, ticket: Ticket) -> int:

scripts/ticket_type.py

@@ -7,5 +7,6 @@ class Ticket(NamedTuple):
     component: str
     original_name: str
     milestone: str
+    area_owner: str
     description: str
     dependencies: List[int]