[ENHANCEMENT] Add dryrun to alerts-index command #249

line · Mar 2, 2020 · 95760fd · 95760fd
2 parents 4686a90 + 44beca3
commit 95760fd
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/promgen/management/commands/alerts-index.py b/promgen/management/commands/alerts-index.py
@@ -7,7 +7,20 @@
 
 
 class Command(BaseCommand):
-    def handle(self, **kargs):
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--force",
+            dest="dryrun",
+            action="store_false",
+            help="Defaults to dry run. Use to execute operation",
+        )
+
+    def handle(self, dryrun, **kargs):
         for alert in models.Alert.objects.filter(alertlabel__isnull=True):
+            if dryrun:
+                labels = alert.json.get("commonLabels")
+                self.stderr.write("alert_id: %s, labels: %s" % (alert.pk, labels))
+                continue
+
             tasks.index_alert.delay(alert.pk)
             time.sleep(0.1)