aDriv4 website!

Linux premium231.web-hosting.com 4.18.0-553.45.1.lve.el8.x86_64 #1 SMP Wed Mar 26 12:08:09 UTC 2025 x86_64
LiteSpeed
66.29.146.61

asyncpg.py

from __future__ import annotations
import contextlib
from typing import Any, TypeVar, Callable, Awaitable, Iterator

import sentry_sdk
from sentry_sdk.consts import OP, SPANDATA
from sentry_sdk.integrations import _check_minimum_version, Integration, DidNotEnable
from sentry_sdk.tracing import Span
from sentry_sdk.tracing_utils import add_query_source, record_sql_queries
from sentry_sdk.utils import (
    ensure_integration_enabled,
    parse_version,
    capture_internal_exceptions,
)

try:
    import asyncpg  # type: ignore[import-not-found]
    from asyncpg.cursor import BaseCursor  # type: ignore

except ImportError:
    raise DidNotEnable("asyncpg not installed.")

class AsyncPGIntegration(Integration):
    identifier = "asyncpg"
    origin = f"auto.db.{identifier}"
    _record_params = False

def __init__(self, *, record_params: bool = False):
        AsyncPGIntegration._record_params = record_params

@staticmethod
    def setup_once() -> None:
        # asyncpg.__version__ is a string containing the semantic version in the form of "<major>.<minor>.<patch>"
        asyncpg_version = parse_version(asyncpg.__version__)
        _check_minimum_version(AsyncPGIntegration, asyncpg_version)

asyncpg.Connection.execute = _wrap_execute(
            asyncpg.Connection.execute,
        )

asyncpg.Connection._execute = _wrap_connection_method(
            asyncpg.Connection._execute
        )
        asyncpg.Connection._executemany = _wrap_connection_method(
            asyncpg.Connection._executemany, executemany=True
        )
        asyncpg.Connection.cursor = _wrap_cursor_creation(asyncpg.Connection.cursor)
        asyncpg.Connection.prepare = _wrap_connection_method(asyncpg.Connection.prepare)
        asyncpg.connect_utils._connect_addr = _wrap_connect_addr(
            asyncpg.connect_utils._connect_addr
        )

T = TypeVar("T")

def _wrap_execute(f: Callable[..., Awaitable[T]]) -> Callable[..., Awaitable[T]]:
    async def _inner(*args: Any, **kwargs: Any) -> T:
        if sentry_sdk.get_client().get_integration(AsyncPGIntegration) is None:
            return await f(*args, **kwargs)

# Avoid recording calls to _execute twice.
        # Calls to Connection.execute with args also call
        # Connection._execute, which is recorded separately
        # args[0] = the connection object, args[1] is the query
        if len(args) > 2:
            return await f(*args, **kwargs)

query = args[1]
        with record_sql_queries(
            cursor=None,
            query=query,
            params_list=None,
            paramstyle=None,
            executemany=False,
            span_origin=AsyncPGIntegration.origin,
        ) as span:
            res = await f(*args, **kwargs)

with capture_internal_exceptions():
            add_query_source(span)

return res

return _inner

SubCursor = TypeVar("SubCursor", bound=BaseCursor)

@contextlib.contextmanager
def _record(
    cursor: SubCursor | None,
    query: str,
    params_list: tuple[Any, ...] | None,
    *,
    executemany: bool = False,
) -> Iterator[Span]:
    integration = sentry_sdk.get_client().get_integration(AsyncPGIntegration)
    if integration is not None and not integration._record_params:
        params_list = None

param_style = "pyformat" if params_list else None

with record_sql_queries(
        cursor=cursor,
        query=query,
        params_list=params_list,
        paramstyle=param_style,
        executemany=executemany,
        record_cursor_repr=cursor is not None,
        span_origin=AsyncPGIntegration.origin,
    ) as span:
        yield span

def _wrap_connection_method(
    f: Callable[..., Awaitable[T]], *, executemany: bool = False
) -> Callable[..., Awaitable[T]]:
    async def _inner(*args: Any, **kwargs: Any) -> T:
        if sentry_sdk.get_client().get_integration(AsyncPGIntegration) is None:
            return await f(*args, **kwargs)
        query = args[1]
        params_list = args[2] if len(args) > 2 else None
        with _record(None, query, params_list, executemany=executemany) as span:
            _set_db_data(span, args[0])
            res = await f(*args, **kwargs)

return res

return _inner

def _wrap_cursor_creation(f: Callable[..., T]) -> Callable[..., T]:
    @ensure_integration_enabled(AsyncPGIntegration, f)
    def _inner(*args: Any, **kwargs: Any) -> T:  # noqa: N807
        query = args[1]
        params_list = args[2] if len(args) > 2 else None

with _record(
            None,
            query,
            params_list,
            executemany=False,
        ) as span:
            _set_db_data(span, args[0])
            res = f(*args, **kwargs)
            span.set_data("db.cursor", res)

return res

return _inner

