wizlib 0.2.6__tar.gz → 0.3.0__tar.gz

wizlib-0.3.0/.version

@@ -0,0 +1 @@
+0.3.0

{wizlib-0.2.6/wizlib.egg-info → wizlib-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wizlib
-Version: 0.2.6
+Version: 0.3.0
 Summary: A collection of Python modules that are useful across multiple projects
 Author-email: Francis Potter <wizlib@steampunkwizard.ca>
 License: MIT

wizlib-0.3.0/test/test_super_wrapper.py

@@ -0,0 +1,36 @@
+from unittest import TestCase
+from unittest.mock import patch
+from io import StringIO
+
+from wizlib.super_wrapper import SuperWrapper
+
+
+class TestSuperWrapper(TestCase):
+
+    def test_super_wrapper(self):
+
+        class Parent(SuperWrapper):
+            def execute(self, method, *args, **kwargs):
+                print(f"Parent execute before")
+                method(self, *args, **kwargs)
+                print(f"Parent execute after")
+
+        class InBetween(Parent):
+            @Parent.wrap
+            def execute(self, method, *args, **kwargs):
+                print(f"IB execute before")
+                method(self, *args, **kwargs)
+                print(f"IB execute after")
+
+        class NewChild(InBetween):
+            @InBetween.wrap
+            def execute(self, name):
+                print(f"Hello {name}")
+
+        with patch('sys.stdout', o:=StringIO()):
+            c = NewChild()
+            c.execute("Jane")
+        o.seek(0)
+        r = o.read()
+        self.assertEqual(r, "Parent execute before\nIB execute before\nHello Jane\nIB execute after\nParent execute after\n")
+

wizlib-0.3.0/wizlib/super_wrapper.py

@@ -0,0 +1,34 @@
+# Provide a decorator to wrap a method so that it's called within the inherited
+# version of that method.
+#
+# Example of use:
+#
+# class Parent(SuperWrapper):
+#     def execute(self, method, *args, **kwargs):
+#         print(f"Parent execute before")
+#         method(self, *args, **kwargs)
+#         print(f"Parent execute after")
+#
+# class InBetween(Parent):
+#     @Parent.wrap
+#     def execute(self, method, *args, **kwargs):
+#         print(f"IB execute before")
+#         method(self, *args, **kwargs)
+#         print(f"IB execute after")
+#
+# class NewChild(InBetween):
+#     @InBetween.wrap
+#     def execute(self, name):
+#         print(f"Hello {name}")
+#
+# c = NewChild()
+# c.execute("Jane")
+
+
+class SuperWrapper:
+    @classmethod
+    def wrap(parent_class, method):
+        def wrapper(self, *args, **kwargs):
+            parent_method = getattr(parent_class, method.__name__)
+            return parent_method(self, method, *args, **kwargs)
+        return wrapper

{wizlib-0.2.6 → wizlib-0.3.0/wizlib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wizlib
-Version: 0.2.6
+Version: 0.3.0
 Summary: A collection of Python modules that are useful across multiple projects
 Author-email: Francis Potter <wizlib@steampunkwizard.ca>
 License: MIT

{wizlib-0.2.6 → wizlib-0.3.0}/wizlib.egg-info/SOURCES.txt

@@ -1,8 +1,10 @@
 .version
 README.md
 pyproject.toml
+test/test_super_wrapper.py
 wizlib/__init__.py
 wizlib/rlinput.py
+wizlib/super_wrapper.py
 wizlib.egg-info/PKG-INFO
 wizlib.egg-info/SOURCES.txt
 wizlib.egg-info/dependency_links.txt

wizlib-0.2.6/.version

@@ -1 +0,0 @@
-0.2.6