mongomock-ng 7.0.0__py3-none-any.whl

mongomock_ng/__init__.py

@@ -0,0 +1,149 @@
+import os
+
+
+try:
+    from pymongo.errors import PyMongoError
+except ImportError:
+
+    class _FallbackPyMongoError(Exception):  # type: ignore[misc]
+        pass
+
+    PyMongoError = _FallbackPyMongoError
+
+
+try:
+    from pymongo.errors import OperationFailure
+except ImportError:
+
+    class _FallbackOperationFailure(PyMongoError):
+        def __init__(self, message, code=None, details=None):
+            super().__init__()
+            self._message = message
+            self._code = code
+            self._details = details
+
+        code = property(lambda self: self._code)
+        details = property(lambda self: self._details)
+
+        def __str__(self):
+            return self._message
+
+    OperationFailure = _FallbackOperationFailure
+
+
+try:
+    from pymongo.errors import WriteError
+except ImportError:
+
+    class _FallbackWriteError(OperationFailure):
+        pass
+
+    WriteError = _FallbackWriteError
+
+
+try:
+    from pymongo.errors import DuplicateKeyError
+except ImportError:
+
+    class _FallbackDuplicateKeyError(WriteError):
+        pass
+
+    DuplicateKeyError = _FallbackDuplicateKeyError
+
+
+try:
+    from pymongo.errors import BulkWriteError
+except ImportError:
+
+    class _FallbackBulkWriteError(OperationFailure):
+        def __init__(self, results):
+            super().__init__('batch op errors occurred', 65, results)
+
+    BulkWriteError = _FallbackBulkWriteError
+
+
+try:
+    from pymongo.errors import CollectionInvalid
+except ImportError:
+
+    class _FallbackCollectionInvalid(PyMongoError):
+        pass
+
+    CollectionInvalid = _FallbackCollectionInvalid
+
+
+try:
+    from pymongo.errors import InvalidName
+except ImportError:
+
+    class _FallbackInvalidName(PyMongoError):
+        pass
+
+    InvalidName = _FallbackInvalidName
+
+
+try:
+    from pymongo.errors import InvalidOperation
+except ImportError:
+
+    class _FallbackInvalidOperation(PyMongoError):
+        pass
+
+    InvalidOperation = _FallbackInvalidOperation
+
+
+try:
+    from pymongo.errors import ConfigurationError
+except ImportError:
+
+    class _FallbackConfigurationError(PyMongoError):
+        pass
+
+    ConfigurationError = _FallbackConfigurationError
+
+
+try:
+    from pymongo.errors import InvalidURI
+except ImportError:
+
+    class _FallbackInvalidURI(ConfigurationError):
+        pass
+
+    InvalidURI = _FallbackInvalidURI
+
+
+from .helpers import ObjectId, utcnow  # noqa
+from .__version__ import __version__
+
+
+__all__ = [
+    'SERVER_VERSION',
+    'Collection',
+    'CollectionInvalid',
+    'Database',
+    'DuplicateKeyError',
+    'InvalidName',
+    'InvalidURI',
+    'MongoClient',
+    'ObjectId',
+    'OperationFailure',
+    'WriteConcern',
+    '__version__',
+    'ignore_feature',
+    'patch',
+    'warn_on_feature',
+]
+
+from .collection import Collection
+from .database import Database
+from .mongo_client import MongoClient
+from .not_implemented import ignore_feature
+from .not_implemented import warn_on_feature
+from .patch import patch
+from .write_concern import WriteConcern
+
+
+# The version of the server faked by mongomock. Callers may patch it before creating connections to
+# update the behavior of mongomock.
+# Keep the default version in sync with docker-compose.yml and travis.yml.
+SERVER_VERSION = os.getenv('MONGODB', '7.0.34')

mongomock_ng/__init__.pyi

@@ -0,0 +1,33 @@
+from typing import Any, Callable, Dict, List, Literal, Optional, Sequence, Tuple, Union
+from unittest import mock
+
+from bson.objectid import ObjectId as ObjectId
+from pymongo import MongoClient as MongoClient
+from pymongo.collection import Collection as Collection
+from pymongo.database import Database as Database
+from pymongo.errors import (
+    BulkWriteError as BulkWriteError,
+    CollectionInvalid as CollectionInvalid,
+    ConfigurationError as ConfigurationError,
+    DuplicateKeyError as DuplicateKeyError,
+    InvalidName as InvalidName,
+    InvalidOperation as InvalidOperation,
+    InvalidURI as InvalidURI,
+    OperationFailure as OperationFailure,
+    PyMongoError as PyMongoError,
+    WriteError as WriteError,
+)
+
+from .write_concern import WriteConcern as WriteConcern
+
+def patch(
+    servers: Union[str, Tuple[str, int], Sequence[Union[str, Tuple[str, int]]]] = ...,
+    on_new: Literal['error', 'create', 'timeout', 'pymongo'] = ...,
+) -> mock._patch: ...
+
+_FeatureName = Literal['collation', 'session']
+
+def ignore_feature(feature: _FeatureName) -> None: ...
+def warn_on_feature(feature: _FeatureName) -> None: ...
+
+SERVER_VERSION: str = ...

mongomock_ng/__version__.py

@@ -0,0 +1,4 @@
+from importlib.metadata import version
+
+
+__version__ = version('mongomock_ng')