def _wrap_connect_addr(f: Callable[..., Awaitable[T]]) -> Callable[..., Awaitable[T]]:
    async def _inner(*args: Any, **kwargs: Any) -> T:
        if sentry_sdk.get_client().get_integration(AsyncPGIntegration) is None:
            return await f(*args, **kwargs)

user = kwargs["params"].user
        database = kwargs["params"].database

with sentry_sdk.start_span(
            op=OP.DB,
            name="connect",
            origin=AsyncPGIntegration.origin,
        ) as span:
            span.set_data(SPANDATA.DB_SYSTEM, "postgresql")
            addr = kwargs.get("addr")
            if addr:
                try:
                    span.set_data(SPANDATA.SERVER_ADDRESS, addr[0])
                    span.set_data(SPANDATA.SERVER_PORT, addr[1])
                except IndexError:
                    pass
            span.set_data(SPANDATA.DB_NAME, database)
            span.set_data(SPANDATA.DB_USER, user)

with capture_internal_exceptions():
                sentry_sdk.add_breadcrumb(
                    message="connect", category="query", data=span._data
                )
            res = await f(*args, **kwargs)

return res

return _inner

def _set_db_data(span: Span, conn: Any) -> None:
    span.set_data(SPANDATA.DB_SYSTEM, "postgresql")

addr = conn._addr
    if addr:
        try:
            span.set_data(SPANDATA.SERVER_ADDRESS, addr[0])
            span.set_data(SPANDATA.SERVER_PORT, addr[1])
        except IndexError:
            pass

database = conn._params.database
    if database:
        span.set_data(SPANDATA.DB_NAME, database)

user = conn._params.user
    if user:
        span.set_data(SPANDATA.DB_USER, user)

