added init, debugged drain function, edited docstrings

Author: ashum68

worker/__init__.py

@@ -1,5 +1,5 @@
 """
-Worker Queue
+Worker Queue.
 """
 
 import multiprocessing.managers
@@ -21,7 +21,7 @@ def __init__(self, mp_manager: multiprocessing.managers.SyncManager, max_size: i
 
     def fill_queue_with_sentinel(self, timeout: float = 0.0) -> None:
         """
-        Fills the queue with sentinel (None)
+        Fills the queue with sentinel (None).
         """
         if timeout <= 0.0:
             timeout = self.__QUEUE_TIMEOUT
@@ -35,19 +35,21 @@ def fill_queue_with_sentinel(self, timeout: float = 0.0) -> None:
 
     def drain_queue(self, timeout: float = 0.0) -> None:
         """
-        Drains the queue
+        Drains the queue.
         """
         if timeout <= 0.0:
             timeout = self.__QUEUE_TIMEOUT
 
         try:
             self.queue.get(timeout=timeout)
+            for _ in range(1, self.max_size):
+                self.queue.get(timeout=timeout)
         except queue.Empty:
             return
 
     def fill_and_drain_queue(self) -> None:
         """
-        Fills the queue with sentinel and then drains it
+        Fills the queue with sentinel and then drains it.
         """
         self.fill_queue_with_sentinel()
         time.sleep(self.__QUEUE_DELAY)

worker/queue_wrapper.py

@@ -15,53 +15,53 @@ class WorkerController:
 
     def __init__(self) -> None:
         """
-        Initializes semaphore and queue
+        Initializes semaphore and queue.
         """
         self.__process_limiter = mp.BoundedSemaphore(1)
         self.__is_paused = False
         self.__exit_queue = mp.Queue(1)
 
     def request_pause(self) -> None:
         """
-        Request a worker to pause its process
+        Request a worker to pause its process.
         """
         if not self.__is_paused:
             self.__process_limiter.acquire()
             self.__is_paused = True
 
     def request_resume(self) -> None:
         """
-        Request a worker to resume its process
+        Request a worker to resume its process.
         """
         if self.__is_paused:
             self.__process_limiter.release()
             self.__is_paused = False
 
     def check_pause(self) -> None:
         """
-        If pause requested by main, worker is blocked, otherwise worker continues
+        If pause requested by main, worker is blocked, otherwise worker continues.
         """
         self.__process_limiter.acquire()
         self.__process_limiter.release()
 
     def request_exit(self) -> None:
         """
-        Requests worker to exit its process
+        Requests worker to exit its process.
         """
         time.sleep(self.__QUEUE_DELAY)
         if not self.__exit_queue.empty():
             self.__exit_queue.put(None)
 
     def clear_exit(self) -> None:
         """
-        Clears exit request
+        Clears exit request.
         """
         time.sleep(self.__QUEUE_DELAY)
         if not self.__exit_queue.empty():
             _ = self.__exit_queue.get()
 
     def is_exit_requested(self) -> bool:
         """
-        Returns whether main has requested a worker to exit
+        Returns whether main has requested a worker to exit.
         """
         return not self.__exit_queue.empty()