cobweb-launcher 0.1.15__tar.gz → 0.1.17__tar.gz

{cobweb-launcher-0.1.15 → cobweb-launcher-0.1.17}/PKG-INFO

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

{cobweb-launcher-0.1.15 → cobweb-launcher-0.1.17}/cobweb/__init__.py

@@ -3,6 +3,7 @@ from .task import Task
 from .log import log
 from .db.redis_db import RedisDB
 from .db.oss_db import OssDB
+from .constant import Setting
 
 from .equip.distributed.launcher import launcher
 from .equip.single.launcher import launcher as single_launcher

{cobweb-launcher-0.1.15 → cobweb-launcher-0.1.17}/cobweb/constant.py

@@ -22,4 +22,6 @@ class Setting:
     CHECK_LOCK_TIME = None
     DEAL_MODEL = None
     LAUNCHER_MODEL = None
-    SPIDER_RUN_TIME = None
+    SCHEDULER_WAIT_TIME = None
+    SCHEDULER_BLOCK_TIME = None
+    SPIDER_WAIT_TIME = None

cobweb-launcher-0.1.17/cobweb/db/storer/redis.py

@@ -0,0 +1,15 @@
+from cobweb import log, StorerInterface
+
+
+class Redis(StorerInterface):
+
+    def store(self, data_list):
+        try:
+            data_str = "\n".join(str(data) for data in data_list)
+            with open(self.table, "a") as fp:
+                fp.write(data_str)
+            log.info(f"save data, data length: {len(data_list)}")
+            return True
+        except Exception as e:
+            return False
+

cobweb-launcher-0.1.17/cobweb/distributed/launcher.py

