소스 검색

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

Signed-off-by: Alexander Zarubkin <alexander.zarubkin@toptal.com>
Alexander Zarubkin 2 년 전
부모
커밋
6e088545fc
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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):
             if self.page_ready_generator is not None:
                 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):
-                    next(self.page_ready_generator)
+                    self.page_ready_generator.send(msg.websocket)
             if self.page_ready_handler:
                 arg_count = len(inspect.signature(self.page_ready_handler).parameters)
                 is_coro = is_coroutine(self.page_ready_handler)