expose register at top-level

Author: kecnry

phoebe/__init__.py

@@ -888,6 +888,27 @@ def add_distl_docstring(obj):
 add_distl_docstring(uniform_around)
 add_distl_docstring(gaussian_around)
 
+# expose registering pluggable features`
+def register_feature(feature_cls, kind=None) -> None:
+    """
+    Register a feature class to be available in the Bundle.
+
+    Arguments
+    -----------
+    * `feature_cls` (class): the class to register
+    * `kind` (string, optional): the kind to register the class under.  If None,
+        the class will be registered under its default kind.
+    """
+
+    from phoebe.features.dataset_features import  _register_feature as ds_register_feature
+    from phoebe.features.component_features import _register_feature as comp_register_feature
+    feature_type = getattr(feature_cls, '_phoebe_custom_feature', None)
+    if feature_type == 'dataset':
+        ds_register_feature(feature_cls, kind=kind)
+    elif feature_type == 'component':
+        comp_register_feature(feature_cls, kind=kind)
+    else:
+        raise TypeError("could not find matching feature type for {}".format(feature_cls))
 
 # expose available "kinds" per-context
 def _get_phoebe_funcs(module, devel=False):

phoebe/features/component_features.py

@@ -8,11 +8,11 @@
 logger.addHandler(logging.NullHandler())
 
 
-__all__ = ['register_feature', 'ComponentFeature', 'Spot', 'Pulsation']
+__all__ = ['ComponentFeature', 'Spot', 'Pulsation']
 
 _skip_filter_checks = {'check_default': False, 'check_visible': False}
 
-def register_feature(feature_cls, kind=None):
+def _register_feature(feature_cls, kind=None):
     if kind is None:
         kind = feature_cls.__name__.lower()
 

phoebe/features/dataset_features.py

@@ -9,11 +9,11 @@
 from phoebe.parameters import constraint
 import phoebe.parameters.feature as _parameters_feature
 
-__all__ = ['register_feature', 'DatasetFeature']
+__all__ = ['DatasetFeature']
 
 _skip_filter_checks = {'check_default': False, 'check_visible': False}
 
-def register_feature(feature_cls, kind=None):
+def _register_feature(feature_cls, kind=None):
     if kind is None:
         kind = feature_cls.__name__.lower()