Name	Type	Size	Owner/Group	Permission	Last Modified	Actions
.	dir	-	root/root	0755	2025-12-03 09:30:45
..	dir	-	root/root	0755	2025-12-03 09:30:45
__pycache__	dir	-	root/root	0755	2025-12-03 09:30:45
celery	dir	-	root/root	0755	2025-12-03 09:30:45
django	dir	-	root/root	0755	2025-12-03 09:30:45
google_genai	dir	-	root/root	0755	2025-12-03 09:30:45
grpc	dir	-	root/root	0755	2025-12-03 09:30:45
openai_agents	dir	-	root/root	0755	2025-12-03 09:30:45
opentelemetry	dir	-	root/root	0755	2025-12-03 09:30:45
pydantic_ai	dir	-	root/root	0755	2025-12-03 09:30:45
redis	dir	-	root/root	0755	2025-12-03 09:30:45
spark	dir	-	root/root	0755	2025-12-03 09:30:45
__init__.py	text/x-script.python	12.44 KB	root/root	0644	2025-12-03 09:30:45
_asgi_common.py	text/x-script.python	3.11 KB	root/root	0644	2025-12-03 09:30:45
_wsgi_common.py	text/x-script.python	7.38 KB	root/root	0644	2025-12-03 09:30:45
aiohttp.py	text/x-script.python	12.7 KB	root/root	0644	2025-12-03 09:30:45
anthropic.py	text/x-script.python	14.17 KB	root/root	0644	2025-12-03 09:30:45
argv.py	text/x-script.python	911 B	root/root	0644	2025-12-03 09:30:45
ariadne.py	text/x-script.python	5.7 KB	root/root	0644	2025-12-03 09:30:45
arq.py	text/x-script.python	7.7 KB	root/root	0644	2025-12-03 09:30:45
asgi.py	text/plain	12.5 KB	root/root	0644	2025-12-03 09:30:45
asyncio.py	text/x-script.python	4.5 KB	root/root	0644	2025-12-03 09:30:45
asyncpg.py	text/x-script.python	6.37 KB	root/root	0644	2025-12-03 09:30:45
atexit.py	text/x-script.python	1.61 KB	root/root	0644	2025-12-03 09:30:45
aws_lambda.py	text/x-script.python	17.65 KB	root/root	0644	2025-12-03 09:30:45
beam.py	text/x-script.python	5.06 KB	root/root	0644	2025-12-03 09:30:45
boto3.py	text/x-script.python	4.31 KB	root/root	0644	2025-12-03 09:30:45
bottle.py	text/x-script.python	6.46 KB	root/root	0644	2025-12-03 09:30:45
chalice.py	text/x-script.python	4.59 KB	root/root	0644	2025-12-03 09:30:45
clickhouse_driver.py	text/x-script.python	5.96 KB	root/root	0644	2025-12-03 09:30:45
cloud_resource_context.py	text/x-script.python	7.6 KB	root/root	0644	2025-12-03 09:30:45
cohere.py	text/x-script.python	9.18 KB	root/root	0644	2025-12-03 09:30:45
dedupe.py	text/x-script.python	1.93 KB	root/root	0644	2025-12-03 09:30:45
dramatiq.py	text/x-script.python	7.28 KB	root/root	0644	2025-12-03 09:30:45
excepthook.py	text/x-script.python	2.35 KB	root/root	0644	2025-12-03 09:30:45
executing.py	text/x-script.python	1.95 KB	root/root	0644	2025-12-03 09:30:45
falcon.py	text/x-script.python	9.28 KB	root/root	0644	2025-12-03 09:30:45
fastapi.py	text/x-script.python	4.48 KB	root/root	0644	2025-12-03 09:30:45
flask.py	text/x-script.python	8.54 KB	root/root	0644	2025-12-03 09:30:45
gcp.py	text/x-script.python	8.26 KB	root/root	0644	2025-12-03 09:30:45
gnu_backtrace.py	text/x-script.python	2.75 KB	root/root	0644	2025-12-03 09:30:45
gql.py	text/x-script.python	4.69 KB	root/root	0644	2025-12-03 09:30:45
graphene.py	text/x-script.python	4.92 KB	root/root	0644	2025-12-03 09:30:45
httpx.py	text/x-script.python	5.94 KB	root/root	0644	2025-12-03 09:30:45
huey.py	text/x-script.python	5.32 KB	root/root	0644	2025-12-03 09:30:45
huggingface_hub.py	text/x-script.python	14.6 KB	root/root	0644	2025-12-03 09:30:45
langchain.py	text/x-script.python	39.02 KB	root/root	0644	2025-12-03 09:30:45
langgraph.py	text/x-script.python	11.56 KB	root/root	0644	2025-12-03 09:30:45
launchdarkly.py	text/x-script.python	1.89 KB	root/root	0644	2025-12-03 09:30:45
litellm.py	text/x-script.python	9.99 KB	root/root	0644	2025-12-03 09:30:45
litestar.py	text/x-script.python	11.55 KB	root/root	0644	2025-12-03 09:30:45
logging.py	text/x-script.python	13.57 KB	root/root	0644	2025-12-03 09:30:45
loguru.py	text/x-script.python	6.38 KB	root/root	0644	2025-12-03 09:30:45
mcp.py	text/plain	19.58 KB	root/root	0644	2025-12-03 09:30:45
modules.py	text/x-script.python	820 B	root/root	0644	2025-12-03 09:30:45
openai.py	text/x-script.python	24.54 KB	root/root	0644	2025-12-03 09:30:45
openfeature.py	text/x-script.python	1.1 KB	root/root	0644	2025-12-03 09:30:45
otlp.py	text/x-script.python	3.06 KB	root/root	0644	2025-12-03 09:30:45
pure_eval.py	text/x-script.python	4.5 KB	root/root	0644	2025-12-03 09:30:45
pymongo.py	text/x-script.python	6.23 KB	root/root	0644	2025-12-03 09:30:45
pyramid.py	text/x-script.python	7.19 KB	root/root	0644	2025-12-03 09:30:45
quart.py	text/x-script.python	7.24 KB	root/root	0644	2025-12-03 09:30:45
ray.py	text/x-script.python	5.2 KB	root/root	0644	2025-12-03 09:30:45
rq.py	text/x-script.python	5.18 KB	root/root	0644	2025-12-03 09:30:45
rust_tracing.py	text/plain	8.87 KB	root/root	0644	2025-12-03 09:30:45
sanic.py	text/x-script.python	12.66 KB	root/root	0644	2025-12-03 09:30:45
serverless.py	text/x-script.python	1.76 KB	root/root	0644	2025-12-03 09:30:45
socket.py	text/x-script.python	3.09 KB	root/root	0644	2025-12-03 09:30:45
sqlalchemy.py	text/x-script.python	4.24 KB	root/root	0644	2025-12-03 09:30:45
starlette.py	text/x-script.python	25.62 KB	root/root	0644	2025-12-03 09:30:45
starlite.py	text/x-script.python	10.31 KB	root/root	0644	2025-12-03 09:30:45
statsig.py	text/x-script.python	1.2 KB	root/root	0644	2025-12-03 09:30:45
stdlib.py	text/x-script.python	8.76 KB	root/root	0644	2025-12-03 09:30:45
strawberry.py	text/x-script.python	13.82 KB	root/root	0644	2025-12-03 09:30:45
sys_exit.py	text/x-script.python	2.43 KB	root/root	0644	2025-12-03 09:30:45
threading.py	text/x-script.python	6.94 KB	root/root	0644	2025-12-03 09:30:45
tornado.py	text/x-script.python	7.04 KB	root/root	0644	2025-12-03 09:30:45
trytond.py	text/x-script.python	1.61 KB	root/root	0644	2025-12-03 09:30:45
typer.py	text/x-script.python	1.77 KB	root/root	0644	2025-12-03 09:30:45
unleash.py	text/x-script.python	1.03 KB	root/root	0644	2025-12-03 09:30:45
unraisablehook.py	text/x-script.python	1.71 KB	root/root	0644	2025-12-03 09:30:45
wsgi.py	text/x-script.python	10.56 KB	root/root	0644	2025-12-03 09:30:45

For more Private tool come to me!

https://t.me/aDriv4vz