raise error when input after page close

Author: wang0618

pywebio/io_ctrl.py

@@ -9,6 +9,7 @@
 
 from .session import chose_impl, next_client_event, get_current_task_id, get_current_session
 from .utils import random_str
+from .exceptions import PageClosedException
 
 logger = logging.getLogger(__name__)
 
@@ -383,6 +384,11 @@ def input_event_handle(item_valid_funcs, form_valid_funcs, preprocess_funcs, onc
         elif event_name == 'from_cancel':
             data = None
             break
+        elif event_name == 'page_close':
+            current_page = get_current_session().get_page_id(check_active=False)
+            closed_page = event_data
+            if closed_page == current_page:
+                raise PageClosedException
         else:
             logger.warning("Unhandled Event: %s", event)
 

pywebio/session/base.py

@@ -97,7 +97,7 @@ def push_scope(self, name):
         task_id = type(self).get_current_task_id()
         self.scope_stack[task_id].append(name)
 
-    def get_page_id(self):
+    def get_page_id(self, check_active=True):
         """
         get the if of current page in task, return `None` when it's master page, 
         raise PageClosedException when current page is closed
@@ -108,7 +108,7 @@ def get_page_id(self):
             return None
 
         page_id = self.page_stack[task_id][-1]
-        if page_id not in self.active_page[task_id]:
+        if page_id not in self.active_page[task_id] and check_active:
             raise PageClosedException(
                 "The page is closed by app user, "
                 "you see this exception mostly because you set `silent_quit=False` in `pywebio.output.page()`"
@@ -150,13 +150,8 @@ def next_client_event(self) -> dict:
         raise NotImplementedError
 
     def send_client_event(self, event):
-        """send event from client to session,
-        return True when this event is handled by this method, which means the subclass must ignore this event.
-        """
-        if event['event'] == 'page_close':
-            self.notify_page_lost(event['task_id'], event['data'])
-            return True
-        return False
+        """send event from client to session"""
+        raise NotImplementedError
 
     def get_task_commands(self) -> list:
         raise NotImplementedError

pywebio/session/coroutinebased.py

@@ -138,9 +138,8 @@ def send_client_event(self, event):
 
         :param dict event: 事件️消息
         """
-        handled = super(CoroutineBasedSession, self).send_client_event(event)
-        if handled:
-            return
+        if event['event'] == 'page_close':
+            self.notify_page_lost(event['task_id'], event['data'])
 
         coro_id = event['task_id']
         coro = self.coros.get(coro_id)

pywebio/session/threadbased.py

@@ -146,9 +146,8 @@ def send_client_event(self, event):
 
         :param dict event: 事件️消息
         """
-        handled = super(ThreadBasedSession, self).send_client_event(event)
-        if handled:
-            return
+        if event['event'] == 'page_close':
+            self.notify_page_lost(event['task_id'], event['data'])
 
         task_id = event['task_id']
         mq = self.task_mqs.get(task_id)