googleapis · chalmerlowe · Apr 16, 2026 · Apr 17, 2026 · Apr 17, 2026 · Apr 17, 2026
@@ -23,7 +23,7 @@
 
 from google.cloud.ndb import version
 
-__version__ = version.__version__
+__version__: str = version.__version__
 
 from google.cloud.ndb.client import Client
 from google.cloud.ndb.context import AutoBatcher

@@ -117,18 +117,18 @@ def done_callback(self, cache_call):
         """
         exception = cache_call.exception()
         if exception:
-            for future in self.futures:
+            for future in self.futures:  # type: ignore[attr-defined]
                 future.set_exception(exception)
 
         else:
-            for future in self.futures:
+            for future in self.futures:  # type: ignore[attr-defined]
                 future.set_result(None)
 
     def make_call(self):
         """Make the actual call to the global cache. To be overridden."""
         raise NotImplementedError
 
-    def future_info(self, key):
+    def future_info(self, key, value=None):
         """Generate info string for Future. To be overridden."""
         raise NotImplementedError
 
@@ -279,7 +279,7 @@ def make_call(self):
         """Call :method:`GlobalCache.get`."""
         return _global_cache().get(self.keys)
 
-    def future_info(self, key):
+    def future_info(self, key, value=None):
         """Generate info string for Future."""
         return "GlobalCache.get({})".format(key)
 
@@ -373,7 +373,7 @@ def make_call(self):
         """Call :method:`GlobalCache.set`."""
         return _global_cache().set(self.todo, expires=self.expires)
 
-    def future_info(self, key, value):
+    def future_info(self, key, value=None):
         """Generate info string for Future."""
         return "GlobalCache.set({}, {})".format(key, value)
 
@@ -436,7 +436,7 @@ def make_call(self):
         """Call :method:`GlobalCache.set`."""
         return _global_cache().set_if_not_exists(self.todo, expires=self.expires)
 
-    def future_info(self, key, value):
+    def future_info(self, key, value=None):
         """Generate info string for Future."""
         return "GlobalCache.set_if_not_exists({}, {})".format(key, value)
 
@@ -482,7 +482,7 @@ def make_call(self):
         """Call :method:`GlobalCache.delete`."""
         return _global_cache().delete(self.keys)
 
-    def future_info(self, key):
+    def future_info(self, key, value=None):
         """Generate info string for Future."""
         return "GlobalCache.delete({})".format(key)
 
@@ -513,7 +513,7 @@ def make_call(self):
         """Call :method:`GlobalCache.watch`."""
         return _global_cache().watch(self.todo)
 
-    def future_info(self, key, value):
+    def future_info(self, key, value=None):
         """Generate info string for Future."""
         return "GlobalCache.watch({}, {})".format(key, value)
 
@@ -543,7 +543,7 @@ def make_call(self):
         """Call :method:`GlobalCache.unwatch`."""
         return _global_cache().unwatch(self.keys)
 
-    def future_info(self, key):
+    def future_info(self, key, value=None):
         """Generate info string for Future."""
         return "GlobalCache.unwatch({})".format(key)
 
@@ -580,7 +580,7 @@ def make_call(self):
         """Call :method:`GlobalCache.compare_and_swap`."""
         return _global_cache().compare_and_swap(self.todo, expires=self.expires)
 
-    def future_info(self, key, value):
+    def future_info(self, key, value=None):
         """Generate info string for Future."""
         return "GlobalCache.compare_and_swap({}, {})".format(key, value)
 
@@ -627,8 +627,7 @@ def global_lock_for_write(key):
         tasklets.Future: Eventual result will be a lock value to be used later with
             :func:`global_unlock`.
     """
-    lock = "." + str(uuid.uuid4())
-    lock = lock.encode("ascii")
+    lock = ("." + str(uuid.uuid4())).encode("ascii")
     utils.logging_debug(log, "lock for write: {}", lock)
 
     def new_value(old_value):

@@ -168,7 +168,7 @@ def lookup(key, options):
         if use_global_cache and not key_locked:
             if entity_pb is not _NOT_FOUND:
                 expires = context._global_cache_timeout(key, options)
-                serialized = entity_pb._pb.SerializeToString()
+                serialized = entity_pb._pb.SerializeToString()  # type: ignore[attr-defined]
                 yield _cache.global_compare_and_swap(
                     cache_key, serialized, expires=expires
                 )

@@ -342,6 +342,8 @@ def has_next_async(self):
         if self._batch is None:
             yield self._next_batch()  # First time
 
+        assert self._batch is not None
+        assert self._index is not None
         if self._index < len(self._batch):
             raise tasklets.Return(True)
 