@@ -0,0 +1,243 @@
+import time
+import threading
+from threading import Thread
+
+from .models import Scheduler, Spider, Storer
+from cobweb import log, Queue, DBItem, RedisDB
+from cobweb.setting import MODEL, RESET_SCORE, CHECK_LOCK_TIME
+from cobweb.utils import (
+    struct_queue_name as sqn,
+    restore_table_name as rtn,
+    parse_import_model as pim,
+)
+
+
+def check(stop, last, spider, scheduler, storer_list, ready_seed_length, spider_queue_length):
+    log.info("run check thread after 30 seconds...")
+    time.sleep(30)
+    spider_info = """
+------------------- check: {0} ------------------
+            running_spider_thread_num: {1}
+            redis_ready_seed_length:   {2}
+            redis_spider_seed_length:  {3}
+            memory_seed_queue_length:  {4}
+            storer_upload_queue_length_info: 
+                {5}
+-----------------------  end  -----------------------"""
+    while True:
+        status = "running"
+        running_spider_thread_num = spider.spider_in_progress.length
+        redis_ready_seed_length = ready_seed_length()
+        redis_spider_seed_length = spider_queue_length()
+        memory_seed_queue_length = scheduler.queue.length
+        storer_upload_queue_list = []
+        for storer in storer_list:
+            storer_upload_queue_list.append(
+                f"{storer.__class__.__name__} storer queue length: {storer.queue.length}"
+            )
+        if (
+                scheduler.stop and
+                # not redis_ready_seed_length and
+                not memory_seed_queue_length and
+                not running_spider_thread_num
+        ):
+            if not MODEL:
+                log.info("spider is done?")
+            last.set()
+            time.sleep(3)
+            storer_queue_empty = True
+            storer_upload_queue_list = []
+            for storer in storer_list:
+                if storer.queue.length:
+                    storer_queue_empty = False
+                storer_upload_queue_list.append(
+                    f"{storer.__class__.__name__} storer queue length: {storer.queue.length}"
+                )
+            if (
+                    storer_queue_empty and
+                    not redis_ready_seed_length and
+                    not redis_spider_seed_length
+            ):
+                if MODEL:
+                    log.info("waiting for push seeds...")
+                    status = "waiting"
+                    time.sleep(30)
+                else:
+                    log.info("spider done!")
+                    break
+
+            last.clear()
+
+        storer_upload_queue_length_info = "\n            ".join(
+            storer_upload_queue_list) if storer_upload_queue_list else "None"
+        log.info(spider_info.format(
+            status,
+            running_spider_thread_num,
+            redis_ready_seed_length,
+            redis_spider_seed_length,
+            memory_seed_queue_length,
+            storer_upload_queue_length_info
+        ))
+
+        time.sleep(3)
+    stop.set()
+
+
+def launcher(task):
+    """
+    任务启动装饰器
+    :param task: 任务配置信息
+    """
+    def decorator(func):
+        """
+        Item:
+            Textfile()
+            Loghub()
+            Console()
+        e.g.
+        task.fields = "a,b"
+        func(item, seed)
+            a = "a"
+            b = "b"
+            data = {"a": "a", "b": "b"}
+            yield item.Loghub(**data)
+            yield item.Loghub(a=a, b=b)
+        """
+        storer_list = []
+
+        # 程序结束事件
+        last = threading.Event()
+        # 停止采集事件
+        stop = threading.Event()
+
+        # 初始化redis信息
+        redis_db = RedisDB(
+            task.project, task.task_name, task.redis_info,
+            model=MODEL, cs_lct=CHECK_LOCK_TIME, rs_time=RESET_SCORE
+        )
+
+        log.info("初始化cobweb!")
+
+        seed_queue = Queue()
+
+        if task.scheduler_info is None:
+            task.scheduler_info = dict()
+
+        # 调度器动态继承
+        sql = task.scheduler_info.get("sql")
+        table = task.scheduler_info.get("table")
+        size = task.scheduler_info.get("size")
+        scheduler_config = task.scheduler_info.get("config")
+        scheduler_db = task.scheduler_info.get("db", "default")
+        DB, class_name = pim(scheduler_db, "scheduler")
+        # SchedulerDB, table, sql, length, size, config = task.scheduler_info
+        SchedulerTmp = type(class_name, (Scheduler, DB), {})
+
+        # 初始化调度器
+        scheduler = SchedulerTmp(
+            table=table, sql=sql, size=size, queue=seed_queue,
+            length=task.scheduler_queue_length, config=scheduler_config
+        )
+
+        # 初始化采集器
+        spider = Spider(seed_queue, task.max_retries)
+
+        # 解析存储器信息
+        storer_info_list = task.storer_info or []
+        if not isinstance(storer_info_list, list):
+            storer_info_list = [storer_info_list]
+
+        # new item
+        item = type("Item", (object,), {"redis_client": redis_db.client})()
+
+        for storer_info in storer_info_list:
+            storer_db = storer_info["db"]
+            fields = storer_info["fields"]
+            storer_table = storer_info.get("table", "console")
+            storer_config = storer_info.get("config")
+
+            StorerDB, class_name = pim(storer_db, "storer")
+            StorerTmp = type(class_name, (Storer, StorerDB), {})
+
+            db_name = class_name.lower()
+            if not getattr(item, db_name, None):
+                instance = type(db_name, (DBItem,), {})
+                setattr(item, db_name, instance)
+
+            storer_item_instance = getattr(item, db_name)
+            storer_item_instance.init_item(storer_table, fields)
+
+            storer_queue = sqn(db_name, storer_table)
+            queue = getattr(storer_item_instance, storer_queue)
+            # 初始话存储器
+            table_name = rtn(table_name=storer_table)
+            storer = StorerTmp(
+                table=table_name, fields=fields,
+                length=task.storer_queue_length,
+                queue=queue, config=storer_config
+            )
+            storer_list.append(storer)
+
+        Thread(target=redis_db.check_spider_queue, args=(stop, len(storer_list))).start()
+        Thread(target=redis_db.set_heartbeat, args=(stop,)).start()
+
+        # 推送初始种子
+        # seeds = start_seeds(task.start_seed)
+        redis_db.add_seed(task.seeds)
+        # 启动调度器, 调度至redis队列
+        Thread(
+            # name="xxxx_schedule_seeds",
+            target=scheduler.schedule_seed,
+            args=(
+                redis_db.ready_seed_length,
+                redis_db.get_scheduler_lock,
+                redis_db.add_seed
+            )
+        ).start()
+
+        # 启动调度器, 调度任务队列
+        Thread(
+            # name="xxxx_schedule_task",
+            target=scheduler.schedule_task,
+            args=(
+                stop, redis_db.get_seed,
+                redis_db.ready_seed_length
+            )
+        ).start()
+
+        # 启动采集器
+        for index in range(task.spider_num):
+            Thread(
+                # name=f"xxxx_spider_task:{index}",
+                target=spider.spider_task,
+                args=(
+                    stop, func, item,
+                    redis_db.del_seed
+                )
+            ).start()
+
+        # 启动存储器
+        for storer in storer_list:
+            Thread(
+                # name=f"xxxx_store_task:{storer.table}",
+                target=storer.store_task,
+                args=(
+                    stop, last,
+                    redis_db.reset_seed,
+                    redis_db.set_storer
+                )
+            ).start()
+
+        Thread(
+            # name="check_spider",
+            target=check,
+            args=(
+                stop, last, spider,
+                scheduler, storer_list,
+                redis_db.ready_seed_length,
+                redis_db.spider_queue_length,
+            )
+        ).start()
+
+    return decorator
+

