mirror of https://github.com/zulip/zulip.git
33 lines
868 B
Python
33 lines
868 B
Python
from typing import TYPE_CHECKING, Dict, Optional
|
|
|
|
from django.conf import settings
|
|
|
|
if TYPE_CHECKING:
|
|
from zerver.tornado.event_queue import ClientDescriptor
|
|
|
|
descriptors_by_handler_id: Dict[int, "ClientDescriptor"] = {}
|
|
|
|
|
|
def get_descriptor_by_handler_id(handler_id: int) -> Optional["ClientDescriptor"]:
|
|
return descriptors_by_handler_id.get(handler_id)
|
|
|
|
|
|
def set_descriptor_by_handler_id(handler_id: int, client_descriptor: "ClientDescriptor") -> None:
|
|
descriptors_by_handler_id[handler_id] = client_descriptor
|
|
|
|
|
|
def clear_descriptor_by_handler_id(handler_id: int) -> None:
|
|
del descriptors_by_handler_id[handler_id]
|
|
|
|
|
|
current_port: Optional[int] = None
|
|
|
|
|
|
def is_current_port(port: int) -> Optional[int]:
|
|
return settings.TEST_SUITE or current_port == port
|
|
|
|
|
|
def set_current_port(port: int) -> None:
|
|
global current_port
|
|
current_port = port
|