pyview-web 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

pyview/async_stream_runner.py

@@ -0,0 +1,65 @@
+import asyncio
+import uuid
+import logging
+from typing import Any, AsyncGenerator, Callable, Optional
+from pyview.events.info_event import InfoEvent, InfoEventScheduler
+
+
+class AsyncStreamRunner:
+    def __init__(self, scheduler: InfoEventScheduler):
+        self._stream_tasks: dict[str, asyncio.Task] = {}
+        self._scheduler = scheduler
+
+    def start_stream(
+        self,
+        gen: AsyncGenerator[Any, None],
+        *,
+        on_yield: Callable[[Any], InfoEvent],
+        on_done: Optional[InfoEvent] = None,
+        on_error: Optional[Callable[[Exception], InfoEvent]] = None,
+        on_cancel: Optional[InfoEvent] = None,
+    ) -> str:
+        """
+        Run `gen` in the background, returning an op_id you can later use
+        to cancel.  Hooks:
+
+        - on_yield(item)  → scheduled per chunk
+        - on_done        → scheduled once at normal completion
+        - on_error(exc)  → scheduled on unexpected exception
+        - on_cancel      → scheduled if the task is cancelled
+        """
+        task_id = uuid.uuid4().hex
+
+        async def driver():
+            try:
+                async for item in gen:
+                    self._scheduler.schedule_info_once(on_yield(item))
+            except asyncio.CancelledError:
+                # user-requested cancellation
+                if on_cancel:
+                    self._scheduler.schedule_info_once(on_cancel)
+                # swallow so it doesn’t log as an “error”
+            except Exception as exc:
+                if on_error:
+                    self._scheduler.schedule_info_once(on_error(exc))
+                else:
+                    logging.exception(f"Error in stream {task_id}", exc_info=True)
+            else:
+                if on_done:
+                    self._scheduler.schedule_info_once(on_done)
+            finally:
+                self._stream_tasks.pop(task_id, None)
+
+        task = asyncio.create_task(driver())
+        self._stream_tasks[task_id] = task
+        return task_id
+
+    def cancel_stream(self, task_id: str) -> bool:
+        """
+        Cancel a running stream.  Returns True if a task was found & cancelled.
+        """
+        task = self._stream_tasks.get(task_id)
+        if not task:
+            return False
+        task.cancel()
+        return True

pyview/events/BaseEventHandler.py

@@ -1,6 +1,9 @@
-from typing import Callable
+from typing import Callable, TYPE_CHECKING
 import logging
 
+if TYPE_CHECKING:
+    from pyview.live_view import InfoEvent
+
 
 def event(*event_names):
     """Decorator that marks methods as event handlers."""
@@ -12,22 +15,37 @@ def event(*event_names):
     return decorator
 
 
+def info(*info_names):
+    """Decorator that marks methods as info handlers."""
+
+    def decorator(func):
+        func._info_names = info_names
+        return func
+
+    return decorator
+
+
 class BaseEventHandler:
-    """Base class for event handlers to handle dispatching events."""
+    """Base class for event handlers to handle dispatching events and info."""
 
     _event_handlers: dict[str, Callable] = {}
+    _info_handlers: dict[str, Callable] = {}
 
     def __init_subclass__(cls, **kwargs):
         super().__init_subclass__(**kwargs)
 
         # Find all decorated methods and register them
         cls._event_handlers = {}
+        cls._info_handlers = {}
         for attr_name in dir(cls):
             if not attr_name.startswith("_"):
                 attr = getattr(cls, attr_name)
                 if hasattr(attr, "_event_names"):
                     for event_name in attr._event_names:
                         cls._event_handlers[event_name] = attr
+                if hasattr(attr, "_info_names"):
+                    for info_name in attr._info_names:
+                        cls._info_handlers[info_name] = attr
 
     async def handle_event(self, event: str, payload: dict, socket):
         handler = self._event_handlers.get(event)
