This commit is contained in:
parent
5bec25fc23
commit
aaa39e0a0d
|
@ -1,10 +1,10 @@
|
||||||
import json
|
import json
|
||||||
import math
|
# import math
|
||||||
import time
|
# import time
|
||||||
|
|
||||||
from typing import Any, Callable, Dict, TypeVar
|
from typing import Any, Callable, Dict, TypeVar
|
||||||
|
# from sqlalchemy import exc, event, Engine,
|
||||||
from sqlalchemy import exc, event, Engine, inspect, Column, Integer, create_engine, JSON
|
from sqlalchemy import inspect, Column, Integer, create_engine, JSON
|
||||||
from sqlalchemy.ext.declarative import declarative_base
|
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
|
||||||
|
@ -12,7 +12,7 @@ from sqlalchemy_searchable import make_searchable
|
||||||
|
|
||||||
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
|
||||||
import warnings
|
# import warnings
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
# Подключение к базе данных SQLAlchemy
|
# Подключение к базе данных SQLAlchemy
|
||||||
|
@ -83,21 +83,21 @@ def warning_with_traceback(message: Warning | str, category, filename: str, line
|
||||||
|
|
||||||
|
|
||||||
# Установка функции вывода трейсбека для предупреждений SQLAlchemy
|
# Установка функции вывода трейсбека для предупреждений SQLAlchemy
|
||||||
warnings.showwarning = warning_with_traceback
|
# warnings.showwarning = warning_with_traceback
|
||||||
warnings.simplefilter('always', exc.SAWarning)
|
# warnings.simplefilter('always', exc.SAWarning)
|
||||||
|
|
||||||
|
|
||||||
# Перехватчики для журнала запросов SQLAlchemy
|
# Перехватчики для журнала запросов SQLAlchemy
|
||||||
@event.listens_for(Engine, 'before_cursor_execute')
|
# @event.listens_for(Engine, 'before_cursor_execute')
|
||||||
def before_cursor_execute(conn, cursor, statement, parameters, context, executemany):
|
# def before_cursor_execute(conn, cursor, statement, parameters, context, executemany):
|
||||||
conn.query_start_time = time.time()
|
# conn.query_start_time = time.time()
|
||||||
|
|
||||||
|
|
||||||
# noinspection PyUnusedLocal
|
# noinspection PyUnusedLocal
|
||||||
@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 hasattr(conn, '_query_start_time'):
|
# if hasattr(conn, '_query_start_time'):
|
||||||
elapsed = time.time() - conn.query_start_time
|
# elapsed = time.time() - conn.query_start_time
|
||||||
conn.query_start_time = None
|
# conn.query_start_time = None
|
||||||
query = f'{statement}'.replace('\n', ' ')
|
# query = f'{statement}'.replace('\n', ' ')
|
||||||
logger.debug(f"\n{query}\n{'*' * math.floor(elapsed)} {elapsed:.3f} s\n")
|
# logger.debug(f"\n{query}\n{'*' * math.floor(elapsed)} {elapsed:.3f} s\n")
|
||||||
|
|
Loading…
Reference in New Issue
Block a user