From 50fe43abc12184943933d0e4c4126a032f193e27 Mon Sep 17 00:00:00 2001 From: Furr Date: Thu, 17 Jan 2019 12:08:21 -0500 Subject: [PATCH] import warnings --- mongolog/exceptions.py | 4 ++++ mongolog/handlers.py | 10 +++++++++- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/mongolog/exceptions.py b/mongolog/exceptions.py index 87c7913..f90f105 100644 --- a/mongolog/exceptions.py +++ b/mongolog/exceptions.py @@ -8,3 +8,7 @@ def __init__(self, *args, **kwargs): class LogConfigError(ValueError): pass + + +class UnsupportedVersionError(ValueError): + pass diff --git a/mongolog/handlers.py b/mongolog/handlers.py index 332b6e2..8e0410f 100644 --- a/mongolog/handlers.py +++ b/mongolog/handlers.py @@ -5,6 +5,8 @@ from logging import Handler, NOTSET from datetime import datetime as dt import sys +import warnings + try: from cStringIO import StringIO # noqa except ImportError: @@ -20,7 +22,10 @@ from pymongo.collection import ReturnDocument from mongolog.models import LogRecord -from mongolog.exceptions import MissingConnectionError +from mongolog.exceptions import ( + MissingConnectionError, + UnsupportedVersionError +) logger = logging.getLogger('') console = logging.getLogger('mongolog-int') @@ -302,6 +307,9 @@ def insert_embedded(self, log_record): self.mongolog.insert_one(log_record) elif pymongo_version == 2: self.mongolog.insert(log_record) + warnings.warn("pymongo version 2 is deprecated", DeprecationWarning) + else: + raise UnsupportedVersionError("mongolog currently on supports pymongo >= 2") else: # record has been seen before so we update the counter and push/pop # the log record time. We keep the 'n' latest log record