private-attribute-cpp 2.1.2__tar.gz → 2.1.3__tar.gz

{private_attribute_cpp-2.1.2/private_attribute_cpp.egg-info → private_attribute_cpp-2.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: private_attribute_cpp
-Version: 2.1.2
+Version: 2.1.3
 Summary: A Python package that provides a way to define private attributes in C++ implementation.
 Home-page: https://github.com/Locked-chess-official/private_attribute_cpp
 Author: HuangHaoHua

{private_attribute_cpp-2.1.2 → private_attribute_cpp-2.1.3}/private_attribute.cpp

@@ -317,11 +317,11 @@ attr_classify(uintptr_t typ_id, const std::string& name, PyCodeObject* code) noe
         chain.insert(chain.end(), parent_ids.begin(), parent_ids.end());
     }
 
-    // Find the class whose code is currently running (the code owner).
+    // Find the class whose code is currently running (the code owner, parent first).
     uintptr_t code_class = 0;
-    for (auto& node : chain) {
-        if (is_class_code(node, code)) {
-            code_class = node;
+    for (auto it = chain.rbegin(); it != chain.rend(); ++it) {
+        if (is_class_code(*it, code)) {
+            code_class = *it;  // first hit from the base-most end
             break;
         }
     }

{private_attribute_cpp-2.1.2 → private_attribute_cpp-2.1.3/private_attribute_cpp.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: private_attribute_cpp
-Version: 2.1.2
+Version: 2.1.3
 Summary: A Python package that provides a way to define private attributes in C++ implementation.
 Home-page: https://github.com/Locked-chess-official/private_attribute_cpp
 Author: HuangHaoHua

{private_attribute_cpp-2.1.2 → private_attribute_cpp-2.1.3}/private_attribute_cpp.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ private_attribute_cpp.egg-info/SOURCES.txt
 private_attribute_cpp.egg-info/dependency_links.txt
 private_attribute_cpp.egg-info/not-zip-safe
 private_attribute_cpp.egg-info/top_level.txt
+tests/test_attr_classify_owner.py
 tests/test_class_private_attrs.py
 tests/test_private_abc.py
 tests/test_private_attrs.py

{private_attribute_cpp-2.1.2 → private_attribute_cpp-2.1.3}/setup.py

@@ -18,7 +18,7 @@ readme = open('README.md').read()
 
 setup(
     name='private_attribute_cpp',
-    version='2.1.2',
+    version='2.1.3',
     author="HuangHaoHua",
     author_email="13140752715@example.com",
     description='A Python package that provides a way to define private attributes in C++ implementation.',

private_attribute_cpp-2.1.3/tests/test_attr_classify_owner.py

@@ -0,0 +1,137 @@
+"""attr_classify ownership rule (v2.1.3).
+
+When a code object is owned by several classes on the MRO chain (e.g. a
+subclass aliases a parent method with `f = g`, or copies a parent method's
+bytecode with `f.__code__ = g.__code__`), attr_classify must attribute the
+code to the base-most (declaring, parent) class. Otherwise the parent code's
+private attribute accesses get split across two storages: the tampered method
+writes the subclass's storage while the untampered parent method reads the
+parent's storage, corrupting the parent's behavior.
+
+Key invariant: parent methods inc() (tampered by the child) and peek() (not
+tampered) must ALWAYS see the same storage on a child instance.
+"""
+import unittest
+
+import private_attribute
+
+
+class Parent(private_attribute.PrivateAttrBase):
+    __private_attrs__ = ["_n"]
+    _n = 0
+
+    def inc(self):
+        self._n += 1
+        return self._n
+
+    def peek(self):
+        return self._n
+
+
+class TestAliasCodeOwnership(unittest.TestCase):
+    """`f = g` in the class body registers the same function object with both
+    classes; the aliased and the untampered parent methods must stay
+    consistent on a child instance."""
+
+    def test_alias_method_sees_same_storage_as_untampered(self):
+        class AliasChild(Parent):
+            __private_attrs__ = ["_n"]
+            _n = 100
+            f = Parent.inc          # same function object in both class dicts
+
+        c = AliasChild()
+        v1 = c.f()                  # parent code, double-owned
+        v2 = c.peek()               # parent code, single-owned
+        self.assertEqual(v1, v2)
+        c.f()
+        self.assertEqual(c.peek(), v2 + 1)
+        self.assertEqual(Parent().peek(), 0)
+
+    def test_alias_without_shadow_works(self):
+        class AliasChild2(Parent):
+            __private_attrs__ = []
+            f = Parent.inc
+
+        c = AliasChild2()
+        v1 = c.f()
+        self.assertEqual(v1, c.peek())
+
+
+class TestCodeSwapOwnership(unittest.TestCase):
+    """`f.__code__ = g.__code__` inside the class body; the swapped method
+    must behave exactly like the parent's method (same storage)."""
+
+    def test_code_swap_sees_same_storage_as_untampered(self):
+        class CodeChild(Parent):
+            __private_attrs__ = ["_n"]
+            _n = 100
+
+            def f(self):
+                return None
+
+            f.__code__ = Parent.inc.__code__
+
+        c = CodeChild()
+        v1 = c.f()
+        v2 = c.peek()
+        self.assertEqual(v1, v2)
+        c.f()
+        self.assertEqual(c.peek(), v2 + 1)
+
+
+class TestFallbackSemantics(unittest.TestCase):
+    """Per-subject fallback is preserved: an unwritten read resolves to the
+    subject's own class-level default, consistently for both methods."""
+
+    def test_unwritten_read_falls_back_per_subject(self):
+        class FreshChild(Parent):
+            __private_attrs__ = ["_n"]
+            _n = 100
+            f = Parent.inc
+
+        c = FreshChild()
+        self.assertEqual(c.peek(), 100)
+        v1 = c.f()
+        self.assertEqual(v1, c.peek())
+
+
+class TestNormalUsageRegression(unittest.TestCase):
+    """Without tampering, nothing changes: child's own code uses child
+    storage, parent's code keeps its per-subject view."""
+
+    def test_child_own_code_uses_child_storage(self):
+        class NormChild(Parent):
+            __private_attrs__ = ["_n"]
+            _n = 100
+
+            def own_inc(self):
+                self._n += 1
+                return self._n
+
+        c = NormChild()
+        self.assertEqual(c.own_inc(), 101)
+        self.assertEqual(c.peek(), 100)
+        self.assertEqual(Parent().peek(), 0)
+
+
+class TestIsolationIntact(unittest.TestCase):
+    """The v2.1.0 isolation rule still holds: the subclass's own code must
+    not touch a parent's private attribute."""
+
+    def test_subclass_own_code_denied(self):
+        class IsoParent(private_attribute.PrivateAttrBase):
+            __private_attrs__ = ["_secret"]
+            _secret = 1
+
+        class IsoChild(IsoParent):
+            __private_attrs__ = []
+
+            def read_secret(self):
+                return self._secret
+
+        with self.assertRaises(AttributeError):
+            IsoChild().read_secret()
+
+
+if __name__ == "__main__":
+    unittest.main()