datacom-device-api 0.1.5__tar.gz

datacom_device_api-0.1.5/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: datacom-device-api
+Version: 0.1.5
+Summary: Add your description here
+Requires-Python: >=3.14
+Description-Content-Type: text/markdown
+Requires-Dist: dataclasses>=0.8
+Requires-Dist: openpyxl>=3.1.5
+Requires-Dist: ops>=3.8.0
+Requires-Dist: pymysql>=1.2.0
+Requires-Dist: rich>=15.0.0

datacom_device_api-0.1.5/pyproject.toml

@@ -0,0 +1,13 @@
+[project]
+name = "datacom-device-api"
+version = "0.1.5"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.14"
+dependencies = [
+    "dataclasses>=0.8",
+    "openpyxl>=3.1.5",
+    "ops>=3.8.0",
+    "pymysql>=1.2.0",
+    "rich>=15.0.0",
+]

datacom_device_api-0.1.5/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

datacom_device_api-0.1.5/src/datacom_device_api/data/table.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass
+from typing import Any, Generic, TypeVarTuple, Unpack
+
+Ts = TypeVarTuple('Ts')
+
+@dataclass
+class Table(Generic[Unpack[Ts]]):
+    table_name: str
+    headers: list[str]
+    fragment_refrence_size: int
+    total_fragment: int
+    total_row: int
+    fragments: list[TableFragment[Unpack[Ts]]]
+    def __post_init__(self):
+        for fragment in self.fragments:
+            if fragment.fragment_offset < 0 or fragment.fragment_size <= 0:
+                raise ValueError('Invalid fragment offset or size')
+            if len(fragment.rows) > fragment.fragment_size:
+                raise ValueError('Invalid fragment size or row count')
+
+@dataclass
+class TableFragment(Generic[Unpack[Ts]]):
+    fragment_offset: int
+    fragment_size: int
+    rows: list[tuple[Unpack[Ts]]]

datacom_device_api-0.1.5/src/datacom_device_api/operation/table_operation.py

@@ -0,0 +1,89 @@
+
+from typing import Any, Callable, Generic, Optional, TypeVar, TypeVarTuple, Union, Unpack, overload
+
+from datacom_device_api.data.table import Table, TableFragment
+
+Ts = TypeVarTuple('Ts')
+
+class TableOperation(Generic[Unpack[Ts]]):
+    def __init__(
+            self, 
+            table_name: str, 
+            headers: list[str], 
+            fragment_refrence_size: int = 1000
+        ) -> None:
+        self.__table = Table[Unpack[Ts]](
+            table_name=table_name,
+            headers=headers,
+            fragment_refrence_size=fragment_refrence_size,
+            total_fragment=0,
+            total_row=0,
+            fragments=[],
+        )
+    
+    def add_fragment(self, fragment: TableFragment[Unpack[Ts]]) -> None:
+        if fragment.fragment_size > self.__table.fragment_refrence_size:
+            raise ValueError('fragment size is too large')
+        
+        self.__table.fragments.append(fragment)
+        self.__table.total_fragment += 1
+        self.__table.total_row += len(fragment.rows)
+        
+    @property
+    def fragment_refrence_size(self) -> int:
+        return self.__table.fragment_refrence_size
+    
+    def get_row_by_row_index(self, row_index: int) -> Optional[tuple[Unpack[Ts]]]:
+        if row_index >= self.__table.total_row:
+            raise ValueError('row index is out of range')
+        
+        self.__table.fragments.sort(key=lambda fragment: fragment.fragment_offset)
+        
+        fragment_index = row_index // self.__table.fragment_refrence_size
+        fragment = self.__table.fragments[fragment_index]
+        
+        return fragment.rows[row_index % fragment.fragment_size]
+    
+    @overload
+    def get_row_by_header(self, header: str, value: Any) -> list[tuple[*Ts]]: ...
+    @overload
+    def get_row_by_header(self, header: int, value: Any) -> list[tuple[*Ts]]: ...
+    
+    def get_row_by_header(self, header: Any, value: Any) -> list[tuple[*Ts]]:
+        if isinstance(header, int): 
+            header_index = header
+        elif isinstance(header, str):
+            if header not in self.__table.headers:
+                raise ValueError('header name not found')
+            header_index = self.__table.headers.index(header)
+        else:
+            raise ValueError('header must be int or str')
+        
+        if type(value) not in [*Ts]:
+            raise ValueError('value type of value')
+        
+        result = []        
+        for fragment in self.__table.fragments:
+            for row in fragment.rows:
+                if row[header_index] == value:
+                    result.append(row)
+        
+        return result
+    
+    @property
+    def table(self) -> Table[Unpack[Ts]]:
+        return self.__table
+
+    @property
+    def headers(self) -> list[str]:
+        return self.__table.headers
+    
+    def for_each_row_by_fragment_index(self, fragment_index: int, callback: Callable[[tuple[*Ts]], None]):
+        fragment = self.__table.fragments[fragment_index]
+        for row in fragment.rows:
+            callback(row)
+            
+    def for_each_all(self, callback: Callable[[tuple[*Ts]], None]):
+        for frament in self.__table.fragments:
+            for row in frament.rows:
+                callback(row)