cobweb-launcher-0.1.17/cobweb/distributed/models.py

@@ -0,0 +1,143 @@
+import time
+from hashlib import md5
+from cobweb import log, Queue, Seed
+from cobweb.utils import issubclass_cobweb_inf
+
+# from pympler import asizeof
+
+
+class Scheduler:
+
+    def schedule_seed(self, ready_seed_length, get_scheduler_lock, add_seed):
+
+        inf_name = "SchedulerInterface"
+        if not issubclass_cobweb_inf(self.__class__, inf_name):
+            raise Exception("not have schedule function!")
+
+        if self.__class__.__name__ == "Default":
+            self.stop = True
+            return None
+
+        while not self.stop:
+            length = ready_seed_length()
+            if length > self.size:
+                time.sleep(15)
+
+            elif get_scheduler_lock():
+                seeds = self.schedule()
+                add_seed(seeds)
+
+        log.info(f"close thread: schedule_seed")
+
+    def schedule_task(self, stop, get_seed, ready_seed_length):
+        time.sleep(3)
+        while not stop.is_set():
+
+            if not ready_seed_length():
+                time.sleep(5)
+                continue
+
+            if self.queue.length >= self.length:
+                time.sleep(3)
+                continue
+
+            seeds = get_seed(self.length)
+            self.queue.push(seeds)
+        log.info(f"close thread: schedule_task")
+
+
+class Spider:
+
+    def __init__(self, queue, max_retries=5):
+        self.spider_in_progress = Queue()
+        self.max_retries = max_retries
+        self.queue = queue
+
+    def spider_task(self, stop, func, item, del_seed):
+        while not stop.is_set():
+            seed = self.queue.pop()
+            if not seed:
+                time.sleep(3)
+                continue
+            elif seed._retry >= self.max_retries:
+                del_seed(seed, spider_status=False)
+                continue
+            try:
+                self.spider_in_progress.push(1, direct_insertion=True)
+                # log.info("spider seed: " + str(seed))
+                ret_count = 0
+                status = None
+                store_queue = None
+                store_data = list()
+                for it in func(item, seed):
+                    ret_count += 1
+                    if getattr(it, "table_name", None):
+                        if not store_queue:
+                            store_queue = it.queue()
+                        store_data.append(it.struct_data)
+                    elif isinstance(it, Seed):
+                        self.queue.push(it)
+                    elif any(isinstance(it, t) for t in (list, tuple)):
+                        self.queue.push([s if isinstance(s, Seed) else Seed(s) for s in it])
+                    elif isinstance(it, bool):
+                        status = it
+
+                if store_queue and store_data:
+                    store_data.append(seed)
+                    store_queue.push(store_data)
+
+                if status:
+                    del_seed(seed, spider_status=True)
+                elif not ret_count or status is False:
+                    seed._retry += 1
+                    self.queue.push(seed)
+
+            except Exception as e:
+                seed._retry += 1
+                self.queue.push(seed)
+                log.info(f"{str(seed)} -> {str(e)}")
+            finally:
+                self.spider_in_progress.pop()
+        log.info(f"close thread: spider")
+
+
+class Storer:
+
+    def store_task(self, stop, last, reset_seed, set_storer):
+
+        inf_name = "StorerInterface"
+        if not issubclass_cobweb_inf(self.__class__, inf_name):
+            return None
+
+        if not getattr(self, "store", None):
+            raise Exception("not have store function!")
+
+        storer_name = self.__class__.__name__ + self.table
+        store_key_id = md5(storer_name.encode()).hexdigest()
+
+        while not stop.is_set():
+
+            if last.is_set() or self.queue.length >= self.length:
+                seeds, data_list = [], []
+
+                while True:
+                    data = self.queue.pop()
+                    if not data:
+                        break
+                    if isinstance(data, Seed):
+                        seeds.append(data)
+                        if len(data_list) >= self.length:
+                            break
+                        continue
+                    data_list.append(data)
+
+                if data_list:
+                    if self.store(data_list):
+                        set_storer(store_key_id, seeds)
+                    else:
+                        reset_seed(seeds)
+                    continue
+
+            time.sleep(3)
+
+        log.info(f"close thread: {storer_name}")