@@ -36,3 +54,11 @@ class BaseEventHandler:
             return await handler(self, event, payload, socket)
         else:
             logging.warning(f"Unhandled event: {event} {payload}")
+
+    async def handle_info(self, info: "InfoEvent", socket):
+        handler = self._info_handlers.get(info.name)
+
+        if handler:
+            return await handler(self, info, socket)
+        else:
+            logging.warning(f"Unhandled info: {info.name} {info}")

pyview/events/__init__.py

@@ -1,4 +1,4 @@
-from .BaseEventHandler import BaseEventHandler, event
+from .BaseEventHandler import BaseEventHandler, event, info
 from .info_event import InfoEvent
 
-__all__ = ["BaseEventHandler", "event", "InfoEvent"]
+__all__ = ["BaseEventHandler", "event", "info", "InfoEvent"]

pyview/events/info_event.py

@@ -1,8 +1,16 @@
 from dataclasses import dataclass
-from typing import Any
+from typing import Any, Optional, Protocol
 
 
 @dataclass
 class InfoEvent:
     name: str
     payload: Any = None
+
+
+class InfoEventScheduler(Protocol):
+    def schedule_info(self, event: InfoEvent, seconds: float):
+        pass
+
+    def schedule_info_once(self, event: InfoEvent, seconds: Optional[float] = None):
+        pass

pyview/live_socket.py

@@ -17,7 +17,9 @@ from pyview.vendor.flet.pubsub import PubSubHub, PubSub
 from pyview.events import InfoEvent
 from pyview.uploads import UploadConstraints, UploadConfig, UploadManager
 from pyview.meta import PyViewMeta
+from pyview.template.render_diff import calc_diff
 import datetime
+from pyview.async_stream_runner import AsyncStreamRunner
 
 
 if TYPE_CHECKING:
@@ -50,8 +52,8 @@ class ConnectedLiveViewSocket(Generic[T]):
     context: T
     live_title: Optional[str] = None
     pending_events: list[tuple[str, Any]]
-
     upload_manager: UploadManager
+    prev_rendered: Optional[dict[str, Any]] = None
 
     def __init__(self, websocket: WebSocket, topic: str, liveview: LiveView):
         self.websocket = websocket
@@ -62,6 +64,7 @@ class ConnectedLiveViewSocket(Generic[T]):
         self.pub_sub = PubSub(pub_sub_hub, topic)
         self.pending_events = []
         self.upload_manager = UploadManager()
+        self.stream_runner = AsyncStreamRunner(self)
 
     @property
     def meta(self) -> PyViewMeta:
@@ -93,9 +96,13 @@ class ConnectedLiveViewSocket(Generic[T]):
         )
 
     def diff(self, render: dict[str, Any]) -> dict[str, Any]:
-        # TODO: not a real diff
-        del render["s"]
-        return render
+        if self.prev_rendered:
+            diff = calc_diff(self.prev_rendered, render)
+        else:
+            diff = render
+
+        self.prev_rendered = render
+        return diff
 
     async def send_info(self, event: InfoEvent):
         await self.liveview.handle_info(event, self)

pyview/ws_handler.py

@@ -1,4 +1,4 @@
-from typing import Optional, Any
+from typing import Optional
 import json
 from starlette.websockets import WebSocket, WebSocketDisconnect
 from urllib.parse import urlparse, parse_qs
@@ -8,7 +8,6 @@ from pyview.csrf import validate_csrf_token
 from pyview.session import deserialize_session
 from pyview.auth import AuthProviderFactory
 from pyview.phx_message import parse_message
-from pyview.template.render_diff import calc_diff
 
 
 class AuthException(Exception):
@@ -60,6 +59,7 @@ class LiveSocketHandler:
                 await lv.handle_params(url, merged_params, socket)
 
                 rendered = await _render(socket)
+                socket.prev_rendered = rendered
 
                 resp = [
                     joinRef,
@@ -70,7 +70,7 @@ class LiveSocketHandler:
                 ]
 
                 await self.manager.send_personal_message(json.dumps(resp), websocket)
