ominfra 0.0.0.dev218__py3-none-any.whl → 0.0.0.dev219__py3-none-any.whl

ominfra/clouds/aws/instancetypes/cli.py

@@ -5,9 +5,9 @@ import typing as ta
 
 from omdev.secrets import load_secrets
 from omlish import lang
-from omlish import secrets as sec
 from omlish.argparse import all as ap
 from omlish.formats import json
+from omlish.secrets import all as sec
 
 from .cache import load_instance_types
 

ominfra/scripts/supervisor.py

@@ -71,6 +71,7 @@ import select
 import shlex
 import signal
 import socket
+import socket as socket_
 import stat
 import string
 import sys
@@ -2820,15 +2821,15 @@ TODO:
 class SocketAddressInfoArgs:
     host: ta.Optional[str]
     port: ta.Union[str, int, None]
-    family: socket.AddressFamily = socket.AddressFamily.AF_UNSPEC
+    family: socket_.AddressFamily = socket_.AddressFamily.AF_UNSPEC
     type: int = 0
     proto: int = 0
-    flags: socket.AddressInfo = socket.AddressInfo(0)
+    flags: socket_.AddressInfo = socket_.AddressInfo(0)
 
 
 @dc.dataclass(frozen=True)
 class SocketAddressInfo:
-    family: socket.AddressFamily
+    family: socket_.AddressFamily
     type: int
     proto: int
     canonname: ta.Optional[str]
@@ -2836,32 +2837,44 @@ class SocketAddressInfo:
 
 
 class SocketFamilyAndAddress(ta.NamedTuple):
-    family: socket.AddressFamily
+    family: socket_.AddressFamily
     address: SocketAddress
 
 
 def get_best_socket_family(
         host: ta.Optional[str],
         port: ta.Union[str, int, None],
-        family: ta.Union[int, socket.AddressFamily] = socket.AddressFamily.AF_UNSPEC,
+        family: ta.Union[int, socket_.AddressFamily] = socket_.AddressFamily.AF_UNSPEC,
 ) -> SocketFamilyAndAddress:
     """https://github.com/python/cpython/commit/f289084c83190cc72db4a70c58f007ec62e75247"""
 
