U ڲg)@sUdZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZerndd lmZe jZd d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2(ZiZd3ed4<ed5d6eDed7d6eDd8d9eDZd:d;d<d=d>Zd;d;d?d@dAZed;dBdCdDdEZedBd;dCdFdGZdNd;d;d;dIdJdKZedLejejej dMZ!dS)Oz3 Mappings between PostgreSQL and Python encodings. ) annotationsN)Any TYPE_CHECKING) ConnStatus)NotSupportedError)cache)BaseConnectionbig5gb2312 euc_jis_2004euc_jpeuc_krgb18030gbkz iso8859-5z iso8859-6z iso8859-7z iso8859-8johabzkoi8-rzkoi8-uz iso8859-1z iso8859-16z iso8859-2z iso8859-3z iso8859-4z iso8859-9z iso8859-10z iso8859-13z iso8859-14z iso8859-15shift_jis_2004 shift_jisasciicp949utf-8cp1250cp1251cp1252cp1253cp1254cp1255cp1256cp1257cp1258cp866cp874)(ZBIG5ZEUC_CNZ EUC_JIS_2004ZEUC_JPZEUC_KRZGB18030ZGBKZ ISO_8859_5Z ISO_8859_6Z ISO_8859_7Z ISO_8859_8ZJOHABZKOI8RZKOI8UZLATIN1ZLATIN10ZLATIN2ZLATIN3ZLATIN4ZLATIN5ZLATIN6ZLATIN7ZLATIN8ZLATIN9ZSHIFT_JIS_2004SJISZ SQL_ASCIIZUHCUTF8ZWIN1250ZWIN1251ZWIN1252ZWIN1253ZWIN1254ZWIN1255ZWIN1256ZWIN1257ZWIN1258ZWIN866ZWIN874zdict[bytes, str] py_codecsccs|]\}}||fVqdS)Nencode.0kvr+6/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/_encodings.py Hsr-ccs.|]&\}}d|kr|dd|fVqdS)_N)replacer&r'r+r+r,r-KscCsi|]\}}||qSr+r%r'r+r+r, Osr1zBaseConnection[Any] | Nonestr)connreturncCs|r |jjSdS)z Return the Python encoding name of a psycopg connection. Default to utf8 if the connection has no encoding info. r)Zpgconn _encoding)r3r+r+r, conn_encodingRsr6)conninfor4cCsNddlm}||}|d}|rJztt|WStk rHYnXdS)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_dictZclient_encodingr)r7r8getpg2pyencr2r&r)r7r8paramsZpgencr+r+r,conninfo_encoding[s  r<bytes)namer4cCstt|jS)zzConvert a Python encoding name to PostgreSQL encoding name. Raise LookupError if the Python encoding is unknown. ) pg_codecscodecslookupr>)r>r+r+r,py2pgencosrBcCsRzt|ddddWStk rL|dd}td|YnXdS)zConvert a PostgreSQL encoding name to Python encoding name. Raise NotSupportedError if the PostgreSQL encoding is not supported by Python. -_utf8r0zcodec not available in Python: N)r$r0upperKeyErrordecoder)r>Zsnamer+r+r,r:xs  r:f)sprefixr4cCsH|s0|ddkr||}|s0td|}|ddkrD||}|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 '_'. rZ 1234567890r.) isidentifier _re_cleansub)rKrLr+r+r,_as_python_identifiers   rPz[^z_])rJ)"__doc__ __future__rrestringr@typingrrZ pq._enumsrerrorsr_compatrZ_connection_baser OKZ _py_codecsr$__annotations__updateitemsr?r6r<rBr:rPcompileascii_lowercaseascii_uppercasedigitsrNr+r+r+r,s     /