o DDiw@sdZddlmZddlZddlmZmZmZddlmZ ddlm Z ddl m Z m Z ejZejZd Zd!d"ddZd!d#ddZd$ddZedZedZd%ddZd&dd ZdS)'z* Functions to manipulate conninfo strings ) annotationsN)_conninfo_attempts_conninfo_attempts_async_conninfo_utils)errors)pq)ConnDict ConnParamconninfostrkwargsr returncKsp|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/var/www/Datamplify/venv/lib/python3.10/site-packages/psycopg/conninfo.py 4sz!make_conninfo.. css*|]\}}|dtt|VqdS)=N) _param_escaperrrrr ;s(z make_conninfo..)_parse_conninforitemsconninfo_to_dictupdatejoin)r rtmprrr make_conninfos r#r cKs<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)roptrrrrSs"z$conninfo_to_dict..N)rr)r roptsrvrrrrrrCsrlist[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.|sdSttd|}rd|d}|S)z9 Apply the escaping rule required by PQconnectdb z''z\\\1')re_spacesearch re_escapesub)r2rrrrns  rparamsintcCsjt|d}|dur t}ztt|}Wnty$td|dw|dkr-t}|S|dkr3d}|S)zG Return the timeout in seconds from the connection parameters. connect_timeoutNzbad value for connect_timeout: r)r get_param_DEFAULT_CONNECT_TIMEOUTr9float ValueErrorr.r0)r8valuetimeoutrrrtimeout_from_conninfo{s  rB)r )r rrr rr)r rrr rr )r rrr*)r2rrr)r8r rr9)__doc__ __future__rrer rrrrr.rabcr r conninfo_attemptsconninfo_attempts_asyncr=r#rrcompiler6r4rrBrrrrs"     )