omlish 0.0.0.dev6__py3-none-any.whl → 0.0.0.dev8__py3-none-any.whl

omlish/__about__.py

@@ -1,6 +1,110 @@
-__author__ = 'wrmsr'
-__url__ = 'https://github.com/wrmsr/omlish'
-__license__ = 'BSD-3-Clause'
-__requires_python__ = '>=3.12'
+__version__ = '0.0.0.dev8'
 
-__version__ = '0.0.0.dev6'
+
+#
+
+
+class ProjectBase:
+    name: str | None = None
+    authors = [{'name': 'wrmsr'}]
+    urls = {'source': 'https://github.com/wrmsr/omlish'}
+    license = {'text': 'BSD-3-Clause'}
+    requires_python = '>=3.12'
+
+    version = __version__
+
+    classifiers = [
+        'License :: OSI Approved :: BSD License',
+        'Development Status :: 2 - Pre-Alpha',
+        'Intended Audience :: Developers',
+
+        'Operating System :: OS Independent',
+        'Operating System :: POSIX',
+    ]
+
+
+class Project(ProjectBase):
+    name = 'omlish'
+    description = 'omlish'
+
+    optional_dependencies = {
+        'async': [
+            'anyio >= 4.4',
+            'sniffio >= 1.3',
+
+            'greenlet >= 3; python_version < "3.13"',
+
+            'trio >= 0.26',
+            'trio-asyncio >= 0.15; python_version < "3.13"',
+        ],
+
+        'compression': [
+            'lz4 >= 4',
+            'python-snappy >= 0.7; python_version < "3.13"',
+            'zstd >= 1.5',
+        ],
+
+        'formats': [
+            'orjson > 3.10',
+            'cloudpickle >= 3',
+            'pyyaml >= 5',
+        ],
+
+        'http': [
+            'httpx[http2] >= 0.27',
+        ],
+
+        'misc': [
+            'jinja2 >= 3.1',
+            'psutil >= 6',
+            'wrapt >= 1.14',
+        ],
+
+        'secrets': [
+            'cryptography >= 43',
+        ],
+
+        'sql': [
+            'sqlalchemy >= 2; python_version >= "3.13"',
+            'sqlalchemy[asyncio] >= 2; python_version < "3.13"',
+
+            'pg8000 >= 1.31',
+            'pymysql >= 1.1',
+
+            'aiomysql >= 0.2',
+            'aiosqlite >= 0.20',
+            'asyncpg >= 0.29; python_version < "3.13"',
+        ],
+
+        'sqlx': [
+            "sqlean.py >= 3.45; python_version < '3.13'",
+
+            'duckdb >= 1',
+        ],
+
+        'testing': [
+            'pytest >= 8',
+        ],
+    }
+
+
+#
+
+
+class SetuptoolsBase:
+    manifest_in = [
+        'global-exclude **/conftest.py',
+    ]
+
+    include_package_data = False
+
+    find_packages = {
+        'exclude': ['*.tests', '*.tests.*'],
+    }
+
+
+class Setuptools(SetuptoolsBase):
+    find_packages = {
+        'include': [Project.name, f'{Project.name}.*'],
+        'exclude': [*SetuptoolsBase.find_packages['exclude']],
+    }

omlish/__init__.py

@@ -1,8 +0,0 @@
-import sys as _sys
-
-
-REQUIRED_PYTHON_VERSION = (3, 12)
-
-
-if _sys.version_info < REQUIRED_PYTHON_VERSION:
-    raise RuntimeError(f'Python version {_sys.version_info=} < {REQUIRED_PYTHON_VERSION}')

omlish/asyncs/__init__.py

@@ -33,12 +33,3 @@ from .flavors import (  # noqa
     mark_trio,
     with_adapter_loop,
 )
-
-from .futures import (  # noqa
-    FutureError,
-    FutureTimeoutError,
-    ImmediateExecutor,
-    new_thread_or_immediate_executor,
-    wait_dependent_futures,
-    wait_futures,
-)

omlish/asyncs/anyio.py

@@ -13,9 +13,20 @@ lookit:
  - https://github.com/M-o-a-T/aevent
  - https://github.com/florimondmanca/aiometer
  - https://github.com/sanitizers/octomachinery/blob/b36c3d3d49da813ac46e361424132955a4e99ac8/octomachinery/utils/asynctools.py
+
+==
+
+async def killer(shutdown: anyio.Event, sleep_s: float) -> None:
+    log.warning('Killing in %d seconds', sleep_s)
+    await anyio.sleep(sleep_s)
+    log.warning('Killing')
+    shutdown.set()
+
 """  # noqa
+import signal
 import typing as ta
 
+import anyio.abc
 import anyio.streams.memory
 import anyio.streams.stapled
 import sniffio
@@ -214,3 +225,32 @@ class LazyFn(ta.Generic[T]):
             self._v = lang.just(await self._fn())
         await self._once.do(do)
         return self._v.must()
+
+
+##
+
+
+async def install_shutdown_signal_handler(
+        tg: anyio.abc.TaskGroup,
+        event: anyio.Event | None = None,
+        *,
+        signals: ta.Iterable[int] = (signal.SIGINT, signal.SIGTERM),
+        echo: bool = False,
+) -> ta.Callable[..., ta.Awaitable[None]] | None:
+    if event is None:
+        event = anyio.Event()
+
+    async def _handler(*, task_status=anyio.TASK_STATUS_IGNORED):
+        with anyio.open_signal_receiver(*signals) as it:  # type: ignore
+            task_status.started()
+            async for signum in it:
+                if echo:
+                    if signum == signal.SIGINT:
+                        print('Ctrl+C pressed!')
+                    else:
+                        print('Terminated!')
+                event.set()
+                return
+
+    await tg.start(_handler)
+    return event.wait