Add additional math methods.

Author: tychedelia

crates/processing_pyo3/mewnala/__init__.py

@@ -1,17 +1,17 @@
 from .mewnala import *
 
-# re-export the native submodules as submodules of this module, if they exist
-# this allows users to import from `mewnala.math` and `mewnala.color`
-# if they exist, without needing to know about the internal structure of the native module
 import sys as _sys
 from . import mewnala as _native
-for _name in ("math", "color"):
-    _sub = getattr(_native, _name, None)
-    if _sub is not None:
-        _sys.modules[f"{__name__}.{_name}"] = _sub
+
+_color = getattr(_native, "color", None)
+if _color is not None:
+    _sys.modules[f"{__name__}.color"] = _color
+
+from . import math  # noqa: E402  (Python submodule, extends native math)
+from .math import *  # noqa: E402,F401,F403
 
 # global var handling. for wildcard import of our module, we copy into globals, otherwise
-# we dispatch to get attr and call the underlying getter method 
+# we dispatch to get attr and call the underlying getter method
 
 _DYNAMIC_GRAPHICS_ATTRS = (
     "width",
@@ -69,6 +69,3 @@ def __getattr__(name):
 
 def __dir__():
     return sorted(set(list(globals().keys()) + list(_DYNAMIC)))
-
-
-del _sys, _name, _sub