cobweb-launcher-0.1.17/cobweb/equip/single/launcher.py

@@ -0,0 +1,202 @@
+import time
+import threading
+
+from .. import log, sqn, rtn, pim
+from .. import Queue, DBItem, RedisDB, Setting, OssDB
+from .models import Scheduler, Spider, Storer
+
+
+def check(stop, last, spider, scheduler, storer, ready_seed_length, spider_queue_length):
+    log.info("run check thread after 30 seconds...")
+    time.sleep(30)
+    spider_info = """
+------------------- check: {0} ------------------
+            redis_spider_seed_length:  {1}
+            redis_ready_seed_length:   {2}
+            running_spider_thread_num: {3}
+            memory_seed_queue_length:  {4}
+            storer_queue_length_info:  {5}
+-----------------------  end  -----------------------"""
+    while True:
+        status = "running"
+        running_spider_thread_num = spider.spider_in_progress.length
+        redis_ready_seed_length = ready_seed_length()
+        redis_spider_seed_length = spider_queue_length()
+        memory_seed_queue_length = scheduler.queue.length
+        storer_upload_queue_length = storer.queue.length if storer else None
+        if (
+                scheduler.stop and
+                not memory_seed_queue_length and
+                not running_spider_thread_num
+        ):
+            if not Setting.LAUNCHER_MODEL:
+                log.info("spider is done?")
+            last.set()
+            time.sleep(3)
+            storer_queue_empty = True
+            if storer and storer.queue.length:
+                storer_queue_empty = False
+            storer_upload_queue_length = storer.queue.length if storer else None
+            if (
+                    storer_queue_empty and
+                    not redis_ready_seed_length and
+                    not redis_spider_seed_length
+            ):
+                if Setting.LAUNCHER_MODEL:
+                    log.info("waiting for push seeds...")
+                    status = "waiting"
+                    time.sleep(30)
+                else:
+                    log.info("spider done!")
+                    break
+
+            last.clear()
+
+        log.info(spider_info.format(
+            status,
+            redis_spider_seed_length,
+            redis_ready_seed_length,
+            running_spider_thread_num,
+            memory_seed_queue_length,
+            storer_upload_queue_length
+        ))
+
+        time.sleep(3)
+    stop.set()
+
+
+def launcher(task):
+    """
+    任务启动装饰器
+    :param task: 任务配置信息
+    """
+    def decorator(func):
+        # 程序结束事件
+        last = threading.Event()
+        # 停止采集事件
+        stop = threading.Event()
+
+        # 初始化redis信息
+        redis_db = RedisDB(task.project, task.task_name, task.redis_info)
+
+        # new item
+        item = type("Item", (object,), {"redis_client": redis_db.client})()
+        if task.oss_config:
+            item.oss = OssDB(**task.oss_config)
+
+        log.info("初始化cobweb!")
+
+        seed_queue = Queue()
+
+        scheduler_info = task.scheduler_info or dict()
+        # 调度器动态继承
+        sql = scheduler_info.get("sql")
+        table = scheduler_info.get("table")
+        size = scheduler_info.get("size")
+        scheduler_config = scheduler_info.get("config")
+        scheduler_db = scheduler_info.get("db", "default")
+        DB, class_name = pim(scheduler_db, "scheduler")
+        # SchedulerDB, table, sql, length, size, config = task.scheduler_info
+        SchedulerTmp = type(class_name, (Scheduler, DB), {})
+        # 初始化调度器
+        scheduler = SchedulerTmp(
+            table=table, sql=sql, size=size, queue=seed_queue,
+            length=task.scheduler_queue_length, config=scheduler_config
+        )
+
+        storer = None
+        storer_info = task.storer_info or dict()
+
+        if storer_info:
+            storer_db = storer_info["db"]
+            fields = storer_info["fields"]
+            storer_table = storer_info.get("table", "console")
+            storer_config = storer_info.get("config")
+
+            StorerDB, class_name = pim(storer_db, "storer")
+            StorerTmp = type(class_name, (Storer, StorerDB), {})
+
+            db_name = class_name.lower()
+            if not getattr(item, db_name, None):
+                instance = type(db_name, (DBItem,), {})
+                setattr(item, db_name, instance)
+
+            storer_item_instance = getattr(item, db_name)
+            storer_item_instance.init_item(storer_table, fields)
+
+            storer_queue = sqn(db_name, storer_table)
+            queue = getattr(storer_item_instance, storer_queue)
+            # 初始话存储器
+            table_name = rtn(table_name=storer_table)
+            storer = StorerTmp(
+                table=table_name, fields=fields,
+                length=task.storer_queue_length,
+                queue=queue, config=storer_config
+            )
+
+        # 初始化采集器
+        spider = Spider(seed_queue, storer and True, task.max_retries)
+
+        threading.Thread(target=redis_db.check_spider_queue, args=(stop, 0)).start()
+        threading.Thread(target=redis_db.set_heartbeat, args=(stop,)).start()
+
+        # 推送初始种子
+        # seeds = start_seeds(task.start_seed)
+        redis_db.add_seed(task.seeds)
+        # 启动调度器, 调度至redis队列
+        threading.Thread(
+            # name="xxxx_schedule_seeds",
+            target=scheduler.schedule_seed,
+            args=(
+                redis_db.ready_seed_length,
+                redis_db.get_scheduler_lock,
+                redis_db.add_seed
+            )
+        ).start()
+
+        # 启动调度器, 调度任务队列
+        threading.Thread(
+            # name="xxxx_schedule_task",
+            target=scheduler.schedule_task,
+            args=(
+                stop, redis_db.get_seed,
+                redis_db.ready_seed_length
+            )
+        ).start()
+
+        # 启动采集器
+        for index in range(task.spider_num):
+            threading.Thread(
+                # name=f"xxxx_spider_task:{index}",
+                target=spider.spider_task,
+                args=(
+                    stop, func, item,
+                    redis_db.del_seed
+                )
+            ).start()
+
+        # 启动存储器
+        if storer:
+            threading.Thread(
+                # name=f"xxxx_store_task:{storer.table}",
+                target=storer.store_task,
+                args=(
+                    stop, last,
+                    redis_db.reset_seed,
+                    redis_db.del_seed
+                )
+            ).start()
+
+        threading.Thread(
+            # name="check_spider",
+            target=check,
+            args=(
+                stop, last, spider,
+                scheduler, storer,
+                redis_db.ready_seed_length,
+                redis_db.spider_queue_length,
+            )
+        ).start()
+
+    return decorator
+