workercommon 0.4.2__tar.gz → 0.4.3__tar.gz

{workercommon-0.4.2 → workercommon-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workercommon
-Version: 0.4.2
+Version: 0.4.3
 Summary: Support code for various projects
 Author-email: "Neil E. Hodges" <47hasbegun@gmail.com>
 License-Expression: BSD-3-Clause

{workercommon-0.4.2 → workercommon-0.4.3}/pyproject.toml

@@ -16,7 +16,7 @@ dependencies = [
   "psycopg2",
   "pika",
 ]
-version = "0.4.2"
+version = "0.4.3"
 license-files = ["LICENSE.TXT"]
 classifiers = [
   "Programming Language :: Python :: 3",

{workercommon-0.4.2 → workercommon-0.4.3}/workercommon/worker.py

@@ -4,19 +4,21 @@ import sys
 if sys.version_info < (3, 8):
     raise RuntimeError("At least Python 3.8 is required")
 
-from pathlib import Path
 from .database import Connection, Cursor
 from . import rabbitmqueue
-import json, re, pika.connection, pika.channel, pika.spec, logging
+import json, pika.connection, pika.channel, pika.spec, logging
 from collections.abc import Mapping
-from time import sleep
 from threading import Thread, Lock
 from typing import Iterable, Optional, Any, Callable, List
-from multiprocessing import Process, Manager, Event
+from multiprocessing import Process
 from multiprocessing.managers import SyncManager
 from concurrent.futures import ThreadPoolExecutor
 
 
+class NoDatabase(RuntimeError):
+    pass
+
+
 class BaseWorker(object):
     def start(self) -> None:
         raise NotImplementedError
@@ -43,7 +45,7 @@ class ThreadWorker(BaseWorker, Thread):
 class ReaderWorker(ThreadWorker):
     def __init__(
         self,
-        database_source: Callable[[], Connection],
+        database_source: Optional[Callable[[], Connection]],
         readq_parameters: rabbitmqueue.Parameters,
         any_exchange: bool = False,
         **readq_arguments: Any,
@@ -152,7 +154,7 @@ class ReaderWorker(ThreadWorker):
     def get_cursor(self) -> Cursor:
         if self.database is None:
             if self.database_source is None:
-                raise RuntimeError("Database source is unavailable")
+                raise NoDatabase("Database source is unavailable")
             self.database = self.database_source()
 
         return self.database.cursor()
@@ -192,8 +194,11 @@ class ReaderWorker(ThreadWorker):
             filtered = self.filter_message(root)
 
             if filtered is not None:
-                with self.get_cursor() as cursor:
-                    self.handle_message(cursor, filtered)
+                try:
+                    with self.get_cursor() as cursor:
+                        self.handle_message(cursor, filtered)
+                except NoDatabase:
+                    self.handle_message_nodb(filtered)
 
             logging.debug(f"Successfully handled {method.delivery_tag}")
             self.get_readq().add_callback_threadsafe(
@@ -242,6 +247,9 @@ class ReaderWorker(ThreadWorker):
     def handle_message(self, cursor: Cursor, message: Any) -> None:
         raise NotImplementedError
 
+    def handle_message_nodb(self, message: Any) -> None:
+        raise NotImplementedError
+
     def stop(self) -> None:
         with self.lock:
             if self.readq is not None:

{workercommon-0.4.2 → workercommon-0.4.3}/workercommon.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workercommon
-Version: 0.4.2
+Version: 0.4.3
 Summary: Support code for various projects
 Author-email: "Neil E. Hodges" <47hasbegun@gmail.com>
 License-Expression: BSD-3-Clause