-    infos = socket.getaddrinfo(
+    infos = socket_.getaddrinfo(
         host,
         port,
         family,
-        type=socket.SOCK_STREAM,
-        flags=socket.AI_PASSIVE,
+        type=socket_.SOCK_STREAM,
+        flags=socket_.AI_PASSIVE,
     )
     ai = SocketAddressInfo(*next(iter(infos)))
     return SocketFamilyAndAddress(ai.family, ai.sockaddr)
 
 
 class SocketAndAddress(ta.NamedTuple):
-    socket: socket.socket
+    socket: socket_.socket
     address: SocketAddress
 
+    def wrap_socket(self, fn: ta.Callable[[socket_.socket], socket_.socket]):
+        return self._replace(socket=fn(self.socket))
+
+    @classmethod
+    def socket_wrapper(
+            cls,
+            fn: ta.Callable[[socket_.socket], socket_.socket],
+    ) -> ta.Callable[['SocketAndAddress'], 'SocketAndAddress']:
+        def inner(conn):
+            return conn.wrap_socket(fn)
+        return inner
+
 
 ########################################
 # ../../../omlish/sockets/io.py
@@ -6391,6 +6404,12 @@ def journald_log_handler_factory(
 ##
 
 
+class SocketHandler_(abc.ABC):  # noqa
+    @abc.abstractmethod
+    def __call__(self, addr: SocketAddress, f: SocketIoPair) -> None:
+        raise NotImplementedError
+
+
 ########################################
 # ../configs.py
 
@@ -6846,6 +6865,12 @@ class UnsupportedMethodHttpHandlerError(Exception):
     pass
 
 
+class HttpHandler_(abc.ABC):  # noqa
+    @abc.abstractmethod
+    def __call__(self, req: HttpHandlerRequest) -> HttpHandlerResponse:
+        raise NotImplementedError
+
+
 ########################################
 # ../../../omlish/lite/configs.py
 
@@ -7710,7 +7735,7 @@ class CoroHttpServer:
 ##
 
 
-class CoroHttpServerSocketHandler:  # SocketHandler
+class CoroHttpServerSocketHandler(SocketHandler_):
     def __init__(
             self,
             server_factory: CoroHttpServerFactory,

ominfra/tools/listresources.py

@@ -19,7 +19,7 @@ from omdev.secrets import load_secrets
 from omlish import check
 from omlish import dataclasses as dc
 from omlish import lang
-from omlish import secrets as sec
+from omlish.secrets import all as sec
 
 
 ##

{ominfra-0.0.0.dev218.dist-info → ominfra-0.0.0.dev219.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ominfra
-Version: 0.0.0.dev218
+Version: 0.0.0.dev219
 Summary: ominfra
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,8 +12,8 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omdev==0.0.0.dev218
-Requires-Dist: omlish==0.0.0.dev218
+Requires-Dist: omdev==0.0.0.dev219
+Requires-Dist: omlish==0.0.0.dev219
 Provides-Extra: all
 Requires-Dist: paramiko~=3.5; extra == "all"
 Requires-Dist: asyncssh~=2.18; extra == "all"

{ominfra-0.0.0.dev218.dist-info → ominfra-0.0.0.dev219.dist-info}/RECORD

@@ -16,7 +16,7 @@ ominfra/clouds/aws/instancetypes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 ominfra/clouds/aws/instancetypes/__main__.py,sha256=Jsrv3P7LX2Cg08W7ByZfZ1JQT4lgLDPW1qNAmShFuMk,75
 ominfra/clouds/aws/instancetypes/cache.json.gz,sha256=JGgZlMGh2nLPsFmn3PwtV5gO5UMWrjFSFnhniGZKbJc,26375
 ominfra/clouds/aws/instancetypes/cache.py,sha256=CrfBzOb0DqQhf3ahQS6qQHHH1kVgZyniYOl22RTvjwo,333
-ominfra/clouds/aws/instancetypes/cli.py,sha256=24DA2Qio27NlyqqOrw-F7rm2NfN3e_Kn2wIYxCeXbMY,2191
+ominfra/clouds/aws/instancetypes/cli.py,sha256=HcfOchROLGFsxmEIuXtXVywFJWjHOk4yvCKrbZFWVGQ,2195
 ominfra/clouds/aws/journald2aws/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
 ominfra/clouds/aws/journald2aws/__main__.py,sha256=d23loR_cKfTYZwYiqpt_CmKI7dd5WcYFgIYzqMep75E,68
 ominfra/clouds/aws/journald2aws/cursor.py,sha256=tQ7O6BHlEdaalbiI_Rqagj0aHfdtTQ_ZJwdOSRUjNvQ,1173
@@ -114,7 +114,7 @@ ominfra/manage/targets/targets.py,sha256=7GP6UAZyJFEhpkJN6UQdpr_WN3p7C76v-s445y-
 ominfra/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ominfra/scripts/journald2aws.py,sha256=CM1nZMpecXJpcg7icJUCJ-_DX-7CduQurxZQ7EmhQmA,171450
 ominfra/scripts/manage.py,sha256=MaUur1_dDdMHt5YnE6Waxa_v70kydHzKCBHXAuTmKd0,362827
-ominfra/scripts/supervisor.py,sha256=ejPO7r3Ta0RQbP28MyL8MJNS1jop6g8d9qdv-vJ0OYA,294602
+ominfra/scripts/supervisor.py,sha256=W_8YHq0Cf5AF88_MBwnZVodxXdQ5AE0zAeafvYMRYFg,295380
 ominfra/supervisor/LICENSE.txt,sha256=ZrHY15PVR98y26Yg6iQfa-SXnUaYTDhrUsPVcEO5OKM,1874
 ominfra/supervisor/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
 ominfra/supervisor/__main__.py,sha256=I0yFw-C08OOiZ3BF6lF1Oiv789EQXu-_j6whDhQUTEA,66
@@ -155,10 +155,10 @@ ominfra/tailscale/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 ominfra/tailscale/api.py,sha256=C5-t_b6jZXUWcy5k8bXm7CFnk73pSdrlMOgGDeGVrpw,1370
 ominfra/tailscale/cli.py,sha256=3FnJbgpLw6gInTfhERd1mDy9ijjMUGxkdYVo43Tnxx4,3555
 ominfra/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ominfra/tools/listresources.py,sha256=4qVg5txsb10EHhvqXXeM6gJ2jx9LbroEnPydDv1uXs0,6176
-ominfra-0.0.0.dev218.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-ominfra-0.0.0.dev218.dist-info/METADATA,sha256=CIoVjeFWhR47YXXqQ65hW7lHeJK6XpLd0jTGOcP6RA4,731
-ominfra-0.0.0.dev218.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ominfra-0.0.0.dev218.dist-info/entry_points.txt,sha256=kgecQ2MgGrM9qK744BoKS3tMesaC3yjLnl9pa5CRczg,37
-ominfra-0.0.0.dev218.dist-info/top_level.txt,sha256=E-b2OHkk_AOBLXHYZQ2EOFKl-_6uOGd8EjeG-Zy6h_w,8
-ominfra-0.0.0.dev218.dist-info/RECORD,,
+ominfra/tools/listresources.py,sha256=BxFoxtyF2aVEcW67ZD9QujoBkJBtvIxGmKqXvLwnEHo,6180
+ominfra-0.0.0.dev219.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+ominfra-0.0.0.dev219.dist-info/METADATA,sha256=E0e_wKkjHZleJ28wnBhOjkR8ooXakDMT7XoBxkebkto,731
+ominfra-0.0.0.dev219.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ominfra-0.0.0.dev219.dist-info/entry_points.txt,sha256=kgecQ2MgGrM9qK744BoKS3tMesaC3yjLnl9pa5CRczg,37
+ominfra-0.0.0.dev219.dist-info/top_level.txt,sha256=E-b2OHkk_AOBLXHYZQ2EOFKl-_6uOGd8EjeG-Zy6h_w,8
+ominfra-0.0.0.dev219.dist-info/RECORD,,