Allow sending of OT creation/activation emails manually (#4706)

* Add manual route for sending OT emails

* fix reference to stage ID

---------

Co-authored-by: DanielRyanSmith <[email protected]>

internals/maintenance_scripts.py

@@ -801,3 +801,39 @@ def calc_needs_work_started_on(
 
     return max(v.set_on for v in votes
                if v.state == Vote.NEEDS_WORK)
+
+
+class SendManualOTCreatedEmail(FlaskHandler):
+  """Manually send an email to origin trial contacts that an origin trial has
+  been created but not yet activated."""
+
+  def get_template_data(self, **kwargs):
+    self.require_cron_header()
+
+    stage_id = kwargs.get('stage_id')
+    stage: Stage|None = Stage.get_by_id(stage_id)
+    if not stage:
+      return f'Stage {stage_id} not found'
+
+    cloud_tasks_helpers.enqueue_task(
+        '/tasks/email-ot-creation-processed',
+        {'stage': converters.stage_to_json_dict(stage)})
+    return 'Email task enqueued.'
+
+
+class SendManualOTActivatedEmail(FlaskHandler):
+  """Manually send an email to origin trial contacts that an origin trial has
+  been created and also activated."""
+
+  def get_template_data(self, **kwargs):
+    self.require_cron_header()
+
+    stage_id = kwargs.get('stage_id')
+    stage: Stage|None = Stage.get_by_id(stage_id)
+    if not stage:
+      return f'Stage {stage_id} not found'
+
+    cloud_tasks_helpers.enqueue_task(
+        '/tasks/email-ot-activated',
+        {'stage': converters.stage_to_json_dict(stage)})
+    return 'Email task enqueued.'

main.py

@@ -348,6 +348,10 @@ class Route:
         maintenance_scripts.BackfillShippingYear),
   Route('/scripts/backfill_gate_dates',
         maintenance_scripts.BackfillGateDates),
+  Route('/scripts/send_ot_creation_email/<int:stage_id>',
+        maintenance_scripts.SendManualOTCreatedEmail),
+  Route('/scripts/send_ot_activation_email/<int:stage_id>',
+        maintenance_scripts.SendManualOTActivatedEmail),
 ]
 
 dev_routes: list[Route] = []