o 6 fx@sdZddlmZddlZddlmZddlmZddlmZ ddlm Z dd lm Z dd lm Z e j Z e jZe jZd Zd#d$ddZd#d%ddZd&ddZedZedZd'ddZd(d!d"ZdS))z* Functions to manipulate conninfo strings ) annotationsN)Any)pq)errors)_conninfo_utils)_conninfo_attempts)_conninfo_attempts_asyncconninfostrkwargsrreturncKsp|s|sdS|st|t|Sdd|D}|r&t|}|||}ddd|D}t||S)a Merge a string and keyword params into a single conninfo string. :param conninfo: A `connection string`__ as accepted by PostgreSQL. :param kwargs: Parameters overriding the ones specified in `!conninfo`. :return: A connection string valid for PostgreSQL, with the `!kwargs` parameters merged. Raise `~psycopg.ProgrammingError` if the input doesn't make a valid conninfo string. .. __: https://www.postgresql.org/docs/current/libpq-connect.html #LIBPQ-CONNSTRING r cSsi|] \}}|dur||qSN.0kvrrI/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/conninfo.py 8sz!make_conninfo.. css*|]\}}|dtt|VqdS)=N) _param_escaper rrrr ?s(z make_conninfo..)_parse_conninfor itemsconninfo_to_dictupdatejoin)r rtmprrr make_conninfos r"ConnDictcKs<t|}dd|D}|D] \}}|dur|||<q|S)a Convert the `!conninfo` string into a dictionary of parameters. :param conninfo: A `connection string`__ as accepted by PostgreSQL. :param kwargs: Parameters overriding the ones specified in `!conninfo`. :return: Dictionary with the parameters parsed from `!conninfo` and `!kwargs`. Raise `~psycopg.ProgrammingError` if `!conninfo` is not a a valid connection string. .. __: https://www.postgresql.org/docs/current/libpq-connect.html #LIBPQ-CONNSTRING cSs(i|]}|jdur|j|jqSr)valkeyworddecode)roptrrrrWs(z$conninfo_to_dict..N)rr)r roptsrvrrrrrrGsrlist[pq.ConninfoOption]c Cs>z tj|WStjy}ztt|dd}~ww)z Verify that `!conninfo` is a valid connection string. Raise ProgrammingError if the string is not valid. Return the result of pq.Conninfo.parse() on success. N)rConninfoparseencodeeOperationalErrorProgrammingErrorr )r exrrrr^s rz([\\'])z\sscCs.|sdStd|}t|rd|d}|S)z9 Apply the escaping rule required by PQconnectdb z''z\\\1') re_escapesubre_spacesearch)r2rrrrps    rparamsintcCsdt|d}|dur t}zt|}Wnty!td|w|dkr*t}|S|dkr0d}|S)zG Return the timeout in seconds from the connection parameters. connect_timeoutNzbad value for connect_timeout: r)r get_param_DEFAULT_CONNECT_TIMEOUTr9 ValueErrorr.r0)r8valuetimeoutrrrtimeout_from_conninfo~s   rA)r )r r rrrr )r r rrrr#)r r rr*)r2r rr )r8r#rr9)__doc__ __future__rretypingrr rrr.rrr r#conninfo_attemptsconninfo_attempts_asyncr=r"rrcompiler4r6rrArrrrs(         )