Add support for disabling the operator

Signed-off-by: Marcelo Henrique Neppel <[email protected]>
canonical · Mar 23, 2024 · ddf5e67 · ddf5e67
1 parent 3b3e0f2
commit ddf5e67
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/charm.py b/src/charm.py
@@ -9,6 +9,8 @@
 import os
 import platform
 import subprocess
+import sys
+from pathlib import Path
 from typing import Dict, List, Literal, Optional, Set, get_args
 
 import psycopg2
@@ -106,6 +108,16 @@ class PostgresqlOperatorCharm(TypedCharmBase[CharmConfig]):
     def __init__(self, *args):
         super().__init__(*args)
 
+        # Support for disabling the operator.
+        disable_file = Path(f"{os.environ.get('CHARM_DIR')}/disable")
+        if disable_file.exists():
+            logger.warning(
+                f"\n\tDisable file `{disable_file.resolve()}` found, the charm will skip all events."
+                "\n\tTo resume normal operations, please remove the file."
+            )
+            self.unit.status = BlockedStatus("Disabled")
+            sys.exit(0)
+
         self.peer_relation_app = DataPeer(
             self,
             relation_name=PEER,