FM / integrations

/opt/imunify360/venv/lib/python3.11/site-packages/sentry_sdk/integrations/

/opt/imunify360/venv/lib/python3.11/site-packages/sentry_sdk/integrations

Name	Size	Mode	Actions
django/	-	0755	rm
spark/	-	0755	rm
__pycache__/	-	0755	rm
aiohttp.py	7960	0644	edit dl rm
argv.py	945	0644	edit dl rm
asgi.py	8368	0644	edit dl rm
atexit.py	1837	0644	edit dl rm
aws_lambda.py	12941	0644	edit dl rm
beam.py	5662	0644	edit dl rm
boto3.py	3727	0644	edit dl rm
bottle.py	6187	0644	edit dl rm
celery.py	9119	0644	edit dl rm
chalice.py	4573	0644	edit dl rm
dedupe.py	1166	0644	edit dl rm
excepthook.py	2190	0644	edit dl rm
executing.py	2023	0644	edit dl rm
falcon.py	6796	0644	edit dl rm
flask.py	7428	0644	edit dl rm
gcp.py	7382	0644	edit dl rm
gnu_backtrace.py	2912	0644	edit dl rm
logging.py	7573	0644	edit dl rm
modules.py	1393	0644	edit dl rm
pure_eval.py	4520	0644	edit dl rm
pyramid.py	7072	0644	edit dl rm
redis.py	3075	0644	edit dl rm
rq.py	4791	0644	edit dl rm
sanic.py	7709	0644	edit dl rm
serverless.py	1971	0644	edit dl rm
sqlalchemy.py	2939	0644	edit dl rm
stdlib.py	7348	0644	edit dl rm
threading.py	2856	0644	edit dl rm
tornado.py	7001	0644	edit dl rm
trytond.py	1728	0644	edit dl rm
wsgi.py	10374	0644	edit dl rm
_wsgi_common.py	4763	0644	edit dl rm
__init__.py	6509	0644	edit dl rm

Edit: /opt/imunify360/venv/lib/python3.11/site-packages/sentry_sdk/integrations/sqlalchemy.py (2939B)

from __future__ import absolute_import

from sentry_sdk._types import MYPY
from sentry_sdk.hub import Hub
from sentry_sdk.integrations import Integration, DidNotEnable
from sentry_sdk.tracing import record_sql_queries

try:
    from sqlalchemy.engine import Engine  # type: ignore
    from sqlalchemy.event import listen  # type: ignore
    from sqlalchemy import __version__ as SQLALCHEMY_VERSION  # type: ignore
except ImportError:
    raise DidNotEnable("SQLAlchemy not installed.")

if MYPY:
    from typing import Any
    from typing import ContextManager
    from typing import Optional

from sentry_sdk.tracing import Span

class SqlalchemyIntegration(Integration):
    identifier = "sqlalchemy"

@staticmethod
    def setup_once():
        # type: () -> None

try:
            version = tuple(map(int, SQLALCHEMY_VERSION.split("b")[0].split(".")))
        except (TypeError, ValueError):
            raise DidNotEnable(
                "Unparsable SQLAlchemy version: {}".format(SQLALCHEMY_VERSION)
            )

if version < (1, 2):
            raise DidNotEnable("SQLAlchemy 1.2 or newer required.")

listen(Engine, "before_cursor_execute", _before_cursor_execute)
        listen(Engine, "after_cursor_execute", _after_cursor_execute)
        listen(Engine, "handle_error", _handle_error)

def _before_cursor_execute(
    conn, cursor, statement, parameters, context, executemany, *args
):
    # type: (Any, Any, Any, Any, Any, bool, *Any) -> None
    hub = Hub.current
    if hub.get_integration(SqlalchemyIntegration) is None:
        return

ctx_mgr = record_sql_queries(
        hub,
        cursor,
        statement,
        parameters,
        paramstyle=context and context.dialect and context.dialect.paramstyle or None,
        executemany=executemany,
    )
    conn._sentry_sql_span_manager = ctx_mgr

