Prod should be true! (#89)

AetherUnbound · web-flow · commit a01cb4bfc3c6 · 2025-03-16T22:12:27.000-07:00
diff --git a/techbloc_airflow/dags/maintenance/backups/offsite_backups.py b/techbloc_airflow/dags/maintenance/backups/offsite_backups.py
@@ -23,6 +23,7 @@ def backup_service(config: OffsiteConfig):
     backup = SSHOperator(
         task_id=f"backup_{config.name}",
         ssh_conn_id=constants.SSH_MONOLITH_CONN_ID,
+        environment={"IS_PROD": "true"},
         command=config.command,
     )
 
@@ -49,6 +50,7 @@ def notify_backup_complete():
     restart_if_failed = SSHOperator(
         task_id=f"start_if_failed_{config.name}",
         ssh_conn_id=constants.SSH_MONOLITH_CONN_ID,
+        environment={"IS_PROD": "true"},
         command=f"cd {config.folder} && just up",
         trigger_rule=TriggerRule.ONE_FAILED,
     )

Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,7 @@ def backup_service(config: OffsiteConfig):`
`23`	`23`	`backup = SSHOperator(`
`24`	`24`	`task_id=f"backup_{config.name}",`
`25`	`25`	`ssh_conn_id=constants.SSH_MONOLITH_CONN_ID,`
	`26`	`+ environment={"IS_PROD": "true"},`
`26`	`27`	`command=config.command,`
`27`	`28`	`)`
`28`	`29`
`@@ -49,6 +50,7 @@ def notify_backup_complete():`
`49`	`50`	`restart_if_failed = SSHOperator(`
`50`	`51`	`task_id=f"start_if_failed_{config.name}",`
`51`	`52`	`ssh_conn_id=constants.SSH_MONOLITH_CONN_ID,`
	`53`	`+ environment={"IS_PROD": "true"},`
`52`	`54`	`command=f"cd {config.folder} && just up",`
`53`	`55`	`trigger_rule=TriggerRule.ONE_FAILED,`
`54`	`56`	`)`