easy-kit 0.0.3__tar.gz → 0.0.4.dev0__tar.gz

{easy_kit-0.0.3 → easy_kit-0.0.4.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: easy-kit
-Version: 0.0.3
+Version: 0.0.4.dev0
 Summary: Python toolkit
 Author: flegac
 Author-email: flegac <florent.legac@gmail.com>

easy_kit-0.0.4.dev0/easy_kit/event.py

@@ -0,0 +1,39 @@
+from typing import Callable
+
+
+class Event:
+    IS_ENABLED = False
+
+    @staticmethod
+    def enable(status: bool = True):
+        Event.IS_ENABLED = status
+
+    def __init__(self, name: str = 'EventDefault'):
+        self.name = name
+        self.observer: list[Callable[[...], ...]] = []
+        self.enabled = True
+        self.verbose = False
+
+    def emitter(self, *args, **kwargs):
+        def _fun():
+            self.emit(*args, **kwargs)
+
+        return _fun
+
+    def emit(self, *args, **kwargs):
+        if self.enabled:
+            for _ in self.observer:
+                _(*args, **kwargs)
+
+    def connect[** P, T](self, func: Callable[P, T]) -> Callable[P, T]:
+        if Event.IS_ENABLED:
+            self.observer.append(func)
+        return func
+
+    def observe[** P, T](self, func: Callable[P, T]) -> Callable[P, T]:
+        def _fun(*args, **kwargs) -> T:
+            res = func(*args, **kwargs)
+            self.emit(res)
+            return res
+
+        return _fun

easy_kit-0.0.4.dev0/easy_kit/timing.py

@@ -0,0 +1,177 @@
+import inspect
+import statistics
+import time
+from collections import defaultdict
+from contextlib import contextmanager
+from dataclasses import dataclass, field
+from functools import wraps
+from pprint import pprint
+from typing import Callable
+from unittest import TestCase
+
+HEADERS = ['label', 'total (s)', 'count', 'min', 'max', 'mean', 'std']
+
+
+def _tabulate(headers: list[str], data: list[list[str]]):
+    try:
+        from tabulate import tabulate
+        return tabulate(
+            headers=headers,
+            floatfmt='.5f',
+            tabular_data=data,
+        )
+    except:
+        pass
+
+    raw = [
+        headers,
+        *[
+            [f'{_:5}' for _ in row]
+            for row in data
+        ]
+    ]
+
+    lengths = [
+        max(len(raw[row][col]) for row in range(len(raw)))
+        for col in range(len(headers))
+    ]
+
+    raw.insert(1, ['-' * _ for _ in lengths])
+
+    return '\n'.join([
+        '   '.join([_.ljust(l) for _, l in zip(row, lengths)])
+        for row in raw
+    ])
+
+
+@dataclass
+class TimeEntry:
+    events: list[float] = field(default_factory=list)
+
+    def raw_line(self, key: str):
+        return [key, self.total, self.count, self.min, self.max, self.mean, self.std]
+
+    @property
+    def total(self):
+        return sum(self.events)
+
+    @property
+    def count(self):
+        return len(self.events)
+
+    @property
+    def min(self):
+        return self._undefined(min(self.events))
+
+    @property
+    def max(self):
+        return self._undefined(max(self.events))
+
+    @property
+    def mean(self):
+        return self._undefined(statistics.mean(self.events))
+
+    @property
+    def std(self):
+        return self._undefined(statistics.pstdev(self.events))
+
+    def _undefined(self, value: float):
+        if len(self.events) <= 1:
+            return ''
+        return value
+
+
+class DefaultLogger:
+    debug = print
+    info = print
+    warning = print
+
+
+class Timings:
+    def __init__(self):
+        self.db: dict[str, TimeEntry] = defaultdict(lambda: TimeEntry())
+        self.active = False
+        self.logs = False
+        self.logger = DefaultLogger
+
+    @contextmanager
+    def timing(self, name: str = None):
+        if name is None:
+            name = inspect.stack()[11].function
+        start = self._before(name)
+        yield
+        self._after(name, start)
+
+    def time_func[** P, R](self, func: Callable[P, R]) -> Callable[P, R]:
+        @wraps(func)
+        def inner(*args: P.args, **kwargs: P.kwargs) -> R:
+            with self.timing(func.__qualname__):
+                return func(*args, **kwargs)
+
+        return inner
+
+    def show_timing(self):
+        if not self.active:
+            return
+
+        try:
+            self.logger.info('\n' + self.format_table())
+        except Exception as e:
+            self.logger.warning(f'Warning: {e}')
+
+    def raw_table(self):
+        return sorted([
+            entry.raw_line(key)
+            for key, entry in self.db.items()
+        ], key=lambda row: row[1], reverse=True)
+
+    def format_table(self):
+        return _tabulate(headers=HEADERS, data=self.raw_table())
+
+    def setup_timing(self, status: bool = True, logs: bool = False):
+        self.active = status
+        self.logs = logs
+
+    def tree_structure(self):
+        groups = {}
+
+        for key, entry in self.db.items():
+            try:
+                major, minor = key.split('.', maxsplit=1)
+            except:
+                major = '___'
+                minor = key
+            if major not in groups:
+                groups[major] = {}
+            groups[major][minor] = entry
+        return groups
+
+    def _before(self, name: str):
+        if self.logs:
+            self.logger.debug(f'+ {name}')
+        if self.active:
+            return time.time()
+
+    def _after(self, name: str, start: float | None):
+        if self.logs:
+            self.logger.debug(f'- {name}')
+        if start is not None:
+            total = time.time() - start
+            self.db[name].events.append(total)
+
+
+_TIMING = Timings()
+timing = _TIMING.timing
+time_func = _TIMING.time_func
+show_timing = _TIMING.show_timing
+setup_timing = _TIMING.setup_timing
+
+
+class TimingTestCase(TestCase):
+    @classmethod
+    def setUpClass(cls):
+        setup_timing()
+
+    @classmethod
+    def tearDownClass(cls):
+        show_timing()

{easy_kit-0.0.3 → easy_kit-0.0.4.dev0}/easy_kit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: easy-kit
-Version: 0.0.3
+Version: 0.0.4.dev0
 Summary: Python toolkit
 Author: flegac
 Author-email: flegac <florent.legac@gmail.com>

{easy_kit-0.0.3 → easy_kit-0.0.4.dev0}/easy_kit.egg-info/SOURCES.txt

@@ -2,6 +2,9 @@ LICENSE
 README.md
 pyproject.toml
 setup.py
+easy_kit/__init__.py
+easy_kit/event.py
+easy_kit/timing.py
 easy_kit.egg-info/PKG-INFO
 easy_kit.egg-info/SOURCES.txt
 easy_kit.egg-info/dependency_links.txt

easy_kit-0.0.4.dev0/easy_kit.egg-info/top_level.txt

@@ -0,0 +1 @@
+easy_kit

{easy_kit-0.0.3 → easy_kit-0.0.4.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "easy-kit"
-version = "0.0.3"
+version = "0.0.4.dev0"
 authors = [
   { name="flegac", email="florent.legac@gmail.com" },
 ]

{easy_kit-0.0.3 → easy_kit-0.0.4.dev0}/setup.py

@@ -1,6 +1,4 @@
-from pathlib import Path
-
-from setuptools import setup, find_packages
+from setuptools import setup
 
 with open("README.md", "r", encoding="utf-8") as fh:
     long_description = fh.read()
@@ -17,7 +15,7 @@ setup(
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
-    packages=find_packages(),
+    packages=['easy_kit'],
     # install_requires=[line for line in open('requirements.txt')],
     python_requires=">=3.7",
     include_package_data=True

easy_kit-0.0.3/easy_kit.egg-info/top_level.txt

@@ -1 +0,0 @@
-