datacom_device_api-0.1.5/src/datacom_device_api/operation/table_read_from_xlsx.py

@@ -0,0 +1,103 @@
+from typing import Any, Callable, Generic, TypeVarTuple
+
+from openpyxl import Workbook, load_workbook
+
+from datacom_device_api.data.table import Table, TableFragment
+from datacom_device_api.operation.table_operation import TableOperation
+
+
+Ts = TypeVarTuple('Ts')
+
+class TableReadFromXlsx(Generic[*Ts]):
+    def __init__(self, xlsx_file_path: str) -> None:
+        try:
+            self.__workbook = load_workbook(xlsx_file_path, read_only=True)
+        except FileNotFoundError:
+            raise FileNotFoundError(f"{xlsx_file_path} is not found.")
+        except Exception as e:
+            raise e
+    
+    @property
+    def get_sheet_names(self) -> list[str]:
+        return self.__workbook.sheetnames
+        
+    
+    def _generate_table_fragment(self, worksheet, min_row: int, max_row: int, convert_func: Callable[[str, Any], Any], headers: list[str]) -> TableFragment[*Ts]:
+        if not hasattr(worksheet, 'iter_rows'):
+            raise AttributeError("worksheet does not have iter_rows method.")
+        
+        row_datas: list[tuple] = []
+        for row_data in worksheet.iter_rows(min_row=min_row, max_row=max_row):
+            tmp = tuple(convert_func(headers[i], row_data[i].value ) for i in range(len(headers)))
+            print(tmp)
+            row_datas.append(tmp)
+        
+        if min_row == 1 or min_row == 2:
+            fragment_offset = 0
+        else:
+            fragment_offset = min_row - 1
+        
+        return TableFragment[*Ts](
+            rows=row_datas,
+            fragment_offset=fragment_offset,
+            fragment_size=len(row_datas),
+        )
+        
+    def generate_table_data(self, sheet_name: str, convert_func: Callable[[str, Any], Any]) -> Table[*Ts]:
+        if sheet_name not in self.__workbook.sheetnames:
+            raise ValueError(f"{sheet_name} is not found.")
+        
+        worksheet = self.__workbook[sheet_name]
+        total_rows = worksheet.max_row - 1
+        
+        headers = []
+        for row_data in worksheet.iter_rows(min_row=1, max_row=1): 
+            headers.extend([item.value for item in row_data])
+
+        table = TableOperation[*Ts](
+            table_name=sheet_name,
+            headers=headers,
+        )
+    
+        fragments_count = -(-total_rows // table.fragment_refrence_size)
+        fragments_size = []
+        
+        remaining_count = total_rows
+        for _ in range(fragments_count):
+            remaining_count -= (table.fragment_refrence_size)
+            if remaining_count < 0:
+                fragments_size.append(remaining_count + table.fragment_refrence_size)
+            else:
+                fragments_size.append(table.fragment_refrence_size)
+        
+        offset = 2
+        for size in fragments_size:
+            fragment = self._generate_table_fragment(
+                worksheet=worksheet,
+                min_row=offset,
+                max_row=offset + size - 1,
+                convert_func=convert_func,
+                headers=headers,
+            )
+            
+            table.add_fragment(fragment)
+            offset += size
+        
+        return table.table
+
+def _convert_func(header: str, value: Any) -> Any:
+    if header == 'USER_NAME':
+        return str(value)
+    return value
+
+if __name__ == "__main__":
+    xlsx_file = "C:\\Users\\24168\\Projects\\Datacom_tools\\tables\\test.xlsx"
+    sheet_name = "Sheet1"
+    xlsx = TableReadFromXlsx(xlsx_file)
+    xlsx.generate_table_data(sheet_name, _convert_func)
+
+    # workbook = load_workbook(xlsx_file, read_only=True)
+    # sheet = workbook[sheet_name]
+    
+    # for row_data in sheet.iter_rows(min_row=1, max_row=10): 
+    #     print(row_data)

datacom_device_api-0.1.5/src/datacom_device_api/operation/table_save_to_db.py

@@ -0,0 +1,20 @@
+from typing import Generic, TypeVarTuple, Unpack
+from pymysql.cursors import Cursor
+from pymysql.connections import Connection
+from datacom_device_api.data.table import Table
+
+Ts = TypeVarTuple('Ts')
+
+class TableSaveToDB(Generic[Unpack[Ts]]):
+    def __init__(self, connect: Connection) -> None:
+        self.__cursor = connect.cursor()
+        self.__conn = connect
+    def save_to_db(self, db_table_name: str, table: Table[Unpack[Ts]]) -> None:
+        sql = f'INSERT INTO {db_table_name} VALUES ({",".join(["%s"] * len(table.headers))})'
+        
+        for fragment in table.fragments:
+            values_list = fragment.rows
+            self.__cursor.executemany(sql, values_list)
+            self.__conn.commit()
+        print(table)
+        

datacom_device_api-0.1.5/src/datacom_device_api/process/terminal_process.py

@@ -0,0 +1,244 @@
+"""多进程任务执行与进度展示封装。
+
+提供 TerminalProcess 类，封装 ProcessPoolExecutor 和 Rich Progress，
+支持任务提交、进度监听、异常处理、上下文管理器等能力。
+"""
+
+from concurrent.futures import Future, ProcessPoolExecutor
+from dataclasses import dataclass
+from functools import partial
+from multiprocessing import Manager
+import os
+import queue
+import threading
+from typing import Any, Callable, Optional
+from uuid import UUID
+import uuid
+
+from rich.progress import Progress, TaskID
+
+REPORT_FUNC_TYPE = Callable[[float, float, str], None]
+EXECUTE_FUNC_TYPE = Callable[[REPORT_FUNC_TYPE], Any]
+
+
+@dataclass
+class ProgressMessage:
+    """进程间传递的进度消息。
+
+    Attributes:
+        type: 消息类型，取值为 'progress' / 'done' / 'error'
+        task_id: 任务唯一标识
+        message: 进度描述文本
+        progress: 当前进度值
+        total: 总进度值
+    """
+
+    type: str
+    task_id: UUID
+    message: str
+    progress: float
+    total: float
+
+
+__PROGRESS_UPDATE_FN_TYPE = Callable[[ProgressMessage], None]
+
+
+class TerminalProcess:
+    """多进程任务执行器，支持进度展示与监听器机制。
+
+    每个实例拥有独立的进程池、进度队列和进度条，
+    多实例之间状态互不干扰。
+
+    用法:
+        with TerminalProcess() as process:
+            process.submit(task1)
+            process.submit(task2)
+    """
+
+    def __init__(self, max_workers: Optional[int] = None):
+        """初始化终端进程执行器。
+
+        Args:
+            max_workers: 进程池最大并发数，默认为 CPU 核心数
+        """
+        self.__pool = ProcessPoolExecutor(max_workers=max_workers or os.cpu_count())
+        self.__manager = Manager()
+        self.__progress_queue: queue.Queue[ProgressMessage] = self.__manager.Queue()
+
+        self.__task_future_count = 0
+        self.__tasks: dict[UUID, Future[Any]] = {}
+        self.__task_update_lisening_fn_dict: dict[UUID, list[__PROGRESS_UPDATE_FN_TYPE]] = {}
+
+        self.__progress_thread_stop_event = threading.Event()
+        self.__progress_tasks: dict[UUID, TaskID] = {}
+        self.__progress = Progress()
+        self.__progress_thread: Optional[threading.Thread] = None
+        self.__progress_main_task: TaskID = self.__progress.add_task(
+            'main', total=self.__task_future_count
+        )
+
+    def trigger_progress(self, task_id: UUID, message: ProgressMessage) -> None:
+        """触发指定任务的所有进度监听器。
+
+        若 task_id 未注册监听器则静默跳过。
+        """
+        listening_fns = self.__task_update_lisening_fn_dict.get(task_id, [])
+        for fn in listening_fns:
+            fn(message)
+
+    def listen_progress(self, task_id: UUID, fn: __PROGRESS_UPDATE_FN_TYPE) -> None:
+        """为指定任务注册一个进度监听器。
+
+        同一任务可注册多个监听器，按注册顺序依次调用。
+        """
+        self.__task_update_lisening_fn_dict.setdefault(task_id, []).append(fn)
+
+    def update_progress(self, message: ProgressMessage) -> None:
+        """更新内置进度条的显示。
+
+        任务完成时自动推进主进度条计数。
+        若消息对应的任务不存在则静默跳过。
+        """
+        task_id = self.__progress_tasks.get(message.task_id)
+        if task_id is None:
+            return
+        self.__progress.update(task_id, completed=message.progress, total=message.total)
+        self.__progress.refresh()
+
+        if message.type == 'done':
+            self.__progress.update(self.__progress_main_task, advance=1)
+
+    @staticmethod
+    def _warrper_fn(
+        fn: EXECUTE_FUNC_TYPE,
+        progress_queue: queue.Queue[ProgressMessage],
+        task_id: UUID,
+        *args,
+        **kwargs,
+    ) -> Any:
+        """任务包装函数，在子进程中执行。
+
+        封装进度上报、完成标记和异常处理，通过队列将进度消息
+        发送回主进程。任务正常结束或异常均会发送对应类型的消息。
+        """
+        done_send = False
+
+        def report(progress: float, total: float, message: str, msg_type: str = 'progress'):
+            nonlocal done_send
+
+            if done_send:
+                return
+
+            current_type = msg_type
+            if progress >= total:
+                current_type = 'done'
+                done_send = True
+
+            progress_queue.put(ProgressMessage(
+                type=current_type,
+                task_id=task_id,
+                message=message,
+                progress=progress,
+                total=total,
+            ))
+
+        try:
+            result = fn(report, *args, **kwargs)
+            if not done_send:
+                report(100, 100, 'done', msg_type='done')
+            return result
+        except Exception as e:
+            progress_queue.put(ProgressMessage(
+                type='error',
+                task_id=task_id,
+                message=str(e),
+                progress=0,
+                total=0,
+            ))
+            raise
+
+    def run_display_progress(self) -> None:
+        """进度消费线程主循环。
+
+        持续从队列中取出进度消息，依次更新内置进度条并触发监听器。
+        使用带超时的轮询避免阻塞，确保停止事件触发后能及时退出。
+        """
+        while not self.__progress_thread_stop_event.is_set() or not self.__progress_queue.empty():
+            try:
+                message = self.__progress_queue.get(timeout=0.1)
+                self.update_progress(message)
+                self.trigger_progress(message.task_id, message)
+            except queue.Empty:
+                if (
+                    self.__progress_thread_stop_event.is_set()
+                    and all(futrue.done() for futrue in self.__tasks.values())
+                ):
+                    break
+
+    def submit(
+        self,
+        fn: EXECUTE_FUNC_TYPE,
+        task_name: Optional[str] = None,
+        *args,
+        **kwargs,
+    ) -> Optional[Future[Any]]:
+        """提交一个任务到进程池执行。
+
+        Args:
+            fn: 任务函数，第一个参数为 report 回调，签名为 (progress, total, message)
+            task_name: 进度条显示的任务名，默认为 task_{uuid}
+            *args: 传递给任务函数的位置参数
+            **kwargs: 传递给任务函数的关键字参数
+
+        Returns:
+            任务的 Future 对象，可用于获取结果或异常
+        """
+        task_id = uuid.uuid4()
+        task_fn = partial(TerminalProcess._warrper_fn, fn, self.__progress_queue, task_id)
+        future = self.__pool.submit(task_fn, *args, **kwargs)
+        self.__tasks[task_id] = future
+        self.__task_update_lisening_fn_dict[task_id] = []
+
+        self.__task_future_count += 1
+        progress_task = self.__progress.add_task(task_name or f'task_{task_id}', total=1)
+        self.__progress_tasks[task_id] = progress_task
+        self.__progress.refresh()
+
+        self.__progress.update(self.__progress_main_task, total=self.__task_future_count)
+        return future
+
+    def start(self) -> None:
+        """启动进度条显示与消息消费线程。"""
+        self.__progress.start()
+        self.__progress_thread_stop_event.clear()
+        self.__progress_thread = threading.Thread(target=self.run_display_progress)
+        self.__progress_thread.start()
+
+    def shutdown(self) -> None:
+        """停止任务执行并释放资源。
+
+        按顺序执行：等待所有任务完成 → 消费完剩余进度消息 →
+        关闭进程池 → 停止进度条。确保不会因消息未消费完而丢失进度。
+        """
+        for future in self.__tasks.values():
+            try:
+                future.result()
+            except Exception:
+                pass
+
+        self.__progress_thread_stop_event.set()
+        if self.__progress_thread is not None:
+            self.__progress_thread.join()
+            self.__progress_thread = None
+
+        self.__pool.shutdown(wait=True)
+        self.__progress.stop()
+
+    def __enter__(self) -> 'TerminalProcess':
+        """进入上下文管理器时自动启动。"""
+        self.start()
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
+        """退出上下文管理器时自动释放资源。"""
+        self.shutdown()

datacom_device_api-0.1.5/src/datacom_device_api/utils/directory_util.py

@@ -0,0 +1,13 @@
+import os
+
+def get_project_root_dir() -> str:
+    return os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+def get_work_dir() -> str:
+    return os.getcwd()
+
+def test():
+    print(get_project_root_dir())
+    print(get_work_dir())
+
+if __name__ == '__main__':
+    test()

datacom_device_api-0.1.5/src/datacom_device_api/utils/logger.py

@@ -0,0 +1,14 @@
+import logging
+class Logger(object):
+    def __init__(self, name: str) -> None:
+        self.__logger = logging.getLogger(name)
+        self.__logger.setLevel(logging.INFO)
+        formatter = logging.Formatter('[%(asctime)s-%(levelname)s](%(name)s): %(message)s')
+
+        consolo_handler = logging.StreamHandler()
+        consolo_handler.setLevel(logging.INFO)
+        consolo_handler.setFormatter(formatter)
+        self.__logger.addHandler(consolo_handler)
+        
+    def __getattr__(self, name):
+        return getattr(self.__logger, name)

datacom_device_api-0.1.5/src/datacom_device_api.egg-info/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: datacom-device-api
+Version: 0.1.5
+Summary: Add your description here
+Requires-Python: >=3.14
+Description-Content-Type: text/markdown
+Requires-Dist: dataclasses>=0.8
+Requires-Dist: openpyxl>=3.1.5
+Requires-Dist: ops>=3.8.0
+Requires-Dist: pymysql>=1.2.0
+Requires-Dist: rich>=15.0.0

datacom_device_api-0.1.5/src/datacom_device_api.egg-info/SOURCES.txt

@@ -0,0 +1,20 @@
+README.md
+pyproject.toml
+src/datacom_device_api/__init__.py
+src/datacom_device_api.egg-info/PKG-INFO
+src/datacom_device_api.egg-info/SOURCES.txt
+src/datacom_device_api.egg-info/dependency_links.txt
+src/datacom_device_api.egg-info/requires.txt
+src/datacom_device_api.egg-info/top_level.txt
+src/datacom_device_api/data/__init__.py
+src/datacom_device_api/data/table.py
+src/datacom_device_api/operation/__init__.py
+src/datacom_device_api/operation/table_operation.py
+src/datacom_device_api/operation/table_read_from_xlsx.py
+src/datacom_device_api/operation/table_save_to_db.py
+src/datacom_device_api/process/__init__.py
+src/datacom_device_api/process/terminal_process.py
+src/datacom_device_api/utils/__init__.py
+src/datacom_device_api/utils/directory_util.py
+src/datacom_device_api/utils/logger.py
+test/test_table_matching.py

datacom_device_api-0.1.5/src/datacom_device_api.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

datacom_device_api-0.1.5/src/datacom_device_api.egg-info/requires.txt

@@ -0,0 +1,5 @@
+dataclasses>=0.8
+openpyxl>=3.1.5
+ops>=3.8.0
+pymysql>=1.2.0
+rich>=15.0.0

datacom_device_api-0.1.5/src/datacom_device_api.egg-info/top_level.txt

@@ -0,0 +1 @@
+datacom_device_api

datacom_device_api-0.1.5/test/test_table_matching.py

@@ -0,0 +1,270 @@
+import re
+from datacom_device_api.operation.table_matching import StringTableMatching
+
+CPU_OUTPUT = """CPU Usage Stat. Cycle: 60 (Second)
+CPU Usage            : 11% Max: 100%
+CPU Usage Stat. Time : 2026-07-26  22:27:23 
+CPU utilization for five seconds: 11%: one minute: 8%: five minutes: 4%.
+
+TaskName             CPU  Runtime(CPU Tick High/Tick Low)  Task Explanation
+BOX                   0%         0/ 676cb3a       BOX Output                   
+_TIL                  0%         0/       0       Infinite loop event task     
+VCLK                  0%         0/ 2b0c04b                                    
+TICK                  0%         0/ 766bb97                                    
+co0                   0%         0/  3fee5e       co0 Line user's task         
+U0                    0%         0/       0       U0   user command process
+                                                  task                         
+RTMR                  0%         0/ 13842b6       RTMR                         
+IPCR                  0%         0/       0       IPCR                         
+VPR                   0%         0/       0       VPR                          
+VPS                   0%         0/       0       VPS                          
+EDBG                  0%         0/       0       EDBG                         
+ECM                   0%         0/       0       ECM                          
+DEV                   0%         0/       0       DEV  Device                  
+FCAT                  0%         0/  187e3a       FCAT FECD task for catch
+                                                  packet                       
+FOAM                  0%         0/       0       FOAM                         
+FTS                   0%         0/       0       FTS                          
+IPCQ                  0%         0/ 557dfd0       IPCQIPC task for single queue
+IPCK                  0%         0/       0       IPCKIPC task for ack message 
+VP                    0%         0/   1b28d       VP  Virtual path task        
+RPCQ                  0%         0/  ca3abd       RPCQRemote procedure call    
+VFSD                  0%         0/       0       VFSDVFS flash task for delete
+                                                  file block                   
+VFS                   0%         0/       0       VFS Virtual file system      
+VMON                  0%         0/       0       VMONSystem monitor           
+VMSH                  0%         0/       0       VMSH                         
+PAT                   0%         0/       0       PAT                          
+HACK                  0%         0/       0       HACKtask for HA ACK          
+CLKI                  0%         0/       0       CLKI                         
+STND                  0%         0/       0       STNDStandby task             
+INFO                  0%         0/   f714a       INFOInformation center       
+SAPP                  0%         0/  724205       SAPP                         
+NQAC                  0%         0/       0       NQAC                         
+NQAS                  0%         0/       0       NQAS                         
+ALM                   0%         0/   36bdb       ALM  Alarm                   
+ENSP                  0%         0/ 11ba520       ENSP                         
+DEVA                  0%         0/       0       DEVA Device assistant        
+SRM                   0%         0/ 4fae8c5       SRM  System Resource Manage  
+METH                  0%         0/  2cab20       METH                         
+FIB6                  0%         0/       0       FIB6IPv6 FIB                 
+BFD                   0%         0/ 1e08781       BFD Bidirection Forwarding
+                                                  Detect                       
+TNLM                  0%         0/ 1e927d2       TNLM                         
+LSPA                  0%         0/       0       LSPA                         
+L2V                   0%         0/       0       L2V                          
+MACL                  0%         0/  7ff9c0       MACL                         
+CCTL                  0%         0/       0       CCTLBulk stat connect control
+TCTL                  0%         0/       0       TCTLBulk stat transmit
+                                                  control                      
+NAP                   0%         0/       0       NAP                          
+TRAF                  0%         0/  55de90       TRAFTraffic Statistics       
+SLAG                  0%         0/       0       SLAG                         
+ITSK                  0%         0/  3c398f       ITSKIPOS common task         
+SNPG                  0%         0/  2a2f7a       SNPG Multicast Snooping      
+MCSW                  0%         0/  2ee900       MCSW Mulitcast Switch        
+L3M4                  0%         0/       0       L3M4                         
+L3I4                  0%         0/       0       L3I4                         
+NDMB                  0%         0/       0       NDMB                         
+NDIO                  0%         0/       0       NDIO                         
+L3MB                  0%         0/       0       L3MB                         
+L3IO                  0%         0/       0       L3IO                         
+PNGM                  0%         0/       0       PNGM                         
+FECD                  0%         0/  8ec6bf       FECD Forward Equal Class
+                                                  Develope                     
+MPSI                  0%         0/   a129b       MPSI                         
+MPSM                  0%         0/       0       MPSM                         
+PPI                   0%         0/  268afa       PPI Product Process Interface
+MAC                   0%         0/  178cbf       MAC Media Access Control     
+IFPD                  0%         0/ 5ad822e       IFPD Ifnet Product Adapt     
+TRUN                  0%         0/       0       TRUNK Product Adapt          
+STFW                  0%         0/       0       STFW Super task forward      
+XQOS                  0%         0/  fe4207       XQOS Quality of service      
+MIRR                  0%         0/       0       MIRR                         
+SMLK                  0%         0/  9902dc       SMLK Smart Link Protocol     
+CDM                   0%         0/  1d895a       CDM                          
+NFPT                  0%         0/ 222e67d       NFPTNFP timer task           
+SOCK                  0%         0/  2fc739       SOCKPacket schedule and
+                                                  process                      
+FIB                   0%         0/       0       FIB Forward Information Base 
+MFIB                  0%         0/   1d53a       MFIBMulticast forward info   
+IFNT                  0%         0/       0       IFNTIfnet task               
+VTYD                  0%         0/  7c0612       VTYDVirtual terminal         
+RSA                   0%         0/       0       RSA RSA public-key algorithms
+AGNT                  0%         0/       0       AGNTSNMP agent task          
+TRAP                  0%         0/       0       TRAPSNMP trap task           
+FMAT                  0%         0/  174892       FMATFault Manage task        
+MDMT                  0%         0/ 606d03e       MDMTModem task               
+CFM                   0%         0/       0       CFM Configuration file
+                                                  management                   
+HS2M                  0%         0/       0       HS2MHigh available task      
+TACH                  0%         0/ 2ef6727       TACHWTACACS                  
+ACL                   0%         0/   d5c88       ACL                          
+SECE                  0%         0/ 2051f1d       SECE Security                
+DEFD                  0%         0/   9eb24       DEFD CPU Defend              
+STRA                  0%         0/   2731b       STRA Source Track            
+MFF                   0%         0/  278c66       MFF MAC Forced Forwarding    
+LBDT                  0%         0/ 74c44c5       LBDT Loopback Detect Mpu     
+BFDA                  0%         0/       0       BFDA BFD Adapter             
+DAD-                  0%         0/       0       DAD-proxy Task               
+MSYN                  0%         0/  f12139       MSYN Mac Synchronization     
+AAA                   0%         0/       0       AAA                          
+RDS                   0%         0/       0       RDS                          
+UCM                   0%         0/   c8e49       UCM  User Control Management 
+DHCP                  0%         0/  4ea069       DHCP Dynamic Host Config
+                                                  Protocol                     
+AM                    0%         0/  661794       AM   Address Management      
+PTAL                  0%         0/       0       PTAL Portal                  
+PS                    0%         0/   7e140       PS   Portal local server     
+HTPD                  0%         0/  23ac4b       HTPD HTTP process for portal 
+EAP                   0%         0/   e652b       EAP  Extensible Authen
+                                                  Protocol                     
+TM                    0%         0/       0       TM   Transmission Management 
+SAM                   0%         0/   4cbf8       SAM  Service Agent Module    
+WEB                   0%         0/       0       WEB  Web                     
+POE+                  0%         0/       0       POE+ PPP Over Ethernet Plus  
+SMAG                  0%         0/       0       SMAG Smart Link Agent        
+GVRP                  0%         0/       0       GVRP Protocol                
+DLDP                  0%         0/  1e7e64       DLDP Protocol                
+LLDP                  0%         0/  3edb7e       LLDP Protocol                
+UDPH                  0%         0/       0       UDPH                         
+BPDU                  0%         0/   3df08       BPDU Adapter                 
+EFMT                  0%         0/  28d270       EFMTEST 802.3AH Test         
+ADPT                  0%         0/       0       ADPT Adapter                 
+ETHA                  0%         0/       0       ETHA                         
+SPM                   0%         0/ 25f5a48       SPM                          
+ROUT                  0%         0/ 4a529b7       ROUTRoute task               
+LSPM                  0%         0/  1ebb95       LSPMLsp management           
+RSVP                  0%         0/       0       RSVP task                    
+LDP                   0%         0/       0       LDP task                     
+GRES                  0%         0/       0       GRESM task                   
+GEM                   0%         0/       0       GEM task                     
+GEMR                  0%         0/       0       GEMRun task                  
+UTSK                  0%         0/       0       UTSK                         
+APP                   0%         0/       0       APP                          
+IP                    0%         0/  21e93f       IP                           
+EOAM                  0%         0/  19df19       EOAM1AG                      
+LINK                  0%         0/  67afb5       LINK                         
+STP                   0%         0/ 22ef8e2       STP                          
+VRPT                  0%         0/  273415       VRPT                         
+HOTT                  0%         0/       0       HOTT                         
+TNQA                  0%         0/  2a0e5d       TNQAC                        
+TTNQ                  0%         0/       0       TTNQAS                       
+TARP                  0%         0/       0       TARPING                      
+TTVP                  0%         0/       0       TTVPLS                       
+L2                    0%         0/  f07caa       L2                           
+VRRP                  0%         0/ 169431e       VRRP                         
+L2_P                  0%         0/ 17b2e2a       L2_PR                        
+ARP                   0%         0/       0       ARP                          
+NTPT                  0%         0/  207e8e       NTPT task                    
+SRMT                  0%         0/  880f55       SRMT System Resource Manage
+                                                  Timer                        
+SRMI                  0%         0/       0       SRMI External Interrupt      
+RMON                  0%         0/  23a925       RMONRemote monitoring        
+IFLP                  0%         0/   de4ef       IFLP                         
+OS                  100%        19/2ccb3fa0       Operation System 
+"""
+
+
+def test_extract_key_value_single():
+    print("=" * 60)
+    print("测试 1: extract_key_value_single - 提取单个 key:value")
+    print("=" * 60)
+    
+    pattern = re.compile(r'CPU Usage\s+:\s*(?P<value>\d+)%')
+    result = StringTableMatching.extract_key_value_single(pattern, CPU_OUTPUT, key_group="value", value_group="value")
+    print(f"CPU 当前使用率: {result}")
+    assert result is not None
+    print("PASS\n")
+
+
+def test_extract_key_value_multiple():
+    print("=" * 60)
+    print("测试 2: extract_key_value_multiple - 提取多个 key:value")
+    print("=" * 60)
+    
+    pattern = re.compile(r'(?P<key>CPU Usage(?:\s+Stat\.\s+\w+)?)\s*:\s*(?P<value>[^\n]+?)\s*$', re.MULTILINE)
+    result = StringTableMatching.extract_key_value_multiple(pattern, CPU_OUTPUT)
+    print(f"提取到 {len(result)} 个 key:value 对:")
+    for k, v in result.items():
+        print(f"  {k}: {v}")
+    assert len(result) >= 2
+    print("PASS\n")
+
+
+def test_extract_values_by_groups():
+    print("=" * 60)
+    print("测试 3: extract_values_by_groups - 按命名分组提取任务表前几列")
+    print("=" * 60)
+    
+    pattern = re.compile(
+        r'^(?P<task_name>\S+)\s+(?P<cpu>\d+)%\s+(?P<tick_high>\w+)\s*/\s*(?P<tick_low>\w+)',
+        re.MULTILINE
+    )
+    result = StringTableMatching.extract_values_by_groups(pattern, CPU_OUTPUT)
+    print(f"提取到 {len(result)} 个任务:")
+    for row in result[:5]:
+        print(f"  {row}")
+    print("  ...")
+    assert len(result) > 0
+    print("PASS\n")
+
+
+def test_extract_table_to_list():
+    print("=" * 60)
+    print("测试 4: extract_table_to_list - 提取表格为二维列表")
+    print("=" * 60)
+    
+    lines = CPU_OUTPUT.strip().split('\n')
+    table_start = next(i for i, line in enumerate(lines) if line.startswith('TaskName'))
+    table_lines = lines[table_start + 1:]
+    table_text = '\n'.join(table_lines)
+    
+    pattern = re.compile(r'(?P<value>\S+)')
+    result = StringTableMatching.extract_table_to_list(pattern, table_text)
+    print(f"提取到 {len(result)} 行:")
+    for row in result[:5]:
+        print(f"  {row[:3]}...")
+    print("  ...")
+    assert len(result) > 0
+    print("PASS\n")
+
+
+def test_cpu_utilization_line():
+    print("=" * 60)
+    print("测试 5: 提取 CPU 利用率行（5秒/1分钟/5分钟）")
+    print("=" * 60)
+    
+    pattern = re.compile(
+        r'five seconds:\s*(?P<five_sec>\d+)%.*?one minute:\s*(?P<one_min>\d+)%.*?five minutes:\s*(?P<five_min>\d+)%'
+    )
+    result = StringTableMatching.extract_values_by_groups(pattern, CPU_OUTPUT)
+    print(f"CPU 利用率: {result}")
+    assert len(result) == 1
+    assert 'five_sec' in result[0]
+    print("PASS\n")
+
+
+def test_os_task():
+    print("=" * 60)
+    print("测试 6: 提取 OS 任务（CPU 100% 的那行）")
+    print("=" * 60)
+    
+    pattern = re.compile(r'^(?P<task>OS)\s+(?P<cpu>100)%\s+(?P<tick_high>\w+)\s*/\s*(?P<tick_low>\w+)\s+(?P<desc>.+)$', re.MULTILINE)
+    result = StringTableMatching.extract_values_by_groups(pattern, CPU_OUTPUT)
+    print(f"OS 任务: {result}")
+    assert len(result) == 1
+    assert result[0]['cpu'] == '100'
+    print("PASS\n")
+
+
+if __name__ == '__main__':
+    test_extract_key_value_single()
+    test_extract_key_value_multiple()
+    test_extract_values_by_groups()
+    test_extract_table_to_list()
+    test_cpu_utilization_line()
+    test_os_task()
+    print("=" * 60)
+    print("所有测试通过!")
+    print("=" * 60)