sensu-plugins-mongodb-mrtrotl 1.4.0

data/lib/gridfs/py.typed

@@ -0,0 +1,2 @@
+# PEP-561 Support File.
+# "Package maintainers who wish to support type checking of their code MUST add a marker file named py.typed to their package supporting typing".

data/lib/pymongo/__init__.py

@@ -0,0 +1,185 @@
+# Copyright 2009-present MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Python driver for MongoDB."""
+
+from typing import ContextManager, Optional, Tuple, Union
+
+__all__ = [
+    "ASCENDING",
+    "DESCENDING",
+    "GEO2D",
+    "GEOSPHERE",
+    "HASHED",
+    "TEXT",
+    "version_tuple",
+    "get_version_string",
+    "__version__",
+    "version",
+    "ReturnDocument",
+    "MAX_SUPPORTED_WIRE_VERSION",
+    "MIN_SUPPORTED_WIRE_VERSION",
+    "CursorType",
+    "MongoClient",
+    "DeleteMany",
+    "DeleteOne",
+    "IndexModel",
+    "InsertOne",
+    "ReplaceOne",
+    "UpdateMany",
+    "UpdateOne",
+    "ReadPreference",
+    "WriteConcern",
+    "has_c",
+    "timeout",
+]
+
+ASCENDING = 1
+"""Ascending sort order."""
+DESCENDING = -1
+"""Descending sort order."""
+
+GEO2D = "2d"
+"""Index specifier for a 2-dimensional `geospatial index`_.
+
+.. _geospatial index: http://mongodb.com/docs/manual/core/2d/
+"""
+
+GEOSPHERE = "2dsphere"
+"""Index specifier for a `spherical geospatial index`_.
+
+.. versionadded:: 2.5
+
+.. _spherical geospatial index: http://mongodb.com/docs/manual/core/2dsphere/
+"""
+
+HASHED = "hashed"
+"""Index specifier for a `hashed index`_.
+
+.. versionadded:: 2.5
+
+.. _hashed index: http://mongodb.com/docs/manual/core/index-hashed/
+"""
+
+TEXT = "text"
+"""Index specifier for a `text index`_.
+
+.. seealso:: MongoDB's `Atlas Search
+   <https://docs.atlas.mongodb.com/atlas-search/>`_ which offers more advanced
+   text search functionality.
+
+.. versionadded:: 2.7.1
+
+.. _text index: http://mongodb.com/docs/manual/core/index-text/
+"""
+
+version_tuple: Tuple[Union[int, str], ...] = (4, 2, 0)
+
+
+def get_version_string() -> str:
+    if isinstance(version_tuple[-1], str):
+        return ".".join(map(str, version_tuple[:-1])) + version_tuple[-1]
+    return ".".join(map(str, version_tuple))
+
+
+__version__: str = get_version_string()
+version = __version__
+
+"""Current version of PyMongo."""
+
+from pymongo import _csot
+from pymongo.collection import ReturnDocument
+from pymongo.common import MAX_SUPPORTED_WIRE_VERSION, MIN_SUPPORTED_WIRE_VERSION
+from pymongo.cursor import CursorType
+from pymongo.mongo_client import MongoClient
+from pymongo.operations import (
+    DeleteMany,
+    DeleteOne,
+    IndexModel,
+    InsertOne,
+    ReplaceOne,
+    UpdateMany,
+    UpdateOne,
+)
+from pymongo.read_preferences import ReadPreference
+from pymongo.write_concern import WriteConcern
+
+
+def has_c() -> bool:
+    """Is the C extension installed?"""
+    try:
+        from pymongo import _cmessage  # type: ignore[attr-defined] # noqa: F401
+
+        return True
+    except ImportError:
+        return False
+
+
+def timeout(seconds: Optional[float]) -> ContextManager:
+    """**(Provisional)** Apply the given timeout for a block of operations.
+
+    .. note:: :func:`~pymongo.timeout` is currently provisional. Backwards
+       incompatible changes may occur before becoming officially supported.
+
+    Use :func:`~pymongo.timeout` in a with-statement::
+
+      with pymongo.timeout(5):
+          client.db.coll.insert_one({})
+          client.db.coll2.insert_one({})
+
+    When the with-statement is entered, a deadline is set for the entire
+    block. When that deadline is exceeded, any blocking pymongo operation
+    will raise a timeout exception. For example::
+
+      try:
+          with pymongo.timeout(5):
+              client.db.coll.insert_one({})
+              time.sleep(5)
+              # The deadline has now expired, the next operation will raise
+              # a timeout exception.
+              client.db.coll2.insert_one({})
+      except PyMongoError as exc:
+          if exc.timeout:
+              print(f"block timed out: {exc!r}")
+          else:
+              print(f"failed with non-timeout error: {exc!r}")
+
+    When nesting :func:`~pymongo.timeout`, the newly computed deadline is capped to at most
+    the existing deadline. The deadline can only be shortened, not extended.
+    When exiting the block, the previous deadline is restored::
+
+      with pymongo.timeout(5):
+          coll.find_one()  # Uses the 5 second deadline.
+          with pymongo.timeout(3):
+              coll.find_one() # Uses the 3 second deadline.
+          coll.find_one()  # Uses the original 5 second deadline.
+          with pymongo.timeout(10):
+              coll.find_one()  # Still uses the original 5 second deadline.
+          coll.find_one()  # Uses the original 5 second deadline.
+
+    :Parameters:
+      - `seconds`: A non-negative floating point number expressing seconds, or None.
+
+    :Raises:
+      - :py:class:`ValueError`: When `seconds` is negative.
+
+    .. versionadded:: 4.2
+    """
+    if not isinstance(seconds, (int, float, type(None))):
+        raise TypeError("timeout must be None, an int, or a float")
+    if seconds and seconds < 0:
+        raise ValueError("timeout cannot be negative")
+    if seconds is not None:
+        seconds = float(seconds)
+    return _csot._TimeoutContext(seconds)

