apply greptile feedback

Author: masenf

packages/reflex-base/src/reflex_base/utils/pyi_generator.py

@@ -318,20 +318,20 @@ def _get_class_prop_comments(clz: type[Component]) -> Mapping[str, tuple[str, ..
     in_docstring = False
     docstring_lines: list[str] = []
     for line in _get_source(clz).splitlines():
-        reached_functions = re.search(r"def ", line)
-        if reached_functions:
-            # We've reached the functions, so stop.
-            break
-
         stripped = line.strip()
 
         # Handle triple-quoted docstrings after prop definitions.
+        # This must be checked before the `def ` boundary so that
+        # docstring prose containing "def " doesn't break the loop.
         if in_docstring:
             if '"""' in stripped or "'''" in stripped:
                 # End of multi-line docstring.
-                end_text = stripped.partition('"""')[0] or stripped.partition("'''")[0]
+                if '"""' in stripped:
+                    end_text = stripped.partition('"""')[0].strip()
+                else:
+                    end_text = stripped.partition("'''")[0].strip()
                 if end_text:
-                    docstring_lines.append(end_text.strip())
+                    docstring_lines.append(end_text)
                 if last_prop and docstring_lines:
                     props_comments[last_prop] = tuple(docstring_lines)
                 in_docstring = False
@@ -341,6 +341,11 @@ def _get_class_prop_comments(clz: type[Component]) -> Mapping[str, tuple[str, ..
                 docstring_lines.append(stripped)
             continue
 
+        reached_functions = re.search(r"def ", line)
+        if reached_functions:
+            # We've reached the functions, so stop.
+            break
+
         # Check for start of a docstring right after a prop.
         if last_prop and (stripped.startswith(('"""', "'''"))):
             quote = '"""' if stripped.startswith('"""') else "'''"

tests/units/reflex_base/utils/pyi_generator/dataset/simple_component.py

@@ -49,6 +49,11 @@ class SimpleComponent(Component):
     """A tooltip that appears on hover
     with additional details."""
 
+    callback: Var[str]
+    """
+    The def of the callback to use when the component is clicked.
+    """
+
     def _private_method(self):
         """This should not appear in the stub.
 

tests/units/reflex_base/utils/pyi_generator/golden/simple_component.pyi

@@ -27,6 +27,7 @@ class SimpleComponent(Component):
         label: Var[str] | str | None = None,
         description: Var[str] | str | None = None,
         tooltip: Var[str] | str | None = None,
+        callback: Var[str] | str | None = None,
         style: Sequence[Mapping[str, Any]]
         | Mapping[str, Any]
         | Var[Mapping[str, Any]]
@@ -66,6 +67,7 @@ class SimpleComponent(Component):
             label: An optional label with a default value.
             description: A detailed description of the component.
             tooltip: A tooltip that appears on hover with additional details.
+            callback: The def of the callback to use when the component is clicked.
             style: The style of the component.
             key: A unique key for the component.
             id: The id for the component.