-                await self.handle_connected(topic, socket, rendered)
+                await self.handle_connected(topic, socket)
 
         except WebSocketDisconnect:
             if socket:
@@ -80,9 +80,7 @@ class LiveSocketHandler:
             await websocket.close()
             self.sessions -= 1
 
-    async def handle_connected(
-        self, myJoinId, socket: ConnectedLiveViewSocket, prev_rendered: dict[str, Any]
-    ):
+    async def handle_connected(self, myJoinId, socket: ConnectedLiveViewSocket):
         while True:
             message = await socket.websocket.receive()
             [joinRef, mesageRef, topic, event, payload] = parse_message(message)
@@ -114,8 +112,7 @@ class LiveSocketHandler:
                     {} if not socket.pending_events else {"e": socket.pending_events}
                 )
 
-                diff = calc_diff(prev_rendered, rendered)
-                prev_rendered = rendered
+                diff = socket.diff(rendered)
 
                 socket.pending_events = []
 
@@ -150,8 +147,7 @@ class LiveSocketHandler:
 
                 await lv.handle_params(url, merged_params, socket)
                 rendered = await _render(socket)
-                diff = calc_diff(prev_rendered, rendered)
-                prev_rendered = rendered
+                diff = socket.diff(rendered)
 
                 resp = [
                     joinRef,
@@ -171,8 +167,7 @@ class LiveSocketHandler:
                 )
 
                 rendered = await _render(socket)
