producer and consumer are mutually exclusive

toluaina · Jan 2, 2024 · a479afc · a479afc
1 parent f6fcbc3
commit a479afc
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/pgsync/sync.py b/pgsync/sync.py
@@ -1339,14 +1339,18 @@ def receive(self) -> None:
 @click.option(
     "--producer",
     is_flag=True,
-    default=True,
+    default=None,
     help="Run as a producer only",
+    cls=MutuallyExclusiveOption,
+    mutually_exclusive=["consumer"],
 )
 @click.option(
     "--consumer",
     is_flag=True,
     help="Run as a consumer only",
-    default=True,
+    default=None,
+    cls=MutuallyExclusiveOption,
+    mutually_exclusive=["producer"],
 )
 @click.option(
     "--polling",
@@ -1451,6 +1455,13 @@ def main(
 
     show_settings(config)
 
+    if producer:
+        consumer = False
+    elif consumer:
+        producer = False
+    else:
+        consumer = producer = True
+
     with Timer():
         if analyze:
             for doc in config_loader(config):