protection from accidently deleting prod

catenarytransit · Jan 6, 2024 · 2f211ac · 2f211ac
1 parent c154a75
commit 2f211ac
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/ingest_gtfs_schedule/main.rs b/src/ingest_gtfs_schedule/main.rs
@@ -238,6 +238,15 @@ async fn main() -> Result<(), Box<dyn Error>> {
         .unwrap()
         .get::<bool>("softinsert");
 
+    let forcewipe = arguments::parse(std::env::args())
+        .unwrap()
+        .get::<bool>("forcewipe")
+        .unwrap_or_else(|| false);
+
+    if (startfresh.unwrap_or(false) && isprod.unwrap_or(false) && forcewipe) {
+        panic!("Cannot wipe the prod server without --forcewipe true");
+    }
+
     let schemaname = match is_prod {
         Some(s) => {
             if s {