From 91601b4f575b21d63c57a13b7c9b02a12319dc25 Mon Sep 17 00:00:00 2001 From: Khaleel Al-Adhami Date: Mon, 21 Oct 2024 23:20:21 -0700 Subject: [PATCH] fix imports --- reflex/components/react_player/__init__.py | 1 + reflex/components/react_player/audio.pyi | 5 ++++- reflex/components/react_player/video.pyi | 5 ++++- reflex/utils/pyi_generator.py | 7 ++++++- 4 files changed, 15 insertions(+), 3 deletions(-) diff --git a/reflex/components/react_player/__init__.py b/reflex/components/react_player/__init__.py index 8c4a4486f..3f807b1a0 100644 --- a/reflex/components/react_player/__init__.py +++ b/reflex/components/react_player/__init__.py @@ -1,5 +1,6 @@ """React Player component for audio and video.""" +from . import react_player from .audio import Audio from .video import Video diff --git a/reflex/components/react_player/audio.pyi b/reflex/components/react_player/audio.pyi index 40c24464d..1841829af 100644 --- a/reflex/components/react_player/audio.pyi +++ b/reflex/components/react_player/audio.pyi @@ -5,6 +5,7 @@ # ------------------------------------------------------ from typing import Any, Dict, Optional, Union, overload +import reflex from reflex.components.react_player.react_player import ReactPlayer from reflex.event import EventType from reflex.style import Style @@ -58,7 +59,9 @@ class Audio(ReactPlayer): on_play: Optional[EventType[[]]] = None, on_playback_quality_change: Optional[EventType[[]]] = None, on_playback_rate_change: Optional[EventType[[]]] = None, - on_progress: Optional[EventType[Progress]] = None, + on_progress: Optional[ + EventType[reflex.components.react_player.react_player.Progress] + ] = None, on_ready: Optional[EventType[[]]] = None, on_scroll: Optional[EventType[[]]] = None, on_seek: Optional[EventType[float]] = None, diff --git a/reflex/components/react_player/video.pyi b/reflex/components/react_player/video.pyi index d39b4f29b..a05e3747b 100644 --- a/reflex/components/react_player/video.pyi +++ b/reflex/components/react_player/video.pyi @@ -5,6 +5,7 @@ # ------------------------------------------------------ from typing import Any, Dict, Optional, Union, overload +import reflex from reflex.components.react_player.react_player import ReactPlayer from reflex.event import EventType from reflex.style import Style @@ -58,7 +59,9 @@ class Video(ReactPlayer): on_play: Optional[EventType[[]]] = None, on_playback_quality_change: Optional[EventType[[]]] = None, on_playback_rate_change: Optional[EventType[[]]] = None, - on_progress: Optional[EventType[Progress]] = None, + on_progress: Optional[ + EventType[reflex.components.react_player.react_player.Progress] + ] = None, on_ready: Optional[EventType[[]]] = None, on_scroll: Optional[EventType[[]]] = None, on_seek: Optional[EventType[float]] = None, diff --git a/reflex/utils/pyi_generator.py b/reflex/utils/pyi_generator.py index 026a53bca..b5f078594 100644 --- a/reflex/utils/pyi_generator.py +++ b/reflex/utils/pyi_generator.py @@ -214,7 +214,9 @@ def _get_type_hint(value, type_hint_globals, is_optional=True) -> str: return res -def _generate_imports(typing_imports: Iterable[str]) -> list[ast.ImportFrom]: +def _generate_imports( + typing_imports: Iterable[str], +) -> list[ast.ImportFrom | ast.Import]: """Generate the import statements for the stub file. Args: @@ -228,6 +230,7 @@ def _generate_imports(typing_imports: Iterable[str]) -> list[ast.ImportFrom]: ast.ImportFrom(module=name, names=[ast.alias(name=val) for val in values]) for name, values in DEFAULT_IMPORTS.items() ], + ast.Import([ast.alias("reflex")]), ] @@ -390,6 +393,8 @@ def type_to_ast(typ) -> ast.AST: # Handle plain types (int, str, custom classes, etc.) if origin is None: if hasattr(typ, "__name__"): + if typ.__module__.startswith("reflex."): + return ast.Name(id=typ.__module__ + "." + typ.__name__) return ast.Name(id=typ.__name__) elif hasattr(typ, "_name"): return ast.Name(id=typ._name)