code: def after_this_websocket(func: Callable) -> Callable:
      
      _websocket_ctx_stack.top._after_websocket_functions.append(func)
      return func
docs: Schedule the func to be called after the current websocket.
  
      This is useful in situations whereby you want an after websocket
      function for a specific route or circumstance only, for example,
  
      .. note::
          The response is an optional argument, and will only be
          passed if the websocket was not active (i.e. there was an
          error).
  
      .. code-block:: python
  
          def index():
              @after_this_websocket
              def set_cookie(response: Optional[Response]):
                  response.set_cookie('special', 'value')
                  return response
  
              ...