-                diff = calc_diff(prev_rendered, rendered)
-                prev_rendered = rendered
+                diff = socket.diff(rendered)
 
                 resp = [
                     joinRef,
@@ -238,8 +233,7 @@ class LiveSocketHandler:
             if event == "progress":
                 socket.upload_manager.update_progress(joinRef, payload)
                 rendered = await _render(socket)
-                diff = calc_diff(prev_rendered, rendered)
-                prev_rendered = rendered
+                diff = socket.diff(rendered)
 
                 resp = [
                     joinRef,

{pyview_web-0.2.2.dist-info → pyview_web-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pyview-web
-Version: 0.2.2
+Version: 0.2.4
 Summary: LiveView in Python
 License: MIT
 Keywords: web,api,LiveView

{pyview_web-0.2.2.dist-info → pyview_web-0.2.4.dist-info}/RECORD

@@ -2,18 +2,19 @@ pyview/__init__.py,sha256=5RJ_KtwJvI_-_Vhb3-py5Qf78YdH1HHvAzZO1ddzzrU,518
 pyview/assets/js/app.js,sha256=XuuSgEMY4hx8v0OuEPwaa7trktu_vppL0tc3Bs9Fw7s,2524
 pyview/assets/package-lock.json,sha256=kFCrEUJc3G7VD7EsBQf6__EKQhaKAok-I5rrwiAoX0w,2425
 pyview/assets/package.json,sha256=E6xaX8KMUAektIIedLmI55jGnmlNMSeD2tgKYXWk1vg,151
+pyview/async_stream_runner.py,sha256=_vXeU1LyuQkJrK5AlaaF1gyhRFWXhzn3Y73BFhSsWVc,2289
 pyview/auth/__init__.py,sha256=vMlirETRhD4va61NOzwg8VY8ep9wVOF96GznJGBmzD0,109
 pyview/auth/provider.py,sha256=fwriy2JZcOStutVXD-8VlMPAFXjILCM0l08lhTgmuyE,935
 pyview/auth/required.py,sha256=ZtNmLFth9nK39RxDiJkSzArXwS5Cvr55MUAzfJ1F2e0,1418
 pyview/changesets/__init__.py,sha256=55CLari2JHZtwy4hapHe7CqUyKjcP4dkM_t5d3CY2gU,46
 pyview/changesets/changesets.py,sha256=hImmvB_jS6RyLr5Mas5L7DO_0d805jR3c41LKJlnNL4,1720
 pyview/csrf.py,sha256=VIURva9EJqXXYGC7engweh3SwDQCnHlhV2zWdcdnFqc,789
-pyview/events/BaseEventHandler.py,sha256=RIv1vYn-sNIAfPNAaRg5iJvM--0ZlQHt3X7av_GR0sw,1138
-pyview/events/__init__.py,sha256=5qQhZUjBwbnL9SrQ9FWHNgrEgDETCI3ysMREx7Tab4E,142
-pyview/events/info_event.py,sha256=Zv8G2F1XeXUk1wrnfomeFfxB0OPYmHdjSvxRjQew3No,125
+pyview/events/BaseEventHandler.py,sha256=0xcjFFMLMN8Aj6toI31vzeYhRQqaX9rm-G7XGXMvqsE,1923
+pyview/events/__init__.py,sha256=oP0SG4Af4uf0GEa0Y_zHYhR7TcBOcXQlTAsgOSaIcC4,156
+pyview/events/info_event.py,sha256=JOwf3KDodHkmH1MzqTD8sPxs0zbI4t8Ff0rLjwRSe2Y,358
 pyview/js.py,sha256=E6HMsUfXQjrcLqYq26ieeYuzTjBeZqfJwwOm3uSR4ME,3498
 pyview/live_routes.py,sha256=IN2Jmy8b1umcfx1R7ZgFXHZNbYDJp_kLIbADtDJknPM,1749
-pyview/live_socket.py,sha256=D8fg7UQSPCVEboxohbDAZuK7J1xBEEoDycVFUuDfZxI,4767
+pyview/live_socket.py,sha256=p1KtX9Exwhgsf0yOp3Eb32zdUOo5hSnYDJrpJuTu3QI,5084
 pyview/live_view.py,sha256=mwAp7jiABSZCBgYF-GLQCB7zcJ7Wpz9cuC84zjzsp2U,1455
 pyview/meta.py,sha256=01Z-qldB9jrewmIJHQpUqyIhuHodQGgCvpuY9YM5R6c,74
 pyview/phx_message.py,sha256=DUdPfl6tlw9K0FNXJ35ehq03JGgynvwA_JItHQ_dxMQ,2007
@@ -43,8 +44,8 @@ pyview/vendor/ibis/nodes.py,sha256=TgFt4q5MrVW3gC3PVitrs2LyXKllRveooM7XKydNATk,2
 pyview/vendor/ibis/template.py,sha256=6XJXnztw87CrOaKeW3e18LL0fNM8AI6AaK_QgMdb7ew,2353
 pyview/vendor/ibis/tree.py,sha256=hg8f-fKHeo6DE8R-QgAhdvEaZ8rKyz7p0nGwPy0CBTs,2509
 pyview/vendor/ibis/utils.py,sha256=nLSaxPR9vMphzV9qinlz_Iurv9c49Ps6Knv8vyNlewU,2768
-pyview/ws_handler.py,sha256=8iFKEse4TUaBzHF4xAcZiOqXaAgv4yB2CT0BKoW2Ny4,9319
-pyview_web-0.2.2.dist-info/LICENSE,sha256=M_bADaBm9_MV9llX3lCicksLhwk3eZUjA2srE0uUWr0,1071
-pyview_web-0.2.2.dist-info/METADATA,sha256=y7PXP4uAyDOEVhkqa-JclJOrPCBfbqJAacMq9XpbcLU,5256
-pyview_web-0.2.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-pyview_web-0.2.2.dist-info/RECORD,,
+pyview/ws_handler.py,sha256=CY1iDx5GETjIkqhgFbo2fkE3FhrqucSdg4AjuJ2P0Qg,9041
+pyview_web-0.2.4.dist-info/LICENSE,sha256=M_bADaBm9_MV9llX3lCicksLhwk3eZUjA2srE0uUWr0,1071
+pyview_web-0.2.4.dist-info/METADATA,sha256=v6WKqv3gIV-4BdP9M91sFKcCjpeg5_Beug68wgqlvrU,5256
+pyview_web-0.2.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+pyview_web-0.2.4.dist-info/RECORD,,