#jLdZddlmZddlmZmZmZmZdZdezZ d d Z dd Z d S)a(Database URL normalization. Hindsight accepts SQLAlchemy-style URLs like ``postgresql+asyncpg://...?ssl=require`` for its async engine, but the same string cannot be handed directly to synchronous SQLAlchemy (psycopg2) or to :func:`asyncpg.create_pool`, which both expect a libpq-compatible URL (``postgresql://...?sslmode=require``). :func:`to_libpq_url` performs that translation. It is idempotent and safe to apply to URLs that are already libpq-compatible, to the ``pg0`` embedded-PG marker, or to any non-PostgreSQL string (returned unchanged). ) annotations) parse_qsl urlencodeurlsplit urlunsplit)zpostgresql+asyncpgzpostgres+asyncpg) postgresqlpostgresurlstrreturnboolc`|rd|vrdSt|jdS)zPTrue if ``url`` is an Oracle SQLAlchemy URL (``oracle`` or ``oracle+oracledb``).://Foracle)rscheme startswith)r s \/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hindsight_api/db_url.py is_oracle_urlrs6 %s""u C==  * *8 4 44c(|rd|vr|St|}|jtvr|Sd}dt|jdD}t |}||jkr ||jkr|St ||j|j||j fS)uNormalize a PostgreSQL URL for libpq-style consumers. Accepts a SQLAlchemy URL (``postgresql+asyncpg://...``) or a plain libpq URL and returns a form suitable for: - :func:`sqlalchemy.create_engine` (sync / psycopg2) - :func:`asyncpg.create_pool` Transformations: - ``postgresql+asyncpg`` / ``postgres+asyncpg`` / ``postgres`` → ``postgresql`` - Query param ``ssl=`` → ``sslmode=`` (SQLAlchemy's asyncpg dialect uses ``ssl=``; libpq uses ``sslmode=``) Any non-PostgreSQL input (e.g. the ``pg0`` embedded-PG marker, a sqlite URL, an empty string) is returned unchanged. Already-normalized URLs are returned unchanged. rrc0g|]\}}|dkrd|fn||fS)sslsslmode).0kvs r z to_libpq_url..8s=59Q!u**A1a&rT)keep_blank_values) rr_POSTGRES_SCHEMESrqueryrrnetlocpathfragment)r parts new_schemenew_query_pairs new_querys r to_libpq_urlr)s& %s"" SMME |,,, J=Fu{fj=k=k=kO/**IU\!!i5;&>&> z5<YW X XXrN)r r r r )r r r r ) __doc__ __future__r urllib.parserrrr_ASYNCPG_SCHEMESr rr)rrrr.s  #"""""CCCCCCCCCCCC=.1AA5555$Y$Y$Y$Y$Y$Yr