fix: only legal widgets in the docstring will be processed

TexteaInc · Jul 15, 2024 · c740cd9 · c740cd9
1 parent caf8fdb
commit c740cd9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 10 deletions.
diff --git a/backend/funix/decorator/__init__.py b/backend/funix/decorator/__init__.py
@@ -63,6 +63,7 @@
     TreatAsType,
     WhitelistType,
     WidgetsType,
+    AcceptableWidgetsList,
 )
 from funix.jupyter import jupyter
 from funix.util.module import funix_menu_to_safe_function_name
@@ -520,16 +521,20 @@ def _function_reactive_update():
                             decorated_param_ = decorated_params.get(
                                 param_.arg_name, None
                             )
-                            widget_ = param_.description.split(",")[0].strip()
-                            if decorated_param_ is None:
-                                decorated_params[param_.arg_name] = {
-                                    "widget": widget_,
-                                }
-                            else:
-                                if "widget" not in decorated_param_:  # no cover
-                                    decorated_params[param_.arg_name][
-                                        "widget"
-                                    ] = widget_
+                            try:
+                                widget_ = param_.description.split(",")[0].strip()
+                                if widget_ in AcceptableWidgetsList:
+                                    if decorated_param_ is None:
+                                        decorated_params[param_.arg_name] = {
+                                            "widget": widget_,
+                                        }
+                                    else:
+                                        if "widget" not in decorated_param_:  # no cover
+                                            decorated_params[param_.arg_name][
+                                                "widget"
+                                            ] = widget_
+                            except:
+                                pass
 
             safe_argument_config = {} if argument_config is None else argument_config
 

diff --git a/backend/funix/hint/__init__.py b/backend/funix/hint/__init__.py
@@ -49,6 +49,8 @@
 Acceptable widgets. Implemented now.
 """
 
+AcceptableWidgetsList = list(getattr(AcceptableWidgets, "__args__"))
+
 WidgetsValue = (
     list[AcceptableWidgets | tuple[AcceptableWidgets, dict]]
     | AcceptableWidgets