Skip to content

Commit 9701a04

Browse files
committed
fix typing
1 parent bc8bdb4 commit 9701a04

File tree

3 files changed

+141
-43
lines changed

3 files changed

+141
-43
lines changed

pymongo/asynchronous/topology.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def __init__(self, topology_settings: TopologySettings):
111111
self._publish_tp = self._listeners is not None and self._listeners.enabled_for_topology
112112

113113
# Create events queue if there are publishers.
114-
self._events = None
114+
self._events: queue.Queue[Any] | None = None
115115
self.__events_executor: Any = None
116116

117117
if self._publish_server or self._publish_tp:
@@ -126,6 +126,7 @@ def __init__(self, topology_settings: TopologySettings):
126126

127127
if self._publish_tp:
128128
assert self._events is not None
129+
assert self._listeners is not None
129130
self._events.put((self._listeners.publish_topology_opened, (self._topology_id,)))
130131
self._settings = topology_settings
131132
topology_description = TopologyDescription(
@@ -143,6 +144,7 @@ def __init__(self, topology_settings: TopologySettings):
143144
)
144145
if self._publish_tp:
145146
assert self._events is not None
147+
assert self._listeners is not None
146148
self._events.put(
147149
(
148150
self._listeners.publish_topology_description_changed,
@@ -161,6 +163,7 @@ def __init__(self, topology_settings: TopologySettings):
161163
for seed in topology_settings.seeds:
162164
if self._publish_server:
163165
assert self._events is not None
166+
assert self._listeners is not None
164167
self._events.put((self._listeners.publish_server_opened, (seed, self._topology_id)))
165168
if _SDAM_LOGGER.isEnabledFor(logging.DEBUG):
166169
_debug_log(
@@ -491,6 +494,7 @@ async def _process_change(
491494
suppress_event = sd_old == server_description
492495
if self._publish_server and not suppress_event:
493496
assert self._events is not None
497+
assert self._listeners is not None
494498
self._events.put(
495499
(
496500
self._listeners.publish_server_description_changed,
@@ -503,6 +507,7 @@ async def _process_change(
503507

504508
if self._publish_tp and not suppress_event:
505509
assert self._events is not None
510+
assert self._listeners is not None
506511
self._events.put(
507512
(
508513
self._listeners.publish_topology_description_changed,
@@ -570,6 +575,7 @@ async def _process_srv_update(self, seedlist: list[tuple[str, Any]]) -> None:
570575

571576
if self._publish_tp:
572577
assert self._events is not None
578+
assert self._listeners is not None
573579
self._events.put(
574580
(
575581
self._listeners.publish_topology_description_changed,
@@ -723,6 +729,7 @@ async def close(self) -> None:
723729
# Publish only after releasing the lock.
724730
if self._publish_tp:
725731
assert self._events is not None
732+
assert self._listeners is not None
726733
self._description = TopologyDescription(
727734
TOPOLOGY_TYPE.Unknown,
728735
{},

pymongo/synchronous/topology.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def __init__(self, topology_settings: TopologySettings):
111111
self._publish_tp = self._listeners is not None and self._listeners.enabled_for_topology
112112

113113
# Create events queue if there are publishers.
114-
self._events = None
114+
self._events: queue.Queue[Any] | None = None
115115
self.__events_executor: Any = None
116116

117117
if self._publish_server or self._publish_tp:
@@ -126,6 +126,7 @@ def __init__(self, topology_settings: TopologySettings):
126126

127127
if self._publish_tp:
128128
assert self._events is not None
129+
assert self._listeners is not None
129130
self._events.put((self._listeners.publish_topology_opened, (self._topology_id,)))
130131
self._settings = topology_settings
131132
topology_description = TopologyDescription(
@@ -143,6 +144,7 @@ def __init__(self, topology_settings: TopologySettings):
143144
)
144145
if self._publish_tp:
145146
assert self._events is not None
147+
assert self._listeners is not None
146148
self._events.put(
147149
(
148150
self._listeners.publish_topology_description_changed,
@@ -161,6 +163,7 @@ def __init__(self, topology_settings: TopologySettings):
161163
for seed in topology_settings.seeds:
162164
if self._publish_server:
163165
assert self._events is not None
166+
assert self._listeners is not None
164167
self._events.put((self._listeners.publish_server_opened, (seed, self._topology_id)))
165168
if _SDAM_LOGGER.isEnabledFor(logging.DEBUG):
166169
_debug_log(
@@ -491,6 +494,7 @@ def _process_change(
491494
suppress_event = sd_old == server_description
492495
if self._publish_server and not suppress_event:
493496
assert self._events is not None
497+
assert self._listeners is not None
494498
self._events.put(
495499
(
496500
self._listeners.publish_server_description_changed,
@@ -503,6 +507,7 @@ def _process_change(
503507

504508
if self._publish_tp and not suppress_event:
505509
assert self._events is not None
510+
assert self._listeners is not None
506511
self._events.put(
507512
(
508513
self._listeners.publish_topology_description_changed,
@@ -570,6 +575,7 @@ def _process_srv_update(self, seedlist: list[tuple[str, Any]]) -> None:
570575

571576
if self._publish_tp:
572577
assert self._events is not None
578+
assert self._listeners is not None
573579
self._events.put(
574580
(
575581
self._listeners.publish_topology_description_changed,
@@ -721,6 +727,7 @@ def close(self) -> None:
721727
# Publish only after releasing the lock.
722728
if self._publish_tp:
723729
assert self._events is not None
730+
assert self._listeners is not None
724731
self._description = TopologyDescription(
725732
TOPOLOGY_TYPE.Unknown,
726733
{},

0 commit comments

Comments
 (0)