refactor: make timeout configurable by the user

pycasbin · Jul 9, 2024 · 241ac5c · 241ac5c
1 parent 6f3c031
commit 241ac5c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/postgresql_watcher/watcher.py b/postgresql_watcher/watcher.py
@@ -105,13 +105,15 @@ def _create_subscription_process(
         if start_listening:
             self.start()
 
-    def start(self):
+    def start(
+        self,
+        timeout=20, # seconds
+    ):
         if not self.subscription_proces.is_alive():
             # Start listening to messages
             self.subscription_proces.start()
             # And wait for the Process to be ready to listen for updates
             # from PostgreSQL
-            timeout = 20  # seconds
             timeout_time = time() + timeout
             while True:
                 if self.parent_conn.poll():