Explorar o código

Pass websocket instance back to generator-style page builder function. Fixes #130.

Signed-off-by: Alexander Zarubkin <alexander.zarubkin@toptal.com>
Alexander Zarubkin %!s(int64=2) %!d(string=hai) anos
pai
achega
6e088545fc
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  1. 2 2
      nicegui/page.py

+ 2 - 2
nicegui/page.py

@@ -89,9 +89,9 @@ class Page(jp.QuasarPage):
         with globals.within_view(self.view):
         with globals.within_view(self.view):
             if self.page_ready_generator is not None:
             if self.page_ready_generator is not None:
                 if isinstance(self.page_ready_generator, types.AsyncGeneratorType):
                 if isinstance(self.page_ready_generator, types.AsyncGeneratorType):
-                    await self.page_ready_generator.__anext__()
+                    await self.page_ready_generator.asend(msg.websocket)
                 elif isinstance(self.page_ready_generator, types.GeneratorType):
                 elif isinstance(self.page_ready_generator, types.GeneratorType):
-                    next(self.page_ready_generator)
+                    self.page_ready_generator.send(msg.websocket)
             if self.page_ready_handler:
             if self.page_ready_handler:
                 arg_count = len(inspect.signature(self.page_ready_handler).parameters)
                 arg_count = len(inspect.signature(self.page_ready_handler).parameters)
                 is_coro = is_coroutine(self.page_ready_handler)
                 is_coro = is_coroutine(self.page_ready_handler)