xkits-thread 0.2__tar.gz → 0.2.1__tar.gz

{xkits_thread-0.2 → xkits_thread-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xkits-thread
-Version: 0.2
+Version: 0.2.1
 Summary: Thread module
 Home-page: https://github.com/bondbox/xthreads/
 Author: Mingzhe Zou
@@ -16,7 +16,8 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: xkits-lib>=0.1
+Requires-Dist: xkits-lib>=0.4
+Requires-Dist: psutil
 
 # xthreads
 

{xkits_thread-0.2 → xkits_thread-0.2.1}/xkits_thread/__init__.py

@@ -3,6 +3,7 @@
 from xkits_thread.executor import ThreadPool  # noqa:F401
 from xkits_thread.executor import hourglass  # noqa:F401,H306
 from xkits_thread.lock import NamedLock  # noqa:F401
+from xkits_thread.proc import Processes  # noqa:F401
 from xkits_thread.task import DaemonTaskJob  # noqa:F401
 from xkits_thread.task import DelayTaskJob  # noqa:F401
 from xkits_thread.task import TaskJob  # noqa:F401

{xkits_thread-0.2 → xkits_thread-0.2.1}/xkits_thread/attribute.py

@@ -1,7 +1,7 @@
 # coding:utf-8
 
 __project__ = "xkits-thread"
-__version__ = "0.2"
+__version__ = "0.2.1"
 __urlhome__ = "https://github.com/bondbox/xthreads/"
 __description__ = "Thread module"
 

xkits_thread-0.2.1/xkits_thread/proc.py

@@ -0,0 +1,68 @@
+# coding:utf-8
+
+from typing import Dict
+from typing import Iterator
+from typing import List
+from typing import Union
+
+from psutil import Process
+
+
+class Processes():
+
+    def __init__(self) -> None:
+        self.__processes: Dict[int, Process] = {}
+
+    def __iter__(self) -> Iterator[Process]:
+        invalid: List[Process] = []
+
+        for pid in sorted(self.__processes.keys()):
+            if not (obj := self.__processes[pid]).is_running():
+                invalid.append(obj)
+                continue
+
+            yield obj
+
+        for obj in invalid:
+            del self.__processes[obj.pid]  # pragma: no cover
+
+    def __getitem__(self, pid: int):
+        if pid not in self.__processes:
+            obj: Process = Process(pid=pid)
+            self.__processes.setdefault(pid, obj)
+        return self.__processes[pid]
+
+    def __len__(self) -> int:
+        return len(self.__processes)
+
+    def add(self, p: Union[Process, int]) -> bool:
+        obj: Process = Process(pid=p) if isinstance(p, int) else p
+        assert isinstance(obj, Process), f"{type(obj)} is not Process"
+
+        if (pid := obj.pid) in self.__processes:
+            if (proc := self.__processes[pid]).is_running():
+                return proc == obj
+            del self.__processes[pid]
+
+        self.__processes.setdefault(pid, obj)
+        return self.__processes[pid] is obj
+
+    def select(self, name: str, exact: bool = True) -> None:
+        def _name_filter(process_name: str) -> bool:
+            return process_name == name if exact else name in process_name
+
+        from psutil import process_iter  # pylint:disable=C0415
+        for proc in process_iter(["name"]):
+            if _name_filter(proc.name()):
+                self.add(proc)
+
+    @classmethod
+    def search(cls, name: str, exact: bool = True) -> "Processes":
+        instance: Processes = cls()
+        instance.select(name, exact)
+        return instance
+
+
+if __name__ == "__main__":
+    for _p in Processes.search("systemd"):
+        print(f"{_p.name()} pid:{_p.pid}")

{xkits_thread-0.2 → xkits_thread-0.2.1}/xkits_thread.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xkits-thread
-Version: 0.2
+Version: 0.2.1
 Summary: Thread module
 Home-page: https://github.com/bondbox/xthreads/
 Author: Mingzhe Zou
@@ -16,7 +16,8 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: xkits-lib>=0.1
+Requires-Dist: xkits-lib>=0.4
+Requires-Dist: psutil
 
 # xthreads
 

{xkits_thread-0.2 → xkits_thread-0.2.1}/xkits_thread.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ xkits_thread/__init__.py
 xkits_thread/attribute.py
 xkits_thread/executor.py
 xkits_thread/lock.py
+xkits_thread/proc.py
 xkits_thread/task.py
 xkits_thread.egg-info/PKG-INFO
 xkits_thread.egg-info/SOURCES.txt

xkits_thread-0.2.1/xkits_thread.egg-info/requires.txt

@@ -0,0 +1,2 @@
+xkits-lib>=0.4
+psutil

xkits_thread-0.2/xkits_thread.egg-info/requires.txt

@@ -1 +0,0 @@
-xkits-lib>=0.1