@@ -359,7 +361,9 @@ def probably_has_next(self):
         return (
             self._batch is None  # Haven't even started yet
             or self._has_next_batch  # There's another batch to fetch
-            or self._index < len(self._batch)  # Not done with current batch
+            or (
+                self._index is not None and self._index < len(self._batch)
+            )  # Not done with current batch
         )
 
     @tasklets.tasklet
@@ -421,6 +425,8 @@ def next(self):
             self._cursor_before = None
             raise StopIteration
 
+        assert self._batch is not None
+        assert self._index is not None
         # Won't block
         next_result = self._batch[self._index]
         self._index += 1
@@ -446,7 +452,7 @@ def _peek(self):
         batch = self._batch
         index = self._index
 
-        if batch and index < len(batch):
+        if batch and index is not None and index < len(batch):
             return batch[index]
 
         raise KeyError(index)
@@ -554,6 +560,7 @@ def next(self):
         if not self.has_next():
             raise StopIteration()
 
+        assert self._next_result is not None
         # Won't block
         next_result = self._next_result
         self._next_result = None
@@ -718,6 +725,7 @@ def next(self):
         if not self.has_next():
             raise StopIteration()
 
+        assert self._next_result is not None
         # Won't block
         next_result = self._next_result
         self._next_result = None
@@ -949,7 +957,7 @@ def _query_to_protobuf(query):
             filter_pb = ancestor_filter_pb
 
         elif isinstance(filter_pb, query_pb2.CompositeFilter):
-            filter_pb.filters._pb.add(property_filter=ancestor_filter_pb._pb)
+            filter_pb.filters._pb.add(property_filter=ancestor_filter_pb._pb)  # type: ignore[attr-defined]
 
         else:
             filter_pb = query_pb2.CompositeFilter(

@@ -116,12 +116,15 @@ class EventLoop(object):
     """
 
     def __init__(self):
-        self.current = collections.deque()
-        self.idlers = collections.deque()
+        self._init()
+
+    def _init(self):
+        self.current: collections.deque = collections.deque()
+        self.idlers: collections.deque = collections.deque()
         self.inactive = 0
         self.queue = []
         self.rpcs = {}
-        self.rpc_results = queue.Queue()
+        self.rpc_results: queue.Queue = queue.Queue()
 
     def clear(self):
         """Remove all pending events without running any."""
@@ -139,7 +142,7 @@ def clear(self):
                 utils.logging_debug(log, "  queue = {}", queue)
             if rpcs:
                 utils.logging_debug(log, "  rpcs = {}", rpcs)
-            self.__init__()
+            self._init()
             current.clear()
             idlers.clear()
             queue[:] = []

@@ -1,6 +1,7 @@
 import datetime
 import re
 import time
+from typing import Any
 
 from google.cloud.ndb import context as context_module
 from google.cloud.ndb import exceptions
@@ -485,7 +486,7 @@ def _Literal(self):
                 a string, integer, floating point value, boolean or None).
         """
 
-        literal = None
+        literal: Any = None
 
         if self._next_symbol < len(self._symbols):
             try:
@@ -770,27 +771,34 @@ def _raise_cast_error(message):
 
 
 def _time_function(values):
+    t_tuple: tuple[int, ...]
     if len(values) == 1:
         value = values[0]
         if isinstance(value, str):
             try:
-                time_tuple = time.strptime(value, "%H:%M:%S")
+                st = time.strptime(value, "%H:%M:%S")
             except ValueError as error:
                 _raise_cast_error(
                     "Error during time conversion, {}, {}".format(error, values)
                 )
-            time_tuple = time_tuple[3:]
-            time_tuple = time_tuple[0:3]
+            t_tuple = (st.tm_hour, st.tm_min, st.tm_sec)
         elif isinstance(value, int):
-            time_tuple = (value,)
+            t_tuple = (value,)
         else:
             _raise_cast_error("Invalid argument for time(), {}".format(value))
     elif len(values) < 4:
-        time_tuple = tuple(values)
+        t_tuple = tuple(values)
     else:
         _raise_cast_error("Too many arguments for time(), {}".format(values))
     try:
-        return datetime.time(*time_tuple)
+        if len(t_tuple) == 1:
+            return datetime.time(t_tuple[0])
+        elif len(t_tuple) == 2:
+            return datetime.time(t_tuple[0], t_tuple[1])
+        elif len(t_tuple) == 3:
+            return datetime.time(t_tuple[0], t_tuple[1], t_tuple[2])
+        else:
+            _raise_cast_error("Invalid arguments for time()")
     except ValueError as error:
         _raise_cast_error("Error during time conversion, {}, {}".format(error, values))
 

@@ -389,11 +389,10 @@ class Property(ProtocolBuffer.ProtocolMessage):
         24: "EMPTY_LIST",
     }
 
+    @classmethod
     def Meaning_Name(cls, x):
         return cls._Meaning_NAMES.get(x, "")
 
-    Meaning_Name = classmethod(Meaning_Name)
-
     has_meaning_ = 0
     meaning_ = 0
     has_meaning_uri_ = 0
@@ -526,7 +525,7 @@ class Path_Element(ProtocolBuffer.ProtocolMessage):
     def type(self):
         # Force legacy byte-str to be a str.
         if type(self.type_) is bytes:
-            return self.type_.decode()
+            return self.type_.decode()  # type: ignore[attr-defined]
         return self.type_
 
     def set_type(self, x):

@@ -32,6 +32,9 @@ def MergePartialFromString(self, s):
         d = Decoder(a, 0, len(a))
         self.TryMerge(d)
 
+    def TryMerge(self, d):
+        raise NotImplementedError
+
 
 class Decoder:
     NUMERIC = 0

@@ -19,11 +19,16 @@
 import logging
 
 from google.cloud.ndb import exceptions
+from typing import Any
 
 log = logging.getLogger(__name__)
 
 
 class Options(object):
+    max_memcache_items: Any
+    force_writes: Any
+    propagation: Any
+
     __slots__ = (
         # Supported
         "retries",

@@ -41,7 +41,7 @@ def __init__(self, future, info):
         self.future = future
         self.info = info
         self.start_time = time.time()
-        self.elapsed_time = 0
+        self.elapsed_time = 0.0
 
         def record_time(future):
             self.elapsed_time = time.time() - self.start_time

@@ -33,7 +33,7 @@ def wraps_safely(obj, attr_names=functools.WRAPPER_ASSIGNMENTS):
     are not copied to the wrappers and thus cause attribute errors. This
     wrapper prevents that problem."""
     return functools.wraps(
-        obj, assigned=(name for name in attr_names if hasattr(obj, name))
+        obj, assigned=tuple(name for name in attr_names if hasattr(obj, name))
     )
 
 
@@ -84,7 +84,7 @@ def retry_wrapper(*args, **kwargs):
                 error = e
             except BaseException as e:
                 # `e` is removed from locals at end of block
-                error = e  # See: https://goo.gl/5J8BMK
+                error = e  # type: ignore[assignment]  # See: https://goo.gl/5J8BMK
 
                 if not is_transient_error(error):
                     # If we are in an inner retry block, use special nested
@@ -107,7 +107,7 @@ def retry_wrapper(*args, **kwargs):
 
         # Unknown errors really want to show up as None, so manually set the error.
         if isinstance(error, core_exceptions.Unknown):
-            error = "google.api_core.exceptions.Unknown"
+            error = "google.api_core.exceptions.Unknown"  # type: ignore[assignment]
 
         raise core_exceptions.RetryError(
             "Maximum number of {} retries exceeded while calling {}".format(

@@ -257,8 +257,8 @@ def _transaction_async(context, callback, read_only=False):
     transaction_id = yield _datastore_api.begin_transaction(read_only, retries=0)
     utils.logging_debug(log, "Transaction Id: {}", transaction_id)
 
-    on_commit_callbacks = []
-    transaction_complete_callbacks = []
+    on_commit_callbacks: list = []
+    transaction_complete_callbacks: list = []
     tx_context = context.new(
         transaction=transaction_id,
         on_commit_callbacks=on_commit_callbacks,

@@ -22,6 +22,7 @@
 import os
 import threading
 import uuid
+from typing import Any, cast
 
 from google.cloud.ndb import _eventloop
 from google.cloud.ndb import exceptions
@@ -254,6 +255,11 @@ class _Context(_ContextTuple):
         client (client.Client): The NDB client for this context.
     """
 
+    cache_policy: Any
+    global_cache_policy: Any
+    global_cache_timeout_policy: Any
+    datastore_policy: Any
+
     def __new__(
         cls,
         client,
@@ -313,11 +319,12 @@ def __new__(
             legacy_data=legacy_data,
         )
 
-        context.set_cache_policy(cache_policy)
-        context.set_global_cache_policy(global_cache_policy)
-        context.set_global_cache_timeout_policy(global_cache_timeout_policy)
-        context.set_datastore_policy(datastore_policy)
-        context.set_retry_state(retry)
+        ctx = cast(Any, context)
+        ctx.set_cache_policy(cache_policy)
+        ctx.set_global_cache_policy(global_cache_policy)
+        ctx.set_global_cache_timeout_policy(global_cache_timeout_policy)
+        ctx.set_datastore_policy(datastore_policy)
+        ctx.set_retry_state(retry)
 
         return context