cobweb-launcher 1.2.8__tar.gz → 1.2.10__tar.gz

{cobweb-launcher-1.2.8/cobweb_launcher.egg-info → cobweb-launcher-1.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cobweb-launcher
-Version: 1.2.8
+Version: 1.2.10
 Summary: spider_hole
 Home-page: https://github.com/Juannie-PP/cobweb
 Author: Juannie-PP

{cobweb-launcher-1.2.8 → cobweb-launcher-1.2.10}/cobweb/launchers/launcher.py

@@ -14,7 +14,7 @@ def check_pause(func):
     def wrapper(self, *args, **kwargs):
         while not self._pause.is_set():
             try:
-                func(*args, **kwargs)
+                func(self, *args, **kwargs)
             except Exception as e:
                 logger.info(f"{func.__name__}: " + str(e))
 

cobweb-launcher-1.2.10/cobweb/pipelines/pipeline.py

@@ -0,0 +1,68 @@
+import time
+import threading
+
+from abc import ABC, abstractmethod
+from cobweb.base import BaseItem, Queue, logger
+
+
+class Pipeline(threading.Thread, ABC):
+
+    def __init__(
+            self,
+            stop: threading.Event,
+            pause: threading.Event,
+            upload: Queue, done: Queue,
+            upload_size: int,
+            wait_seconds: int
+    ):
+        super().__init__()
+        self._stop = stop
+        self._pause = pause
+        self._upload = upload
+        self._done = done
+
+        self.upload_size = upload_size
+        self.wait_seconds = wait_seconds
+
+    @abstractmethod
+    def build(self, item: BaseItem) -> dict:
+        pass
+
+    @abstractmethod
+    def upload(self, table: str, data: list) -> bool:
+        pass
+
+    def run(self):
+        while not self._stop.is_set():
+            if not self._upload.length:
+                time.sleep(self.wait_seconds)
+                continue
+            if self._upload.length < self.upload_size:
+                time.sleep(self.wait_seconds)
+            status = True
+            data_info, seeds = {}, []
+            try:
+                for _ in range(self.upload_size):
+                    item = self._upload.pop()
+                    if not item:
+                        break
+                    seeds.append(item.seed)
+                    data = self.build(item)
+                    data_info.setdefault(item.table, []).append(data)
+                for table, datas in data_info.items():
+                    try:
+                        self.upload(table, datas)
+                    except Exception as e:
+                        logger.info(e)
+                        status = False
+            except Exception as e:
+                logger.info(e)
+                status = False
+            if not status:
+                for seed in seeds:
+                    seed.params.seed_status = "deal model: fail"
+            self._done.push(seeds)
+
+        logger.info("upload pipeline close!")
+
+

{cobweb-launcher-1.2.8 → cobweb-launcher-1.2.10/cobweb_launcher.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cobweb-launcher
-Version: 1.2.8
+Version: 1.2.10
 Summary: spider_hole
 Home-page: https://github.com/Juannie-PP/cobweb
 Author: Juannie-PP

{cobweb-launcher-1.2.8 → cobweb-launcher-1.2.10}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="cobweb-launcher",
-    version="1.2.8",
+    version="1.2.10",
     packages=find_packages(),
     url="https://github.com/Juannie-PP/cobweb",
     license="MIT",

cobweb-launcher-1.2.8/cobweb/pipelines/pipeline.py

@@ -1,60 +0,0 @@
-import time
-import threading
-
-from abc import ABC, abstractmethod
-from cobweb.base import BaseItem, Queue, logger
-
-
-class Pipeline(threading.Thread, ABC):
-
-    def __init__(
-            self,
-            stop: threading.Event,
-            pause: threading.Event,
-            upload: Queue, done: Queue,
-            upload_size: int,
-            wait_seconds: int
-    ):
-        super().__init__()
-        self._stop = stop
-        self._pause = pause
-        self._upload = upload
-        self._done = done
-
-        self.upload_size = upload_size
-        self.wait_seconds = wait_seconds
-
-    @abstractmethod
-    def build(self, item: BaseItem) -> dict:
-        pass
-
-    @abstractmethod
-    def upload(self, table: str, data: list) -> bool:
-        pass
-
-    def run(self):
-        while not self._stop.is_set():
-            status = self._upload.length < self.upload_size
-            if status:
-                time.sleep(self.wait_seconds)
-            data_info, seeds = {}, []
-            for _ in range(self.upload_size):
-                item = self._upload.pop()
-                if not item:
-                    break
-                data = self.build(item)
-                seeds.append(item.seed)
-                data_info.setdefault(item.table, []).append(data)
-            for table, datas in data_info.items():
-                try:
-                    self.upload(table, datas)
-                    status = True
-                except Exception as e:
-                    logger.info(e)
-                    status = False
-                if status:
-                    self._done.push(seeds)
-
-        logger.info("upload pipeline close!")
-
-