.config/coc/extensions/node_modules/coc-pyright/node_modules/pyright/dist/typeshed-fallback/stubs/google-cloud-ndb/google/cloud/ndb/context.pyi

   1 from typing import Any, Callable, NamedTuple
   2
   3 from google.cloud.ndb import Key, exceptions as exceptions
   4
   5 class _LocalState:
   6     def __init__(self) -> None: ...
   7     @property
   8     def context(self): ...
   9     @context.setter
  10     def context(self, value) -> None: ...
  11     @property
  12     def toplevel_context(self): ...
  13     @toplevel_context.setter
  14     def toplevel_context(self, value) -> None: ...
  15
  16 def get_context(raise_context_error: bool = ...): ...
  17 def get_toplevel_context(raise_context_error: bool = ...): ...
  18
  19 class _ContextTuple(NamedTuple):
  20     client: Any
  21     namespace: Any
  22     eventloop: Any
  23     batches: Any
  24     commit_batches: Any
  25     transaction: Any
  26     cache: Any
  27     global_cache: Any
  28     on_commit_callbacks: Any
  29     legacy_data: Any
  30
  31 class _Context(_ContextTuple):
  32     def __new__(
  33         cls,
  34         client,
  35         namespace=...,
  36         eventloop: Any | None = ...,
  37         batches: Any | None = ...,
  38         commit_batches: Any | None = ...,
  39         transaction: Any | None = ...,
  40         cache: Any | None = ...,
  41         cache_policy: Any | None = ...,
  42         global_cache: Any | None = ...,
  43         global_cache_flush_keys: object | None = ...,
  44         global_cache_policy: Callable[[Key], bool] | None = ...,
  45         global_cache_timeout_policy: Any | None = ...,
  46         datastore_policy: Any | None = ...,
  47         on_commit_callbacks: Any | None = ...,
  48         legacy_data: bool = ...,
  49         retry: Any | None = ...,
  50         rpc_time: Any | None = ...,
  51         wait_time: Any | None = ...,
  52     ): ...
  53     def new(self, **kwargs): ...
  54     rpc_time: int
  55     wait_time: int
  56     def use(self) -> None: ...
  57
  58 class Context(_Context):
  59     def clear_cache(self) -> None: ...
  60     def flush(self) -> None: ...
  61     def get_namespace(self): ...
  62     def get_cache_policy(self): ...
  63     def get_datastore_policy(self) -> None: ...
  64     def get_global_cache_policy(self): ...
  65     get_memcache_policy: Any
  66     def get_global_cache_timeout_policy(self): ...
  67     get_memcache_timeout_policy: Any
  68     cache_policy: Any
  69     def set_cache_policy(self, policy): ...
  70     datastore_policy: Any
  71     def set_datastore_policy(self, policy): ...
  72     global_cache_policy: Any
  73     def set_global_cache_policy(self, policy): ...
  74     set_memcache_policy: Any
  75     global_cache_timeout_policy: Any
  76     def set_global_cache_timeout_policy(self, policy): ...
  77     set_memcache_timeout_policy: Any
  78     def get_retry_state(self): ...
  79     def set_retry_state(self, state) -> None: ...
  80     def clear_retry_state(self) -> None: ...
  81     def call_on_commit(self, callback) -> None: ...
  82     def in_transaction(self): ...
  83     def in_retry(self): ...
  84     def memcache_add(self, *args, **kwargs) -> None: ...
  85     def memcache_cas(self, *args, **kwargs) -> None: ...
  86     def memcache_decr(self, *args, **kwargs) -> None: ...
  87     def memcache_delete(self, *args, **kwargs) -> None: ...
  88     def memcache_get(self, *args, **kwargs) -> None: ...
  89     def memcache_gets(self, *args, **kwargs) -> None: ...
  90     def memcache_incr(self, *args, **kwargs) -> None: ...
  91     def memcache_replace(self, *args, **kwargs) -> None: ...
  92     def memcache_set(self, *args, **kwargs) -> None: ...
  93     def urlfetch(self, *args, **kwargs) -> None: ...
  94
  95 class ContextOptions:
  96     def __init__(self, *args, **kwargs) -> None: ...
  97
  98 class TransactionOptions:
  99     NESTED: int
 100     MANDATORY: int
 101     ALLOWED: int
 102     INDEPENDENT: int
 103
 104 class AutoBatcher:
 105     def __init__(self, *args, **kwargs) -> None: ...