From 60cc09525273c523f4b85ee237d26755579ddff5 Mon Sep 17 00:00:00 2001 From: Siting Ren Date: Sun, 12 Feb 2023 16:23:13 +0000 Subject: [PATCH] add warnings --- vertica_python/vertica/connection.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/vertica_python/vertica/connection.py b/vertica_python/vertica/connection.py index 8a763134..2b01cdd3 100644 --- a/vertica_python/vertica/connection.py +++ b/vertica_python/vertica/connection.py @@ -42,6 +42,7 @@ import ssl import getpass import uuid +import warnings from struct import unpack from collections import deque, namedtuple import random @@ -518,7 +519,9 @@ def balance_load(self, raw_socket): raw_socket = self.establish_socket_connection(self.address_list) else: self._logger.debug('<= LoadBalanceResponse: %s', response) - self._logger.warning("Load balancing requested but not supported by server") + no_load_balancing = "Load balancing requested but not supported by server" + warnings.warn(no_load_balancing) + self._logger.warning(no_load_balancing) return raw_socket @@ -646,6 +649,8 @@ def handle_asynchronous_message(self, message): if getattr(self, 'notice_handler', None) is not None: self.notice_handler(message) else: + notice = f'[{message.severity}] {message.message}' + warnings.warn(notice) self._logger.warning(message.error_message()) def read_string(self): @@ -833,8 +838,9 @@ def startup_connection(self): self._logger.error(msg) raise errors.ConnectionError(msg) elif message.code == messages.Authentication.PASSWORD_GRACE: - self._logger.warning('The password for user {} will expire soon.' - ' Please consider changing it.'.format(self.options['user'])) + password_grace = f'The password for user {self.options["user"]} will expire soon. Please consider changing it.' + warnings.warn(password_grace) + self._logger.warning(password_grace) elif message.code == messages.Authentication.GSS: self.make_GSS_authentication() else: