2020-06-11 00:54:34 +02:00
|
|
|
from typing import TYPE_CHECKING, Dict, Optional
|
2016-11-27 06:14:48 +01:00
|
|
|
|
2022-09-22 22:09:34 +02:00
|
|
|
from django.conf import settings
|
|
|
|
|
2019-07-30 20:58:48 +02:00
|
|
|
if TYPE_CHECKING:
|
2018-03-15 01:21:27 +01:00
|
|
|
from zerver.tornado.event_queue import ClientDescriptor
|
2016-11-27 06:14:48 +01:00
|
|
|
|
python: Convert assignment type annotations to Python 3.6 style.
This commit was split by tabbott; this piece covers the vast majority
of files in Zulip, but excludes scripts/, tools/, and puppet/ to help
ensure we at least show the right error messages for Xenial systems.
We can likely further refine the remaining pieces with some testing.
Generated by com2ann, with whitespace fixes and various manual fixes
for runtime issues:
- invoiced_through: Optional[LicenseLedger] = models.ForeignKey(
+ invoiced_through: Optional["LicenseLedger"] = models.ForeignKey(
-_apns_client: Optional[APNsClient] = None
+_apns_client: Optional["APNsClient"] = None
- notifications_stream: Optional[Stream] = models.ForeignKey('Stream', related_name='+', null=True, blank=True, on_delete=CASCADE)
- signup_notifications_stream: Optional[Stream] = models.ForeignKey('Stream', related_name='+', null=True, blank=True, on_delete=CASCADE)
+ notifications_stream: Optional["Stream"] = models.ForeignKey('Stream', related_name='+', null=True, blank=True, on_delete=CASCADE)
+ signup_notifications_stream: Optional["Stream"] = models.ForeignKey('Stream', related_name='+', null=True, blank=True, on_delete=CASCADE)
- author: Optional[UserProfile] = models.ForeignKey('UserProfile', blank=True, null=True, on_delete=CASCADE)
+ author: Optional["UserProfile"] = models.ForeignKey('UserProfile', blank=True, null=True, on_delete=CASCADE)
- bot_owner: Optional[UserProfile] = models.ForeignKey('self', null=True, on_delete=models.SET_NULL)
+ bot_owner: Optional["UserProfile"] = models.ForeignKey('self', null=True, on_delete=models.SET_NULL)
- default_sending_stream: Optional[Stream] = models.ForeignKey('zerver.Stream', null=True, related_name='+', on_delete=CASCADE)
- default_events_register_stream: Optional[Stream] = models.ForeignKey('zerver.Stream', null=True, related_name='+', on_delete=CASCADE)
+ default_sending_stream: Optional["Stream"] = models.ForeignKey('zerver.Stream', null=True, related_name='+', on_delete=CASCADE)
+ default_events_register_stream: Optional["Stream"] = models.ForeignKey('zerver.Stream', null=True, related_name='+', on_delete=CASCADE)
-descriptors_by_handler_id: Dict[int, ClientDescriptor] = {}
+descriptors_by_handler_id: Dict[int, "ClientDescriptor"] = {}
-worker_classes: Dict[str, Type[QueueProcessingWorker]] = {}
-queues: Dict[str, Dict[str, Type[QueueProcessingWorker]]] = {}
+worker_classes: Dict[str, Type["QueueProcessingWorker"]] = {}
+queues: Dict[str, Dict[str, Type["QueueProcessingWorker"]]] = {}
-AUTH_LDAP_REVERSE_EMAIL_SEARCH: Optional[LDAPSearch] = None
+AUTH_LDAP_REVERSE_EMAIL_SEARCH: Optional["LDAPSearch"] = None
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
2020-04-22 01:09:50 +02:00
|
|
|
descriptors_by_handler_id: Dict[int, "ClientDescriptor"] = {}
|
2016-11-27 06:14:48 +01:00
|
|
|
|
2021-02-12 08:19:30 +01:00
|
|
|
|
2021-02-12 08:20:45 +01:00
|
|
|
def get_descriptor_by_handler_id(handler_id: int) -> Optional["ClientDescriptor"]:
|
2016-11-27 06:14:48 +01:00
|
|
|
return descriptors_by_handler_id.get(handler_id)
|
|
|
|
|
2021-02-12 08:19:30 +01:00
|
|
|
|
2021-02-12 08:20:45 +01:00
|
|
|
def set_descriptor_by_handler_id(handler_id: int, client_descriptor: "ClientDescriptor") -> None:
|
2016-11-27 06:14:48 +01:00
|
|
|
descriptors_by_handler_id[handler_id] = client_descriptor
|
|
|
|
|
2021-02-12 08:19:30 +01:00
|
|
|
|
2020-07-05 02:29:31 +02:00
|
|
|
def clear_descriptor_by_handler_id(handler_id: int) -> None:
|
2016-11-27 06:14:48 +01:00
|
|
|
del descriptors_by_handler_id[handler_id]
|
2022-09-22 22:09:34 +02:00
|
|
|
|
|
|
|
|
|
|
|
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
|