data/lib/pymongo/_csot.py

@@ -0,0 +1,118 @@
+# Copyright 2022-present MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you
+# may not use this file except in compliance with the License.  You
+# may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied.  See the License for the specific language governing
+# permissions and limitations under the License.
+
+"""Internal helpers for CSOT."""
+
+import functools
+import time
+from contextvars import ContextVar, Token
+from typing import Any, Callable, MutableMapping, Optional, Tuple, TypeVar, cast
+
+from pymongo.write_concern import WriteConcern
+
+TIMEOUT: ContextVar[Optional[float]] = ContextVar("TIMEOUT", default=None)
+RTT: ContextVar[float] = ContextVar("RTT", default=0.0)
+DEADLINE: ContextVar[float] = ContextVar("DEADLINE", default=float("inf"))
+
+
+def get_timeout() -> Optional[float]:
+    return TIMEOUT.get(None)
+
+
+def get_rtt() -> float:
+    return RTT.get()
+
+
+def get_deadline() -> float:
+    return DEADLINE.get()
+
+
+def set_rtt(rtt: float) -> None:
+    RTT.set(rtt)
+
+
+def remaining() -> Optional[float]:
+    if not get_timeout():
+        return None
+    return DEADLINE.get() - time.monotonic()
+
+
+def clamp_remaining(max_timeout: float) -> float:
+    """Return the remaining timeout clamped to a max value."""
+    timeout = remaining()
+    if timeout is None:
+        return max_timeout
+    return min(timeout, max_timeout)
+
+
+class _TimeoutContext(object):
+    """Internal timeout context manager.
+
+    Use :func:`pymongo.timeout` instead::
+
+      with pymongo.timeout(0.5):
+          client.test.test.insert_one({})
+    """
+
+    __slots__ = ("_timeout", "_tokens")
+
+    def __init__(self, timeout: Optional[float]):
+        self._timeout = timeout
+        self._tokens: Optional[Tuple[Token, Token, Token]] = None
+
+    def __enter__(self):
+        timeout_token = TIMEOUT.set(self._timeout)
+        prev_deadline = DEADLINE.get()
+        next_deadline = time.monotonic() + self._timeout if self._timeout else float("inf")
+        deadline_token = DEADLINE.set(min(prev_deadline, next_deadline))
+        rtt_token = RTT.set(0.0)
+        self._tokens = (timeout_token, deadline_token, rtt_token)
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self._tokens:
+            timeout_token, deadline_token, rtt_token = self._tokens
+            TIMEOUT.reset(timeout_token)
+            DEADLINE.reset(deadline_token)
+            RTT.reset(rtt_token)
+
+
+# See https://mypy.readthedocs.io/en/stable/generics.html?#decorator-factories
+F = TypeVar("F", bound=Callable[..., Any])
+
+
+def apply(func: F) -> F:
+    """Apply the client's timeoutMS to this operation."""
+
+    @functools.wraps(func)
+    def csot_wrapper(self, *args, **kwargs):
+        if get_timeout() is None:
+            timeout = self._timeout
+            if timeout is not None:
+                with _TimeoutContext(timeout):
+                    return func(self, *args, **kwargs)
+        return func(self, *args, **kwargs)
+
+    return cast(F, csot_wrapper)
+
+
+def apply_write_concern(cmd: MutableMapping, write_concern: Optional[WriteConcern]) -> None:
+    """Apply the given write concern to a command."""
+    if not write_concern or write_concern.is_server_default:
+        return
+    wc = write_concern.document
+    if get_timeout() is not None:
+        wc.pop("wtimeout", None)
+    if wc:
+        cmd["writeConcern"] = wc