omlish 0.0.0.dev470__tar.gz → 0.0.0.dev476__tar.gz

{omlish-0.0.0.dev470/omlish.egg-info → omlish-0.0.0.dev476}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omlish
-Version: 0.0.0.dev470
+Version: 0.0.0.dev476
 Summary: omlish
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -18,12 +18,12 @@ Provides-Extra: all
 Requires-Dist: anyio~=4.11; extra == "all"
 Requires-Dist: sniffio~=1.3; extra == "all"
 Requires-Dist: greenlet~=3.2; extra == "all"
-Requires-Dist: trio~=0.31; extra == "all"
+Requires-Dist: trio~=0.32; extra == "all"
 Requires-Dist: trio-asyncio~=0.15; extra == "all"
 Requires-Dist: lz4~=4.4; extra == "all"
 Requires-Dist: python-snappy~=0.7; extra == "all"
 Requires-Dist: zstandard~=0.25; python_version < "3.14" and extra == "all"
-Requires-Dist: brotli~=1.1; extra == "all"
+Requires-Dist: brotli~=1.2; extra == "all"
 Requires-Dist: asttokens~=3.0; extra == "all"
 Requires-Dist: executing~=2.2; extra == "all"
 Requires-Dist: psutil~=7.1; extra == "all"
@@ -41,12 +41,12 @@ Requires-Dist: pymysql~=1.1; extra == "all"
 Requires-Dist: aiomysql~=0.3; extra == "all"
 Requires-Dist: aiosqlite~=0.21; extra == "all"
 Requires-Dist: asyncpg~=0.30; extra == "all"
-Requires-Dist: apsw~=3.50; extra == "all"
+Requires-Dist: apsw~=3.51; extra == "all"
 Requires-Dist: sqlean.py~=3.50; extra == "all"
 Requires-Dist: duckdb~=1.4; extra == "all"
 Requires-Dist: markupsafe~=3.0; extra == "all"
 Requires-Dist: jinja2~=3.1; extra == "all"
-Requires-Dist: pytest~=8.4; extra == "all"
+Requires-Dist: pytest~=9.0; extra == "all"
 Requires-Dist: anyio~=4.11; extra == "all"
 Requires-Dist: sniffio~=1.3; extra == "all"
 Requires-Dist: asttokens~=3.0; extra == "all"
@@ -58,13 +58,13 @@ Provides-Extra: async
 Requires-Dist: anyio~=4.11; extra == "async"
 Requires-Dist: sniffio~=1.3; extra == "async"
 Requires-Dist: greenlet~=3.2; extra == "async"
-Requires-Dist: trio~=0.31; extra == "async"
+Requires-Dist: trio~=0.32; extra == "async"
 Requires-Dist: trio-asyncio~=0.15; extra == "async"
 Provides-Extra: compress
 Requires-Dist: lz4~=4.4; extra == "compress"
 Requires-Dist: python-snappy~=0.7; extra == "compress"
 Requires-Dist: zstandard~=0.25; python_version < "3.14" and extra == "compress"
-Requires-Dist: brotli~=1.1; extra == "compress"
+Requires-Dist: brotli~=1.2; extra == "compress"
 Provides-Extra: diag
 Requires-Dist: asttokens~=3.0; extra == "diag"
 Requires-Dist: executing~=2.2; extra == "diag"
@@ -89,14 +89,14 @@ Requires-Dist: pymysql~=1.1; extra == "sqldrivers"
 Requires-Dist: aiomysql~=0.3; extra == "sqldrivers"
 Requires-Dist: aiosqlite~=0.21; extra == "sqldrivers"
 Requires-Dist: asyncpg~=0.30; extra == "sqldrivers"
-Requires-Dist: apsw~=3.50; extra == "sqldrivers"
+Requires-Dist: apsw~=3.51; extra == "sqldrivers"
 Requires-Dist: sqlean.py~=3.50; extra == "sqldrivers"
 Requires-Dist: duckdb~=1.4; extra == "sqldrivers"
 Provides-Extra: templates
 Requires-Dist: markupsafe~=3.0; extra == "templates"
 Requires-Dist: jinja2~=3.1; extra == "templates"
 Provides-Extra: testing
-Requires-Dist: pytest~=8.4; extra == "testing"
+Requires-Dist: pytest~=9.0; extra == "testing"
 Provides-Extra: plus
 Requires-Dist: anyio~=4.11; extra == "plus"
 Requires-Dist: sniffio~=1.3; extra == "plus"
