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.orm import Session, configure_mappers
|
||||||
from sqlalchemy.sql.schema import Table
|
from sqlalchemy.sql.schema import Table
|
||||||
from sqlalchemy_searchable import make_searchable
|
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 services.logger import root_logger as logger
|
||||||
from settings import DB_URL
|
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')
|
@event.listens_for(Engine, 'after_cursor_execute')
|
||||||
def after_cursor_execute(conn, cursor, statement, parameters, context, executemany):
|
def after_cursor_execute(conn, cursor, statement, parameters, context, executemany):
|
||||||
if not isinstance(statement, Executable):
|
compiled_statement = context.compiled.string
|
||||||
return
|
compiled_parameters = context.compiled.params
|
||||||
|
if compiled_statement:
|
||||||
|
elapsed = time.time() - conn.query_start_time
|
||||||
|
query = compiled_statement % compiled_parameters
|
||||||
|
|
||||||
if isinstance(context, PGExecutionContext):
|
if elapsed > 1 and conn.last_statement != query:
|
||||||
compiled_statement = context.compiled.string
|
conn.last_statement = query
|
||||||
compiled_parameters = context.compiled.params
|
logger.debug(f"\n{query}\n{'*' * math.floor(elapsed)} {elapsed:.3f} s\n")
|
||||||
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")
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user