laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix imports

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2024-10-21 23:20:21 -07:00
parent 5fa9a2c9ba
commit 91601b4f57
4 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
reflex/components/react_player/__init__.py
View File

@ -1,5 +1,6 @@
"""React Player component for audio and video.""" """React Player component for audio and video."""
from . import react_player
from .audio import Audio from .audio import Audio
from .video import Video from .video import Video

5
reflex/components/react_player/audio.pyi
View File

@ -5,6 +5,7 @@
# ------------------------------------------------------ # ------------------------------------------------------
from typing import Any, Dict, Optional, Union, overload from typing import Any, Dict, Optional, Union, overload
import reflex
from reflex.components.react_player.react_player import ReactPlayer from reflex.components.react_player.react_player import ReactPlayer
from reflex.event import EventType from reflex.event import EventType
from reflex.style import Style from reflex.style import Style
@ -58,7 +59,9 @@ class Audio(ReactPlayer):
on_play: Optional[EventType[[]]] = None, on_play: Optional[EventType[[]]] = None,
on_playback_quality_change: Optional[EventType[[]]] = None, on_playback_quality_change: Optional[EventType[[]]] = None,
on_playback_rate_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_ready: Optional[EventType[[]]] = None,
on_scroll: Optional[EventType[[]]] = None, on_scroll: Optional[EventType[[]]] = None,
on_seek: Optional[EventType[float]] = None, on_seek: Optional[EventType[float]] = None,

5
reflex/components/react_player/video.pyi
View File

@ -5,6 +5,7 @@
# ------------------------------------------------------ # ------------------------------------------------------
from typing import Any, Dict, Optional, Union, overload from typing import Any, Dict, Optional, Union, overload
import reflex
from reflex.components.react_player.react_player import ReactPlayer from reflex.components.react_player.react_player import ReactPlayer
from reflex.event import EventType from reflex.event import EventType
from reflex.style import Style from reflex.style import Style
@ -58,7 +59,9 @@ class Video(ReactPlayer):
on_play: Optional[EventType[[]]] = None, on_play: Optional[EventType[[]]] = None,
on_playback_quality_change: Optional[EventType[[]]] = None, on_playback_quality_change: Optional[EventType[[]]] = None,
on_playback_rate_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_ready: Optional[EventType[[]]] = None,
on_scroll: Optional[EventType[[]]] = None, on_scroll: Optional[EventType[[]]] = None,
on_seek: Optional[EventType[float]] = None, on_seek: Optional[EventType[float]] = None,

7
reflex/utils/pyi_generator.py
View File

@ -214,7 +214,9 @@ def _get_type_hint(value, type_hint_globals, is_optional=True) -> str:
return res 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. """Generate the import statements for the stub file.
Args: 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]) ast.ImportFrom(module=name, names=[ast.alias(name=val) for val in values])
for name, values in DEFAULT_IMPORTS.items() 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.) # Handle plain types (int, str, custom classes, etc.)
if origin is None: if origin is None:
if hasattr(typ, "__name__"): if hasattr(typ, "__name__"):
if typ.__module__.startswith("reflex."):
return ast.Name(id=typ.__module__ + "." + typ.__name__)
return ast.Name(id=typ.__name__) return ast.Name(id=typ.__name__)
elif hasattr(typ, "_name"): elif hasattr(typ, "_name"):
return ast.Name(id=typ._name) return ast.Name(id=typ._name)