__init__.py 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. try:
  2. from pymongo.errors import PyMongoError
  3. except ImportError:
  4. class PyMongoError(Exception):
  5. pass
  6. try:
  7. from pymongo.errors import DuplicateKeyError
  8. except ImportError:
  9. class DuplicateKeyError(PyMongoError):
  10. pass
  11. try:
  12. from pymongo.errors import BulkWriteError
  13. except ImportError:
  14. class BulkWriteError(PyMongoError):
  15. pass
  16. try:
  17. from pymongo.errors import OperationFailure
  18. except ImportError:
  19. class OperationFailure(PyMongoError):
  20. pass
  21. try:
  22. from pymongo.errors import CollectionInvalid
  23. except ImportError:
  24. class CollectionInvalid(PyMongoError):
  25. pass
  26. try:
  27. from pymongo.errors import InvalidName
  28. except ImportError:
  29. class InvalidName(PyMongoError):
  30. pass
  31. try:
  32. from pymongo.errors import InvalidOperation
  33. except ImportError:
  34. class InvalidOperation(PyMongoError):
  35. pass
  36. try:
  37. from pymongo.errors import ConfigurationError
  38. except ImportError:
  39. class ConfigurationError(PyMongoError):
  40. pass
  41. try:
  42. from pymongo.errors import InvalidURI
  43. except ImportError:
  44. class InvalidURI(ConfigurationError):
  45. pass
  46. try:
  47. from pymongo.errors import WriteError
  48. except ImportError:
  49. class WriteError(OperationFailure):
  50. pass
  51. from .helpers import ObjectId # noqa
  52. from mongomock.__version__ import __version__
  53. __all__ = [
  54. '__version__',
  55. 'Database',
  56. 'DuplicateKeyError',
  57. 'Collection',
  58. 'CollectionInvalid',
  59. 'InvalidName',
  60. 'MongoClient',
  61. 'ObjectId',
  62. 'OperationFailure',
  63. 'WriteConcern'
  64. ]
  65. from .collection import Collection
  66. from .database import Database
  67. from .mongo_client import MongoClient
  68. from .write_concern import WriteConcern