omlish 0.0.0.dev327__py3-none-any.whl → 0.0.0.dev328__py3-none-any.whl

omlish/__about__.py

@@ -1,5 +1,5 @@
-__version__ = '0.0.0.dev327'
-__revision__ = 'ce58b88c3e3c1e9bd7b09f89667cb4047b61684c'
+__version__ = '0.0.0.dev328'
+__revision__ = '80b9ed17dd2aefd7dd86d4aa2f14fac4281cf8d6'
 
 
 #
@@ -113,7 +113,7 @@ class Project(ProjectBase):
         ],
 
         'testing': [
-            'pytest ~= 8.3',
+            'pytest ~= 8.4',
         ],
     }
 

omlish/collections/__init__.py

@@ -59,7 +59,7 @@ from .hasheq import (  # noqa
 from .identity import (  # noqa
     IdentityKeyDict,
     IdentitySet,
-    IdentityWrapper,
+    IdentityWeakKeyDictionary,
     IdentityWeakSet,
 )
 

omlish/collections/identity.py

@@ -13,29 +13,6 @@ V = ta.TypeVar('V')
 ##
 
 
-class IdentityWrapper(ta.Generic[T]):
-    def __init__(self, value: T) -> None:
-        super().__init__()
-
-        self._value = value
-
-    def __repr__(self) -> str:
-        return lang.attr_repr(self, 'value')
-
-    @property
-    def value(self) -> T:
-        return self._value
-
-    def __eq__(self, other):
-        return isinstance(other, IdentityWrapper) and other._value is self._value
-
-    def __ne__(self, other):
-        return not (self == other)
-
-    def __hash__(self):
-        return id(self._value)
-
-
 class IdentityKeyDict(ta.MutableMapping[K, V]):
     def __init__(self, *args, **kwargs) -> None:
         super().__init__()
@@ -115,31 +92,90 @@ class IdentitySet(ta.MutableSet[T]):
         return iter(self._dict.values())
 
 
+##
+
+
+class IdentityWeakKeyDictionary(ta.MutableMapping[K, V]):
+    """
+    https://ideone.com/G4iIri
+    https://stackoverflow.com/questions/75314250/python-weakkeydictionary-for-unhashable-types#comment135919973_77100606
+
+    See also:
+    https://github.com/python-trio/trio/blob/efd785a20721707b52a6e2289a65e25722b30c96/src/trio/_core/_ki.py#L81
+    """
+
+    def __init__(self, *args: ta.Any, **kwargs: ta.Any) -> None:
+        super().__init__()
+
+        self._keys: ta.MutableMapping[IdentityWeakKeyDictionary._Id[K], K] = weakref.WeakValueDictionary()
+        self._values: ta.MutableMapping[IdentityWeakKeyDictionary._Id[K], V] = weakref.WeakKeyDictionary()
+
+        for k, v in lang.yield_dict_init(*args, **kwargs):
+            self[k] = v
+
+    def __len__(self) -> int:
+        return len(self._keys)
+
+    def __iter__(self) -> ta.Iterator[K]:
+        return iter(self._keys.values())
+
+    class _Id(ta.Generic[T]):
+        def __init__(self, key: T) -> None:
+            super().__init__()
+
+            self._id = id(key)
+            self._key_ref = weakref.ref(key)
+
+        def __repr__(self) -> str:
+            return f'{self.__class__.__name__}<id={self._id}>'
+
+        def __hash__(self) -> int:
+            return self._id
+
+        def __eq__(self, other: object) -> bool:
+            return (
+                type(other) is type(self) and
+                self._id == other._id and  # type: ignore  # noqa
+                self._key_ref() is other._key_ref()  # type: ignore  # noqa
+            )
+
+        def __ne__(self, other: object) -> bool:
+            return not (self == other)
+
+    def __getitem__(self, key: K) -> V:
+        return self._values.__getitem__(self._Id(key))
+
+    def __setitem__(self, key: K, value: V) -> None:
+        id_obj = self._Id(key)
+        self._keys.__setitem__(id_obj, key)
+        self._values.__setitem__(id_obj, value)
+
+    def __delitem__(self, key: K) -> None:
+        self._values.__delitem__(self._Id(key))
+        self._keys.__delitem__(self._Id(key))
+
+
 class IdentityWeakSet(ta.MutableSet[T]):
     def __init__(self, init: ta.Iterable[T] | None = None) -> None:
         super().__init__()
 
-        self._dict: weakref.WeakValueDictionary[int, T] = weakref.WeakValueDictionary()
+        self._dict: IdentityWeakKeyDictionary[T, None] = IdentityWeakKeyDictionary()
+
+        if init is not None:
+            for e in init:
+                self._dict[e] = None
 
     def add(self, value):
-        # FIXME: race with weakref callback?
-        self._dict[id(value)] = value
+        self._dict[value] = None
 
     def discard(self, value):
-        try:
-            del self._dict[id(value)]
-        except KeyError:
-            pass
+        del self._dict[value]
 
     def __contains__(self, x):
-        try:
-            o = self._dict[id(x)]
-        except KeyError:
-            return False
-        return x is o
+        return x in self._dict
 
     def __len__(self):
         return len(self._dict)
 
     def __iter__(self):
-        return self._dict.values()
+        return self._dict.keys()

omlish/dataclasses/api/classes/decorator.py

@@ -49,6 +49,7 @@ def dataclass(
         cache_hash: bool | None = None,
         generic_init: bool | None = None,
         override: bool | None = None,
+        allow_dynamic_dunder_attrs: bool | None = None,
 
         repr_id: bool | None = None,
         terse_repr: bool | None = None,
@@ -153,6 +154,7 @@ def dataclass(
                 cache_hash=cache_hash,
                 generic_init=generic_init,
                 override=override,
+                allow_dynamic_dunder_attrs=allow_dynamic_dunder_attrs,
 
                 repr_id=repr_id,
                 terse_repr=terse_repr,

omlish/dataclasses/api/classes/make.py

@@ -39,6 +39,7 @@ def make_dataclass(  # noqa
         cache_hash: bool | None = None,
         generic_init: bool | None = None,
         override: bool | None = None,
+        allow_dynamic_dunder_attrs: bool | None = None,
 
         repr_id: bool | None = None,
         terse_repr: bool | None = None,
@@ -109,6 +110,7 @@ def make_dataclass(  # noqa
         cache_hash=cache_hash,
         generic_init=generic_init,
         override=override,
+        allow_dynamic_dunder_attrs=allow_dynamic_dunder_attrs,
 
         repr_id=repr_id,
         terse_repr=terse_repr,

omlish/dataclasses/concerns/frozen.py

@@ -65,9 +65,10 @@ def check_frozen_bases(cls: type, frozen: bool) -> None:
 ##
 
 
-@dc.dataclass(frozen=True)
+@dc.dataclass(frozen=True, kw_only=True)
 class FrozenPlan(Plan):
     fields: tuple[str, ...]
+    allow_dynamic_dunder_attrs: bool
 
 
 @register_generator_type(FrozenPlan)
@@ -78,7 +79,10 @@ class FrozenGenerator(Generator[FrozenPlan]):
         if not ctx.cs.frozen:
             return None
 
-        return PlanResult(FrozenPlan(tuple(f.name for f in ctx.cs.fields)))
+        return PlanResult(FrozenPlan(
+            fields=tuple(f.name for f in ctx.cs.fields),
+            allow_dynamic_dunder_attrs=ctx.cs.allow_dynamic_dunder_attrs,
+        ))
 
     def _generate_one(
             self,
@@ -88,8 +92,20 @@ class FrozenGenerator(Generator[FrozenPlan]):
             exc_args: str,
     ) -> AddMethodOp:
         preamble = []
+        condition = []
+
         # https://github.com/python/cpython/commit/ee6f8413a99d0ee4828e1c81911e203d3fff85d5
-        condition = f'type(self) is {CLS_IDENT}'
+        base_condition = f'type(self) is {CLS_IDENT}'
+
+        if plan.allow_dynamic_dunder_attrs:
+            condition.extend([
+                f'(',
+                f'    {base_condition}',
+                f'    and not (len(name) > 4 and name[:2] == name[-2:] == "__")',
+                f')',
+            ])
+        else:
+            condition.append(base_condition)
 
         if plan.fields:
             set_ident = f'{IDENT_PREFIX}_{mth}_frozen_fields'
@@ -102,14 +118,19 @@ class FrozenGenerator(Generator[FrozenPlan]):
                 f'}}',
                 f'',
             ])
-            condition += f' or name in {set_ident}'
+            condition.append(f' or name in {set_ident}')
 
         return AddMethodOp(
             f'__{mth}__',
             '\n'.join([
                 *preamble,
                 f'def __{mth}__(self, {", ".join(params)}):',
-                f'    if {condition}:',
+                f'    if (',
+                *[
+                    f'        {l}'
+                    for l in condition
+                ],
+                f'    ):',
                 f'        raise {FROZEN_INSTANCE_ERROR_GLOBAL.ident}{exc_args}',
                 f'    super({CLS_IDENT}, self).__{mth}__({", ".join(params)})',
             ]),

omlish/dataclasses/specs.py

@@ -183,6 +183,7 @@ class ClassSpec(lang.Final):
     cache_hash: bool = False
     generic_init: bool = False
     override: bool = False
+    allow_dynamic_dunder_attrs: bool = False
 
     repr_id: bool = False
     terse_repr: bool = False

omlish/lang/objects.py

@@ -250,3 +250,8 @@ class Identity(ta.Generic[T]):
         if type(other) is not type(self):
             return NotImplemented
         return self._obj is other._obj  # noqa
+
+    def __ne__(self, other):
+        if type(other) is not type(self):
+            return NotImplemented
+        return self._obj is not other._obj  # noqa

{omlish-0.0.0.dev327.dist-info → omlish-0.0.0.dev328.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omlish
-Version: 0.0.0.dev327
+Version: 0.0.0.dev328
 Summary: omlish
 Author: wrmsr
 License: BSD-3-Clause
@@ -44,7 +44,7 @@ Requires-Dist: sqlean.py~=3.49; extra == "all"
 Requires-Dist: duckdb~=1.3; extra == "all"
 Requires-Dist: markupsafe~=3.0; extra == "all"
 Requires-Dist: jinja2~=3.1; extra == "all"
-Requires-Dist: pytest~=8.3; extra == "all"
+Requires-Dist: pytest~=8.4; extra == "all"
 Requires-Dist: anyio~=4.9; extra == "all"
 Requires-Dist: sniffio~=1.3; extra == "all"
 Requires-Dist: asttokens~=3.0; extra == "all"
@@ -94,7 +94,7 @@ Provides-Extra: templates
 Requires-Dist: markupsafe~=3.0; extra == "templates"
 Requires-Dist: jinja2~=3.1; extra == "templates"
 Provides-Extra: testing
-Requires-Dist: pytest~=8.3; extra == "testing"
+Requires-Dist: pytest~=8.4; extra == "testing"
 Provides-Extra: plus
 Requires-Dist: anyio~=4.9; extra == "plus"
 Requires-Dist: sniffio~=1.3; extra == "plus"

{omlish-0.0.0.dev327.dist-info → omlish-0.0.0.dev328.dist-info}/RECORD

@@ -1,5 +1,5 @@
 omlish/.manifests.json,sha256=orgsRvtpHu8tdhaCvlP9v3P495OJopYYiHKjK68WtWg,8587
-omlish/__about__.py,sha256=d4Xjb6KLvYaunh9KWRpIDX2rC1dTJogXQi_w5nINTKg,3478
+omlish/__about__.py,sha256=0MrPN5vIAd-Q-O4VVWJF_WDLWv-jHeJBNJZTsBJ_214,3478
 omlish/__init__.py,sha256=SsyiITTuK0v74XpKV8dqNaCmjOlan1JZKrHQv5rWKPA,253
 omlish/c3.py,sha256=rer-TPOFDU6fYq_AWio_AmA-ckZ8JDY5shIzQ_yXfzA,8414
 omlish/cached.py,sha256=MLap_p0rdGoDIMVhXVHm1tsbcWobJF0OanoodV03Ju8,542
@@ -76,14 +76,14 @@ omlish/codecs/funcs.py,sha256=p4imNt7TobyZVXWC-WhntHVu9KfJrO4QwdtPRh-cVOk,850
 omlish/codecs/registry.py,sha256=2FnO5YP7ui1LzkguwESY0MP3WIdwgPTIJTM_4RyTOEg,3896
 omlish/codecs/standard.py,sha256=eiZ4u9ep0XrA4Z_D1zJI0vmWyuN8HLrX4Se_r_Cq_ZM,60
 omlish/codecs/text.py,sha256=JzrdwMpQPo2NBBg3K1EZszzQy5vEWmd82SIerJd4yeQ,5723
-omlish/collections/__init__.py,sha256=7r4qY9paGL-pC8ZNJn6taod9T2qAyMjIpcL7ptb8GaI,2394
+omlish/collections/__init__.py,sha256=_7sct5tPAzV3BadIGYRdz6KzazntZIws9e8fEDlUwsU,2404
 omlish/collections/abc.py,sha256=ikTJlJ5dhXjU6tlNsI0Wm0_7GaIEpe3mftpvdGY_nc8,2620
 omlish/collections/bimap.py,sha256=3szDCscPJlFRtkpyVQNWneg4s50mr6Rd0jdTzVEIcnE,1661
 omlish/collections/coerce.py,sha256=tAls15v_7p5bUN33R7Zbko87KW5toWHl9fRialCqyNY,7030
 omlish/collections/errors.py,sha256=shcS-NCnEUudF8qC_SmO2TQyjivKlS4TDjaz_faqQ0c,44
 omlish/collections/frozen.py,sha256=LMbAHYDENIQk1hvjCTvpnx66m1TalrHa4CSn8n_tsXQ,4142
 omlish/collections/hasheq.py,sha256=uHypfZlHhicQPvx9OOlpT9MSLwfc_mFil-WaxF9dTOo,3732
-omlish/collections/identity.py,sha256=xtoczgBPYzr6r2lJS-eti2kEnN8rVDvNGDCG3TA6vRo,3405
+omlish/collections/identity.py,sha256=S_W508EkU-AW0TZ7cv1wWUc6EG54vww4XbcfGjDFTgg,4793
 omlish/collections/mappings.py,sha256=iXb7oq1rCQak0KgzblgrzWCJLrkfJAYHFvl9lprOVUI,2804
 omlish/collections/ordered.py,sha256=7zTbrAt12rf6i33XHkQERKar258fJacaw_WbtGEBgWo,2338
 omlish/collections/ranked.py,sha256=McB8C2UQfUvrbmxGTpBz1-EZuyCLkBFtktzncMdt8_Y,2287
@@ -141,13 +141,13 @@ omlish/dataclasses/errors.py,sha256=tyv3WR6az66uGGiq9FIuCHvy1Ef-G7zeMY7mMG6hy2Y,
 omlish/dataclasses/inspect.py,sha256=BlpPghVCU3w_YDnONEqqE99YHzJM2q3eoqe39YX25Ko,4596
 omlish/dataclasses/internals.py,sha256=vIGCZnStgD3ef4drYRtVOrxhxmAPa0vJpo4pXcDcQvM,3073
 omlish/dataclasses/reflection.py,sha256=FSSTiS7pGTmDrYNkS6fZzaMSe8qtSvDv5Pgz0T55qXg,2404
-omlish/dataclasses/specs.py,sha256=r9hpIWy83ODiB1n7wTf7JL5fiCeeT8fWbMvxs7c4e3g,6148
+omlish/dataclasses/specs.py,sha256=JFpMILJhxiWYmfTD4TDl03O13o1i0Rwgf14WpI7U9go,6193
 omlish/dataclasses/utils.py,sha256=gv6za6oJYBr1VaeGd7oXqq9lhBs_sxkpC27XYzJggno,1870
 omlish/dataclasses/api/__init__.py,sha256=k5iS9QOwf_f4iOfGffYhnqDOcmEIwEUUTp00u11kIPM,455
 omlish/dataclasses/api/classes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omlish/dataclasses/api/classes/conversion.py,sha256=U3g5S-HefJrg4to1BF8XnlEWqMxFeGEqVt8o5OTQK-U,902
-omlish/dataclasses/api/classes/decorator.py,sha256=q-m05j6VhJ9_00TmrXBvwmv0_TFPTD8OURi1KbKo9-4,3933
-omlish/dataclasses/api/classes/make.py,sha256=Q_8ZVjhhFEUstIZA3Ht_vlGptYjTcJqX48qT57W7SZw,2820
+omlish/dataclasses/api/classes/decorator.py,sha256=swqe6wZ3nAb9kZaYvxXJNhr-KiOPkEERCCV4Zu2SlfA,4060
+omlish/dataclasses/api/classes/make.py,sha256=qcsELy9tpwMLr0RJmvlSgR0deh40nvxgy7RaG3SJ_nw,2939
 omlish/dataclasses/api/classes/metadata.py,sha256=ocUu75VWOdZA8-R7K95yPhQtS0sFZB1NtqCdzXyDNeQ,2769
 omlish/dataclasses/api/classes/params.py,sha256=GmLJotHT4BuXcK9lAwP00boR0_mfM0xITWBFKrk0brY,1994
 omlish/dataclasses/api/fields/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -161,7 +161,7 @@ omlish/dataclasses/concerns/copy.py,sha256=U8kSHEvVupvXArS3OPB7GK_M7aPO9R3_Uk1nq
 omlish/dataclasses/concerns/doc.py,sha256=SL0hj4EzXrVOhaDzHVDGnzylkOU7mcXtr9Tm3wuZnQQ,1344
 omlish/dataclasses/concerns/eq.py,sha256=PI0KKzJyIziRLgcjl8efjapf8baaRCdcuyGW83bCpEo,1720
 omlish/dataclasses/concerns/fields.py,sha256=-VnI7al9sKRs0xTCy-IUhl3EgCKBX5vczDUZNu72_KY,3311
-omlish/dataclasses/concerns/frozen.py,sha256=IWfar9VIkMezE8Bh9duB5vVXy1qyABa1YEOwHVJOCqE,3872
+omlish/dataclasses/concerns/frozen.py,sha256=_skp52Rknas_WaSm2UAzHPX44L2dtM1CMq1478GNz8M,4506
 omlish/dataclasses/concerns/hash.py,sha256=ouX8zyae_IUAMEd-0dvUAPO9bbp5AM1evumyUSr5inY,4815
 omlish/dataclasses/concerns/init.py,sha256=_Yv4g7paBtkxnm3CgglBSAOQnoRAULv9HwvZg_od3WM,14542
 omlish/dataclasses/concerns/matchargs.py,sha256=maHlfPCcHvAOGuXoLGu66ZizHdizU-UIrFCYbNMPR5g,748
@@ -418,7 +418,7 @@ omlish/lang/functions.py,sha256=qNqzWF6vI6PGTzKhgkhnNXP8e1gSerb8GsHG3_wVBP8,6386
 omlish/lang/generators.py,sha256=a4D5HU_mySs2T2z3xCmE_s3t4QJkj0YRrK4-hhpGd0A,5197
 omlish/lang/imports.py,sha256=y9W9Y-d_cQ35QCLuSIPoa6vnEqSErFCz8b-34IH128U,10552
 omlish/lang/iterables.py,sha256=StoGp9yaP3njdLKHoWYcEevO3eE8SHEPYl5_avZob24,2149
-omlish/lang/objects.py,sha256=nbxBHfQHVw0OG4qeSTP2GvIiFIcH2tbbitY8y-mYPPo,5959
+omlish/lang/objects.py,sha256=ZsibJwNp1EXorbaObm9TlCNtuuM65snCmVb7H4_llqI,6116
 omlish/lang/outcomes.py,sha256=mpFy_VoM-b74L1aCFsjsZVUHx_icZ1AHMOKeVesjOp4,8628
 omlish/lang/overrides.py,sha256=IBzK6ljfLX6TLgIyKTSjhqTLcuKRkQNVtEOnBLS4nuA,2095
 omlish/lang/params.py,sha256=sfbNoGrKCsAtubFufj_uh_WKshIgA8fqJ4PmLH1PH00,6639
@@ -857,9 +857,9 @@ omlish/typedvalues/holder.py,sha256=ZTnHiw-K38ciOBLEdwgrltr7Xp8jjEs_0Lp69DH-G-o,
 omlish/typedvalues/marshal.py,sha256=hWHRLcrGav7lvXJDtb9bNI0ickl4SKPQ6F4BbTpqw3A,4219
 omlish/typedvalues/reflect.py,sha256=Ih1YgU-srUjsvBn_P7C66f73_VCvcwqE3ffeBnZBgt4,674
 omlish/typedvalues/values.py,sha256=ym46I-q2QJ_6l4UlERqv3yj87R-kp8nCKMRph0xQ3UA,1307
-omlish-0.0.0.dev327.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omlish-0.0.0.dev327.dist-info/METADATA,sha256=mliaL7sX9zbiogrFKjPXwlZWP3vroB5tneucBGOSeHE,4416
-omlish-0.0.0.dev327.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-omlish-0.0.0.dev327.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
-omlish-0.0.0.dev327.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
-omlish-0.0.0.dev327.dist-info/RECORD,,
+omlish-0.0.0.dev328.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omlish-0.0.0.dev328.dist-info/METADATA,sha256=j4SbGhFfYtXvAosOCJp8sZ1zRExSDAF3wT4rhu7TCQw,4416
+omlish-0.0.0.dev328.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+omlish-0.0.0.dev328.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
+omlish-0.0.0.dev328.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
+omlish-0.0.0.dev328.dist-info/RECORD,,