o 6 f@sVUdZddlZddlZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZer;ddlmZdd lmZe jZid d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+id,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTZiZeeefedU<edVdWeDedXdWeDdYdZeDZd[d\d]efd^d_Zd`dad]efdbdcZdded]efdedfZedged]efdhdiZedged]efdjdkZ dsdmedned]efdodpZ!e"dqej#ej$ej%drZ&dS)tz3 Mappings between PostgreSQL and Python encodings. N)AnyDictOptional TYPE_CHECKING) ConnStatus)NotSupportedError)cache)PGconn)BaseConnectionBIG5big5EUC_CNgb2312 EUC_JIS_2004 euc_jis_2004EUC_JPeuc_jpEUC_KReuc_krGB18030gb18030GBKgbk ISO_8859_5z iso8859-5 ISO_8859_6z iso8859-6 ISO_8859_7z iso8859-7 ISO_8859_8z iso8859-8JOHABjohabKOI8Rzkoi8-rKOI8Uzkoi8-uLATIN1z iso8859-1LATIN10z iso8859-16LATIN2z iso8859-2LATIN3z iso8859-3LATIN4z iso8859-4LATIN5z iso8859-9LATIN6z iso8859-10LATIN7z iso8859-13LATIN8z iso8859-14LATIN9z iso8859-15SHIFT_JIS_2004shift_jis_2004SJIS shift_jis SQL_ASCIIasciiUHCcp949UTF8utf-8WIN1250cp1250WIN1251cp1251WIN1252cp1252WIN1253cp1253WIN1254cp1254cp1255cp1256cp1257cp1258cp866cp874)WIN1255WIN1256WIN1257WIN1258WIN866WIN874 py_codecsccs |] \}}||fVqdS)Nencode.0kvrSK/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/_encodings.py GsrUccs0|]\}}d|vr|dd|fVqdS)_N)replacerNrOrSrSrTrUJs&cCsi|] \}}||qSrSrMrOrSrSrT NsrYconnzOptional[BaseConnection[Any]]returncCs&|r|jrdS|jdpd}t|S)z Return the Python encoding name of a psycopg connection. Default to utf8 if the connection has no encoding info. r5client_encodingUTF8)closedpgconnparameter_statuspg2pyenc)rZpgencrSrSrT conn_encodingQs rcr_r cCs$|jtkrdS|dp d}t|S)z} Return the Python encoding name of a libpq connection. Default to utf8 if the connection has no encoding info. r5r\r])statusOKr`ra)r_rbrSrSrTpgconn_encoding^s rfconninfocCsJddlm}||}|d}|r#zt|WSty"YdSwdS)z Return the Python encoding name passed in a conninfo string. Default to utf8. Because the input is likely to come from the user and not normalised by the server, be somewhat lenient (non-case-sensitive lookup, ignore noise chars). r)conninfo_to_dictclient_encodingr5)rgrhgetrarNr)rgrhparamsrbrSrSrTconninfo_encodingks   rlnamecCstt|jS)zzConvert a Python encoding name to PostgreSQL encoding name. Raise LookupError if the Python encoding is unknown. ) pg_codecscodecslookuprm)rmrSrSrTpy2pgencsrqcCsHzt|ddddWSty#|dd}td|w)zConvert a PostgreSQL encoding name to Python encoding name. Raise NotSupportedError if the PostgreSQL encoding is not supported by Python. -_utf8rXzcodec not available in Python: )rLrXupperKeyErrordecoder)rmsnamerSrSrTras   rafsprefixcCsH|s|ddvr||}|std|}|ddkr"||}|S)z Reduce a string to a valid Python identifier. Replace all non-valid chars with '_' and prefix the value with `!prefix` if the first letter is an '_'. r 1234567890rV) isidentifier _re_cleansub)r{r|rSrSrT_as_python_identifiers   rz[^z_])rz)'__doc__restringrotypingrrrr pq._enumsrerrorsr_compatr pq.abcr connectionr re _py_codecsrLbytesstr__annotations__updateitemsrnrcrfrlrqrarcompileascii_lowercaseascii_uppercasedigitsrrSrSrSrTs           !"#$%&'/