psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-10-31 15:57:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Made the log method a weakreference

Browse Source
This commit is contained in:
Sharvil Patankar 2024-02-06 11:03:44 -05:00
parent 00870545b7
commit f511222705
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/extras.py
View File

@ -29,8 +29,8 @@ and classes until a better place in the distribution is found.
import os as _os import os as _os
import time as _time import time as _time
import re as _re import re as _re
import weakref as _weakref
from collections import namedtuple, OrderedDict from collections import namedtuple, OrderedDict
import logging as _logging import logging as _logging
import psycopg2 import psycopg2
@ -401,9 +401,15 @@ class LoggingConnection(_connection):
self._logobj = logobj self._logobj = logobj
if _logging and isinstance( if _logging and isinstance(
logobj, (_logging.Logger, _logging.LoggerAdapter)): logobj, (_logging.Logger, _logging.LoggerAdapter)):
self.log = self._logtologger self._log = _weakref.WeakMethod(self._logtologger)
else: else:
self.log = self._logtofile self._log = _weakref.WeakMethod(self._logtofile)
def log(self, *args, **kwargs):
"""
Public interface of the log method defined in initialize
"""
return self._log()(*args, **kwargs)
def filter(self, msg, curs): def filter(self, msg, curs):
"""Filter the query before logging it. """Filter the query before logging it.