@@ -234,7 +234,8 @@ dependencies of any kind**.
   - **[queries](https://github.com/wrmsr/omlish/blob/master/omlish/sql/queries)** - A SQL query builder with a fluent
     interface.
   - **[alchemy](https://github.com/wrmsr/omlish/blob/master/omlish/sql/alchemy)** - SQLAlchemy utilities. The codebase
-    is moving away from SQLAlchemy however in favor of its own internal SQL api.
+    has moved away from SQLAlchemy in favor of its own internal SQL api, but it will likely still remain as an optional
+    dep for the api adapter.
 
 - **[testing](https://github.com/wrmsr/omlish/blob/master/omlish/testing)** - Test - primarily pytest - helpers,
   including:
@@ -294,8 +295,8 @@ examples are:
 - **pytest** - What is used for all standard testing - as lite code has no dependencies of any kind its testing uses
   stdlib's [unittest](https://docs.python.org/3/library/unittest.html).
 - **wrapt** - For (optionally-enabled) injector circular proxies.
-- **sqlalchemy** - Parts of the codebase use SQLAlchemy for db stuff, but it is being migrated away from in favor of the
-  internal api. It will however likely still remain as an optional dep for the api adapter.
+- **sqlalchemy** - The codebase has migrated away from SQLAlchemy in favor of the internal api but it retains it as an
+  optional dep to support adapting the internal api to it.
 
 Additionally, some catchall dep categories include:
 

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/README.md

@@ -125,7 +125,8 @@ dependencies of any kind**.
   - **[queries](https://github.com/wrmsr/omlish/blob/master/omlish/sql/queries)** - A SQL query builder with a fluent
     interface.
   - **[alchemy](https://github.com/wrmsr/omlish/blob/master/omlish/sql/alchemy)** - SQLAlchemy utilities. The codebase
-    is moving away from SQLAlchemy however in favor of its own internal SQL api.
+    has moved away from SQLAlchemy in favor of its own internal SQL api, but it will likely still remain as an optional
+    dep for the api adapter.
 
 - **[testing](https://github.com/wrmsr/omlish/blob/master/omlish/testing)** - Test - primarily pytest - helpers,
   including:
@@ -185,8 +186,8 @@ examples are:
 - **pytest** - What is used for all standard testing - as lite code has no dependencies of any kind its testing uses
   stdlib's [unittest](https://docs.python.org/3/library/unittest.html).
 - **wrapt** - For (optionally-enabled) injector circular proxies.
-- **sqlalchemy** - Parts of the codebase use SQLAlchemy for db stuff, but it is being migrated away from in favor of the
-  internal api. It will however likely still remain as an optional dep for the api adapter.
+- **sqlalchemy** - The codebase has migrated away from SQLAlchemy in favor of the internal api but it retains it as an
+  optional dep to support adapting the internal api to it.
 
 Additionally, some catchall dep categories include:
 

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/__about__.py

@@ -1,5 +1,5 @@
-__version__ = '0.0.0.dev470'
-__revision__ = '967906fd30a40d061ede9f7aeda9dad0ec608508'
+__version__ = '0.0.0.dev476'
+__revision__ = '4be61621e1f66ad752c0ae86067058d44b4d8b8f'
 
 
 #
@@ -41,7 +41,7 @@ class Project(ProjectBase):
 
             'greenlet ~= 3.2',
 
-            'trio ~= 0.31',
+            'trio ~= 0.32',
             'trio-asyncio ~= 0.15',
         ],
 
@@ -53,7 +53,7 @@ class Project(ProjectBase):
 
             'zstandard ~= 0.25; python_version < "3.14"',
 
-            'brotli ~= 1.1',
+            'brotli ~= 1.2',
         ],
 
         'diag': [
@@ -103,7 +103,7 @@ class Project(ProjectBase):
             'aiosqlite ~= 0.21',
             'asyncpg ~= 0.30',
 
-            'apsw ~= 3.50',
+            'apsw ~= 3.51',
 
             'sqlean.py ~= 3.50',
 
@@ -117,7 +117,7 @@ class Project(ProjectBase):
         ],
 
         'testing': [
-            'pytest ~= 8.4',
+            'pytest ~= 9.0',
         ],
     }
 

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/argparse/cli.py

@@ -10,6 +10,7 @@ TODO:
  - pre-run, post-run hooks
  - exitstack?
  - suggestion - difflib.get_close_matches
+ - add_argument_group - group kw on ArgparseKwarg?
 """
 import argparse
 import dataclasses as dc
@@ -36,6 +37,7 @@ ArgparseCmdFn = ta.Callable[[], ta.Optional[int]]  # ta.TypeAlias
 class ArgparseArg:
     args: ta.Sequence[ta.Any]
     kwargs: ta.Mapping[str, ta.Any]
+    group: ta.Optional[str] = None
     dest: ta.Optional[str] = None
 
     def __get__(self, instance, owner=None):
@@ -45,7 +47,11 @@ class ArgparseArg:
 
 
 def argparse_arg(*args, **kwargs) -> ArgparseArg:
-    return ArgparseArg(args, kwargs)
+    return ArgparseArg(
+        args=args,
+        group=kwargs.pop('group', None),
+        kwargs=kwargs,
+    )
 
 
 def argparse_arg_(*args, **kwargs) -> ta.Any:
@@ -215,6 +221,10 @@ class ArgparseCli:
                     subparser.set_defaults(_cmd=obj)
 
             elif isinstance(obj, ArgparseArg):
+                if obj.group is not None:
+                    # FIXME: add_argument_group
+                    raise NotImplementedError
+
                 if att in anns:
                     ann_kwargs = _get_argparse_arg_ann_kwargs(anns[att])
                     obj.kwargs = {**ann_kwargs, **obj.kwargs}
@@ -260,7 +270,7 @@ class ArgparseCli:
 
         if self._unknown_args and not (cmd is not None and cmd.accepts_unknown):
             msg = f'unrecognized arguments: {" ".join(self._unknown_args)}'
-            if (parser := self.get_parser()).exit_on_error:
+            if (parser := self.get_parser()).exit_on_error:  # noqa
                 parser.error(msg)
             else:
                 raise argparse.ArgumentError(None, msg)
@@ -280,7 +290,10 @@ class ArgparseCli:
         return fn()
 
     def cli_run_and_exit(self) -> ta.NoReturn:
-        sys.exit(rc if isinstance(rc := self.cli_run(), int) else 0)
+        rc = self.cli_run()
+        if not isinstance(rc, int):
+            rc = 0
+        raise SystemExit(rc)
 
     def __call__(self, *, exit: bool = False) -> ta.Optional[int]:  # noqa
         if exit:

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/bootstrap/diag.py

@@ -13,23 +13,14 @@ from .base import ContextBootstrap
 from .base import SimpleBootstrap
 
 
-if ta.TYPE_CHECKING:
+with lang.auto_proxy_import(globals()):
     import cProfile  # noqa
     import pstats
 
-    from ..diag import debug as ddbg
-    from ..diag import pycharm as diagpc
-    from ..diag import replserver as diagrs
-    from ..diag import threads as diagt
-
-else:
-    cProfile = lang.proxy_import('cProfile')  # noqa
-    pstats = lang.proxy_import('pstats')
-
-    ddbg = lang.proxy_import('..diag.debug', __package__)
-    diagpc = lang.proxy_import('..diag.pycharm', __package__)
-    diagrs = lang.proxy_import('..diag.replserver', __package__)
-    diagt = lang.proxy_import('..diag.threads', __package__)
+    from ..diag import debug as d_debug
+    from ..diag import pycharm as d_pycharm
+    from ..diag import replserver
+    from ..diag import threads as d_threads
 
 
 ##
@@ -110,7 +101,7 @@ class ThreadDumpBootstrap(ContextBootstrap['ThreadDumpBootstrap.Config']):
     @contextlib.contextmanager
     def enter(self) -> ta.Iterator[None]:
         if self._config.interval_s:
-            tdt = diagt.create_thread_dump_thread(
+            tdt = d_threads.create_thread_dump_thread(
                 interval_s=self._config.interval_s,
                 start=True,
                 nodaemon=self._config.nodaemon,
@@ -119,7 +110,7 @@ class ThreadDumpBootstrap(ContextBootstrap['ThreadDumpBootstrap.Config']):
             tdt = None
 
         if self._config.on_sigquit:
-            dump_threads_str = diagt.dump_threads_str
+            dump_threads_str = d_threads.dump_threads_str
 
             def handler(signum, frame):
                 print(dump_threads_str(), file=sys.stderr)
@@ -153,7 +144,7 @@ class TimebombBootstrap(ContextBootstrap['TimebombBootstrap.Config']):
             yield
             return
 
-        tbt = diagt.create_timebomb_thread(
+        tbt = d_threads.create_timebomb_thread(
             self._config.delay_s,
             start=True,
         )
@@ -173,8 +164,8 @@ class PycharmBootstrap(SimpleBootstrap['PycharmBootstrap.Config']):
 
     def run(self) -> None:
         if self._config.debug is not None:
-            prd = diagpc.PycharmRemoteDebugger.parse(self._config.debug)
-            diagpc.pycharm_remote_debugger_attach(prd)
+            prd = d_pycharm.PycharmRemoteDebugger.parse(self._config.debug)
+            d_pycharm.pycharm_remote_debugger_attach(prd)
 
 
 ##
@@ -190,7 +181,7 @@ class ReplServerBootstrap(ContextBootstrap['ReplServerBootstrap.Config']):
         if self._config.path is None:
             return
 
-        with diagrs.ReplServer(diagrs.ReplServer.Config(
+        with replserver.ReplServer(replserver.ReplServer.Config(
             path=self._config.path,
         )) as rs:
             thread = threading.Thread(target=rs.run, name='replserver')
@@ -212,5 +203,5 @@ class DebugBootstrap(ContextBootstrap['DebugBootstrap.Config']):
         if not self._config.enable:
             return
 
-        with ddbg.debugging_on_exception():
+        with d_debug.debugging_on_exception():
             yield

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/bootstrap/main.py

@@ -1,9 +1,9 @@
+# ruff: noqa: UP006 UP007 UP045
 """
 TODO:
  - -x / --exec - os.exec entrypoint
  - refuse to install non-exec-relevant Bootstraps when chosen
 """
-# ruff: noqa: UP006 UP007 UP045
 import argparse
 import dataclasses as dc
 import io
@@ -19,12 +19,9 @@ from .harness import BOOTSTRAP_TYPES_BY_NAME
 from .harness import bootstrap
 
 
-if ta.TYPE_CHECKING:
+with lang.auto_proxy_import(globals()):
     import runpy
 
-else:
-    runpy = lang.proxy_import('runpy')
-
 
 ##
 

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/bootstrap/sys.py

@@ -2,14 +2,10 @@
 import contextlib
 import dataclasses as dc
 import enum
-import faulthandler
 import gc
 import importlib
 import logging
 import os
-import pwd
-import resource
-import signal
 import sys
 import typing as ta
 
@@ -19,18 +15,17 @@ from .base import ContextBootstrap
 from .base import SimpleBootstrap
 
 
-if ta.TYPE_CHECKING:
+with lang.auto_proxy_import(globals()):
+    import faulthandler
+    import pwd
+    import resource
+    import signal
+
     from .. import libc
     from ..formats import dotenv
     from ..logs import all as logs
     from ..os.pidfiles import pidfile
 
-else:
-    libc = lang.proxy_import('..libc', __package__)
-    logs = lang.proxy_import('..logs', __package__)
-    dotenv = lang.proxy_import('..formats.dotenv', __package__)
-    pidfile = lang.proxy_import('..os.pidfiles.pidfile', __package__)
-
 
 ##
 
@@ -184,14 +179,16 @@ class FaulthandlerBootstrap(ContextBootstrap['FaulthandlerBootstrap.Config']):
 ##
 
 
-SIGNALS_BY_NAME = {
-    a[len('SIG'):]: v  # noqa
-    for a in dir(signal)
-    if a.startswith('SIG')
-    and not a.startswith('SIG_')
-    and a == a.upper()
-    and isinstance((v := getattr(signal, a)), int)
-}
+@lang.cached_function
+def signals_by_name() -> ta.Mapping[str, int]:
+    return {
+        a[len('SIG'):]: v  # noqa
+        for a in dir(signal)
+        if a.startswith('SIG')
+        and not a.startswith('SIG_')
+        and a == a.upper()
+        and isinstance((v := getattr(signal, a)), int)
+    }
 
 
 class PrctlBootstrap(SimpleBootstrap['PrctlBootstrap.Config']):
@@ -208,20 +205,22 @@ class PrctlBootstrap(SimpleBootstrap['PrctlBootstrap.Config']):
             if isinstance(self._config.deathsig, int):
                 sig = self._config.deathsig
             else:
-                sig = SIGNALS_BY_NAME[self._config.deathsig.upper()]
+                sig = signals_by_name()[self._config.deathsig.upper()]
             libc.prctl(libc.PR_SET_PDEATHSIG, sig, 0, 0, 0, 0)
 
 
 ##
 
 
-RLIMITS_BY_NAME = {
-    a[len('RLIMIT_'):]: v  # noqa
-    for a in dir(resource)
-    if a.startswith('RLIMIT_')
-    and a == a.upper()
-    and isinstance((v := getattr(resource, a)), int)
-}
+@lang.cached_function
+def rlimits_by_name() -> ta.Mapping[str, int]:
+    return {
+        a[len('RLIMIT_'):]: v  # noqa
+        for a in dir(resource)
+        if a.startswith('RLIMIT_')
+        and a == a.upper()
+        and isinstance((v := getattr(resource, a)), int)
+    }
 
 
 class RlimitBootstrap(ContextBootstrap['RlimitBootstrap.Config']):
@@ -240,7 +239,7 @@ class RlimitBootstrap(ContextBootstrap['RlimitBootstrap.Config']):
 
         prev = {}
         for k, (s, h) in self._config.limits.items():
-            i = RLIMITS_BY_NAME[k.upper()]
+            i = rlimits_by_name()[k.upper()]
             prev[i] = resource.getrlimit(i)
             resource.setrlimit(i, (or_infin(s), or_infin(h)))
 

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/collections/__init__.py

@@ -8,7 +8,9 @@ with _lang.auto_proxy_init(globals()):
 
     from .attrregistry import (  # noqa
         AttrRegistry,
+
         AttrRegistryCache,
+        SimpleAttrRegistryCache,
     )
 
     from .bimap import (  # noqa
@@ -17,6 +19,8 @@ with _lang.auto_proxy_init(globals()):
         make_bi_map,
     )
 
+    from . import cache  # noqa
+
     from .coerce import (  # noqa
         abs_set,
         abs_set_of,

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/collections/attrregistry.py

@@ -1,3 +1,7 @@
+"""
+TODO:
+ - lock?
+"""
 import dataclasses as dc
 import typing as ta
 import weakref
@@ -50,13 +54,32 @@ class AttrRegistry(ta.Generic[K, V]):
     def add_invalidate_callback(self, callback: ta.Callable[[], None]) -> None:
         self._invalidate_callbacks.append(callback)
 
+    @ta.overload
     def register(self, obj: K, val: V) -> None:
-        check.not_in(obj, self._objs)
+        ...
 
-        self._objs[obj] = val
+    @ta.overload
+    def register(self, val: V) -> ta.Callable[[T], T]:
+        ...
 
-        for iv in self._invalidate_callbacks:
-            iv()
+    def register(self, *args):
+        def inner(obj, val):
+            check.not_in(obj, self._objs)
+
+            self._objs[obj] = val
+
+            for iv in self._invalidate_callbacks:
+                iv()
+
+            return obj
+
+        if len(args) == 1:
+            return lambda obj: inner(obj, args[0])
+        elif len(args) == 2:
+            inner = inner(*args)
+            return None
+        else:
+            raise TypeError(args)
 
     def _lookup(self, obj: ta.Any) -> lang.Maybe[V]:
         if not self._identity:
@@ -180,3 +203,8 @@ class AttrRegistryCache(ta.Generic[K, V, T]):
         out = self._prepare(instance_cls, collected)
         self._cache[weakref.ref(instance_cls, self._cache_remove)] = out
         return out
+
+
+class SimpleAttrRegistryCache(AttrRegistryCache[K, V, dict[str, tuple[K, V]]], ta.Generic[K, V]):
+    def __init__(self, registry: AttrRegistry[K, V]) -> None:
+        super().__init__(registry, lambda _, dct: dct)

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/concurrent/threadlets.py

@@ -75,7 +75,7 @@ class GreenletThreadlet(Threadlet):
 
     @property
     def parent(self) -> ta.Optional['GreenletThreadlet']:
-        return GreenletThreadlet(self.g.parent)
+        return GreenletThreadlet(p) if (p := self.g.parent) is not None else None
 
     @property
     def dead(self) -> bool:

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/daemons/reparent.py

@@ -10,13 +10,12 @@ def reparent_process(
         no_close_stdio: bool = False,
 ) -> None:
     if (pid := os.fork()):  # noqa
-        sys.exit(0)
-        raise RuntimeError('Unreachable')  # noqa
+        raise SystemExit(0)
 
     os.setsid()
 
     if (pid := os.fork()):  # noqa
-        sys.exit(0)
+        raise SystemExit(0)
 
     if not no_close_stdio:
         rn_fd = os.open('/dev/null', os.O_RDONLY)

{omlish-0.0.0.dev470 → omlish-0.0.0.dev476}/omlish/daemons/spawning.py

@@ -2,7 +2,6 @@ import abc
 import enum
 import functools
 import os
-import sys
 import threading
 import typing as ta
 
@@ -166,9 +165,9 @@ class ForkSpawner(Spawner, dc.Frozen):
         try:
             spawn.fn()
         except BaseException:  # noqa
-            sys.exit(1)
+            raise SystemExit(1) from None
         else:
-            sys.exit(0)
+            raise SystemExit(0)
 
         raise RuntimeError('Unreachable')  # noqa
 

omlish-0.0.0.dev476/omlish/dataclasses/impl/configs.py

@@ -0,0 +1,117 @@
+"""
+TODO:
+ - better support anonymous / weakref'd / unlaoded modules / etc.
+  - anonymous=False on class spec?
+"""
+import dataclasses as dc
+import threading
+import typing as ta
+
+from ... import check
+
+
+##
+
+
+@dc.dataclass(frozen=True, kw_only=True)
+class PackageConfig:
+    codegen: bool = False
+
+    def __init_subclass__(cls, **kwargs):
+        raise TypeError
+
+
+DEFAULT_PACKAGE_CONFIG = PackageConfig()
+
+
+##
+
+
+@ta.final
+class PackageConfigCache:
+    def __init__(self) -> None:
+        super().__init__()
+
+        self._lock = threading.Lock()
+        self._root = PackageConfigCache._Node('', None)
+        self._nodes: dict[str, PackageConfigCache._Node] = {}
+
+    @ta.final
+    class _Node:
+        def __init__(self, pkg: str, cfg: PackageConfig | None) -> None:
+            self.pkg = pkg
+            self.cfg = cfg
+
+            self.children: dict[str, PackageConfigCache._Node] = {}
+
+        def __repr__(self) -> str:
+            return f'{self.__class__.__name__}(pkg={self.pkg!r}, cfg={self.cfg!r})'
+
+    def _navigate(self, *parts: str) -> _Node:
+        node = self._root
+        for i, p in enumerate(parts):
+            if (child := node.children.get(p)) is None:
+                child_pkg = '.'.join(parts[:i + 1])
+                check.not_in(child_pkg, self._nodes)
+                child = node.children[p] = self._nodes[child_pkg] = PackageConfigCache._Node(child_pkg, node.cfg)
+            node = child
+        return node
+
+    def put(self, pkg: str, cfg: PackageConfig) -> None:
+        check.non_empty_str(pkg)
+        with self._lock:
+            check.not_in(pkg, self._nodes)
+            parts = pkg.split('.')
+            parent = self._navigate(*parts[:-1])
+            check.not_in(parts[-1], parent.children)
+            parent.children[parts[-1]] = self._nodes[pkg] = PackageConfigCache._Node(pkg, cfg)
+
+    def get(self, pkg: str) -> PackageConfig | None:
+        if not pkg:
+            return None
+
+        try:
+            node = self._nodes[pkg]
+        except KeyError:
+            pass
+        else:
+            return node.cfg
+
+        # Flimsy - if no config anywhere for root package then don't cache. Want to support unlimited anonymous modules
+        # which may be unloaded without polluting cache forever.
+        parts = pkg.split('.')
+        if parts[0] not in self._root.children:
+            return None
+
+        with self._lock:
+            try:
+                node = self._nodes[pkg]
+            except KeyError:
+                pass
+            else:
+                return node.cfg
+
+            node = self._navigate(*pkg.split('.'))
+            return node.cfg
+
+
+PACKAGE_CONFIG_CACHE = PackageConfigCache()
+
+
+##
+
+
+def init_package(
+        init_globals: ta.MutableMapping[str, ta.Any],
+        *,
+        codegen: bool = False,
+) -> None:
+    pkg = check.non_empty_str(init_globals['__package__'])
+    if init_globals['__name__'] not in (pkg, '__main__'):
+        raise NameError('Must call dataclasses.init_package from __init__')
+
+    pkg_cfg = PackageConfig(
+        codegen=codegen,
+    )
+
+    PACKAGE_CONFIG_CACHE.put(pkg, pkg_cfg)