Add compatablity with upstream wildcard imports

Fixes #19

Author: DocOtak

gsw_xarray/__init__.py

@@ -4,6 +4,8 @@
 
 import gsw
 from ._core import _wrapped_funcs
+from ._util import submodule_all_compat
+
 
 _compat = {
     "conversions",
@@ -16,6 +18,8 @@
 }
 _compat_modules = {name: import_module(f".{name}", "gsw_xarray") for name in _compat}
 
+__all__ = submodule_all_compat("gsw") + list(_compat)
+
 # See PEP 562
 def __getattr__(name):
     if name in _compat:

gsw_xarray/_core.py

@@ -5,6 +5,7 @@
 
 from ._cf_names import _func_standard_name_units
 
+
 def cf_attrs(standard_name, units, extra=None):
     def cf_attrs_decorator(func):
         @wraps(func)
@@ -20,6 +21,7 @@ def cf_attrs_wrapper(*args, **kwargs):
 
     return cf_attrs_decorator
 
+
 def _init_funcs():
     _wrapped_funcs = {}
     for func, name, units in _func_standard_name_units:

gsw_xarray/_util.py

@@ -16,6 +16,18 @@ def getattr_(name):
 
 def submodule_dir_compat(submodule_name):
     submodule = import_module(submodule_name)
+
     def dir_():
         return dir(submodule)
+
     return dir_
+
+
+def submodule_all_compat(submodule_name):
+    submodule = import_module(submodule_name)
+    try:
+        return submodule.__all__
+    except AttributeError:
+        all = dir(submodule)
+        # as per python docs, an import * should return everything not starting with "_"
+        return list(filter(lambda s: not s.startswith("_"), all))

gsw_xarray/conversions.py

@@ -1,4 +1,5 @@
-from ._util import submodule_attr_compat, submodule_dir_compat
+from ._util import submodule_attr_compat, submodule_dir_compat, submodule_all_compat
 
 __getattr__ = submodule_attr_compat("gsw.conversions")
 __dir__ = submodule_dir_compat("gsw.conversions")
+__all__ = submodule_all_compat("gsw.conversions")

gsw_xarray/density.py

@@ -1,4 +1,5 @@
-from ._util import submodule_attr_compat, submodule_dir_compat
+from ._util import submodule_attr_compat, submodule_dir_compat, submodule_all_compat
 
 __getattr__ = submodule_attr_compat("gsw.density")
 __dir__ = submodule_dir_compat("gsw.density")
+__all__ = submodule_all_compat("gsw.density")

gsw_xarray/energy.py

@@ -1,4 +1,5 @@
-from ._util import submodule_attr_compat, submodule_dir_compat
+from ._util import submodule_attr_compat, submodule_dir_compat, submodule_all_compat
 
 __getattr__ = submodule_attr_compat("gsw.energy")
 __dir__ = submodule_dir_compat("gsw.energy")
+__all__ = submodule_all_compat("gsw.energy")

gsw_xarray/freezing.py

@@ -1,4 +1,5 @@
-from ._util import submodule_attr_compat, submodule_dir_compat
+from ._util import submodule_attr_compat, submodule_dir_compat, submodule_all_compat
 
 __getattr__ = submodule_attr_compat("gsw.freezing")
 __dir__ = submodule_dir_compat("gsw.freezing")
+__all__ = submodule_all_compat("gsw.freezing")

gsw_xarray/geostrophy.py

@@ -1,4 +1,5 @@
-from ._util import submodule_attr_compat, submodule_dir_compat
+from ._util import submodule_attr_compat, submodule_dir_compat, submodule_all_compat
 
 __getattr__ = submodule_attr_compat("gsw.geostrophy")
 __dir__ = submodule_dir_compat("gsw.geostrophy")
+__all__ = submodule_all_compat("gsw.geostrophy")

gsw_xarray/ice.py

@@ -1,4 +1,5 @@
-from ._util import submodule_attr_compat, submodule_dir_compat
+from ._util import submodule_attr_compat, submodule_dir_compat, submodule_all_compat
 
 __getattr__ = submodule_attr_compat("gsw.ice")
 __dir__ = submodule_dir_compat("gsw.ice")
+__all__ = submodule_all_compat("gsw.ice")

gsw_xarray/stability.py

@@ -1,4 +1,5 @@
-from ._util import submodule_attr_compat, submodule_dir_compat
+from ._util import submodule_attr_compat, submodule_dir_compat, submodule_all_compat
 
 __getattr__ = submodule_attr_compat("gsw.stability")
 __dir__ = submodule_dir_compat("gsw.stability")
+__all__ = submodule_all_compat("gsw.stability")