This commit is contained in:
parent
9a12cbcdde
commit
e103b283cb
|
@ -9,8 +9,6 @@ from sqlalchemy.ext.declarative import declarative_base
|
|||
from sqlalchemy.orm import Session, configure_mappers
|
||||
from sqlalchemy.sql.schema import Table
|
||||
from sqlalchemy_searchable import make_searchable
|
||||
from sqlalchemy.sql import Executable
|
||||
from sqlalchemy.dialects.postgresql.psycopg2 import PGExecutionContext
|
||||
|
||||
from services.logger import root_logger as logger
|
||||
from settings import DB_URL
|
||||
|
@ -97,16 +95,12 @@ def before_cursor_execute(conn, cursor, statement, parameters, context, executem
|
|||
|
||||
@event.listens_for(Engine, 'after_cursor_execute')
|
||||
def after_cursor_execute(conn, cursor, statement, parameters, context, executemany):
|
||||
if not isinstance(statement, Executable):
|
||||
return
|
||||
compiled_statement = context.compiled.string
|
||||
compiled_parameters = context.compiled.params
|
||||
if compiled_statement:
|
||||
elapsed = time.time() - conn.query_start_time
|
||||
query = compiled_statement % compiled_parameters
|
||||
|
||||
if isinstance(context, PGExecutionContext):
|
||||
compiled_statement = context.compiled.string
|
||||
compiled_parameters = context.compiled.params
|
||||
if compiled_statement:
|
||||
elapsed = time.time() - conn.query_start_time
|
||||
query = compiled_statement % compiled_parameters
|
||||
|
||||
if elapsed > 1 and conn.last_statement != query:
|
||||
conn.last_statement = query
|
||||
logger.debug(f"\n{query}\n{'*' * math.floor(elapsed)} {elapsed:.3f} s\n")
|
||||
if elapsed > 1 and conn.last_statement != query:
|
||||
conn.last_statement = query
|
||||
logger.debug(f"\n{query}\n{'*' * math.floor(elapsed)} {elapsed:.3f} s\n")
|
||||
|
|
Loading…
Reference in New Issue
Block a user