span = ctx_mgr.__enter__()

if span is not None:
        conn._sentry_sql_span = span

def _after_cursor_execute(conn, cursor, statement, *args):
    # type: (Any, Any, Any, *Any) -> None
    ctx_mgr = getattr(
        conn, "_sentry_sql_span_manager", None
    )  # type: ContextManager[Any]

if ctx_mgr is not None:
        conn._sentry_sql_span_manager = None
        ctx_mgr.__exit__(None, None, None)

def _handle_error(context, *args):
    # type: (Any, *Any) -> None
    conn = context.connection
    span = getattr(conn, "_sentry_sql_span", None)  # type: Optional[Span]

if span is not None:
        span.set_status("internal_error")

# _after_cursor_execute does not get called for crashing SQL stmts. Judging
    # from SQLAlchemy codebase it does seem like any error coming into this
    # handler is going to be fatal.
    ctx_mgr = getattr(
        conn, "_sentry_sql_span_manager", None
    )  # type: ContextManager[Any]

if ctx_mgr is not None:
        conn._sentry_sql_span_manager = None
        ctx_mgr.__exit__(None, None, None)

from __future__ import absolute_import from sentry_sdk._types import MYPY from sentry_sdk.hub import Hub from sentry_sdk.integrations import Integration, DidNotEnable from sentry_sdk.tracing import record_sql_queries try: from sqlalchemy.engine import Engine # type: ignore from sqlalchemy.event import listen # type: ignore from sqlalchemy import __version__ as SQLALCHEMY_VERSION # type: ignore except ImportError: raise DidNotEnable("SQLAlchemy not installed.") if MYPY: from typing import Any from typing import ContextManager from typing import Optional from sentry_sdk.tracing import Span class SqlalchemyIntegration(Integration): identifier = "sqlalchemy" @staticmethod def setup_once(): # type: () -> None try: version = tuple(map(int, SQLALCHEMY_VERSION.split("b")[0].split("."))) except (TypeError, ValueError): raise DidNotEnable( "Unparsable SQLAlchemy version: {}".format(SQLALCHEMY_VERSION) ) if version < (1, 2): raise DidNotEnable("SQLAlchemy 1.2 or newer required.") listen(Engine, "before_cursor_execute", _before_cursor_execute) listen(Engine, "after_cursor_execute", _after_cursor_execute) listen(Engine, "handle_error", _handle_error) def _before_cursor_execute( conn, cursor, statement, parameters, context, executemany, *args ): # type: (Any, Any, Any, Any, Any, bool, *Any) -> None hub = Hub.current if hub.get_integration(SqlalchemyIntegration) is None: return ctx_mgr = record_sql_queries( hub, cursor, statement, parameters, paramstyle=context and context.dialect and context.dialect.paramstyle or None, executemany=executemany, ) conn._sentry_sql_span_manager = ctx_mgr span = ctx_mgr.__enter__() if span is not None: conn._sentry_sql_span = span def _after_cursor_execute(conn, cursor, statement, *args): # type: (Any, Any, Any, *Any) -> None ctx_mgr = getattr( conn, "_sentry_sql_span_manager", None ) # type: ContextManager[Any] if ctx_mgr is not None: conn._sentry_sql_span_manager = None ctx_mgr.__exit__(None, None, None) def _handle_error(context, *args): # type: (Any, *Any) -> None conn = context.connection span = getattr(conn, "_sentry_sql_span", None) # type: Optional[Span] if span is not None: span.set_status("internal_error") # _after_cursor_execute does not get called for crashing SQL stmts. Judging # from SQLAlchemy codebase it does seem like any error coming into this # handler is going to be fatal. ctx_mgr = getattr( conn, "_sentry_sql_span_manager", None ) # type: ContextManager[Any] if ctx_mgr is not None: conn._sentry_sql_span_manager = None ctx_mgr.__exit__(None, None, None)