*JfbUdZddlZddlZddlZddlmZmZmZmZddl 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<edVeDedWeDdXeDZdYdZd[efd\Zd]d^d[efd_Zd`ed[efdaZedbed[efdcZedbed[efddZ dkdfedged[efdhZ!ej"diej#ej$ej%djZ&dS)lz3 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_codecsc#HK|]\}}||fVdS)Nencode.0kvs U/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/_encodings.py rVGs2@@TQ!((**a@@@@@@c#xK|]5\}}d|v |dd|fV6dS)_N)replacerPrQs rUrVrVJsR)-AsaxxQYYsB  !$xxxxrWc>i|]\}}||SrOrQs rU r^Ns& : : :tq!Q : : :rWconnzOptional[BaseConnection[Any]]returncn|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. r6client_encodingUTF8)closedpgconnparameter_statuspg2pyenc)r_pgencs rU conn_encodingriQsA 4;w K ( (); < < GE E??rWrer cr|jtkrdS|dpd}t|S)z} Return the Python encoding name of a libpq connection. Default to utf8 if the connection has no encoding info. r6rbrc)statusOKrfrg)rerhs rUpgconn_encodingrm^s; }w  # #$6 7 7 B7E E??rWconninfocddlm}||}|d}|r2 t|S#t $rYnwxYwdS)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_encodingr6)rnrpgetrgrPr )rnrpparamsrhs rUconninfo_encodingrtks+*****  h ' 'F JJ( ) )E  ELLNN++ +     D  7s A AAnamecJttj|jS)zzConvert a Python encoding name to PostgreSQL encoding name. Raise LookupError if the Python encoding is unknown. ) pg_codecscodecslookupru)rus rUpy2pgencrzs V]4((- ..rWc t|ddddS#t$r)|dd}t d|wxYw)zConvert a PostgreSQL encoding name to Python encoding name. Raise NotSupportedError if the PostgreSQL encoding is not supported by Python. -rW_utf8r[zcodec not available in Python: )rMr[upperKeyErrordecoder )rusnames rUrgrgsMdC0088sCCIIKKLL MMM FI.. K% K KLLLMs AA 3A<fsprefixc|s>|ddvr||z}|std|}|ddkr||z}|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 1234567890rY) isidentifier _re_cleansub)rrs rU_as_python_identifierrsn >>  & Q4<   A~~ & c1%%Ats{{ QJ HrWz[^z_])r)'__doc__restringrxtypingrrrr pq._enumsrerrorsr _compatr pq.abcr connectionr rl _py_codecsrMbytesstr__annotations__updateitemsrwrirmrtrzrgrcompileascii_lowercaseascii_uppercasedigitsrr]rWrUrsu  555555555555!!!!!!%%%%%%+******]- F- h-N- h -  h -y- 5-+-+-+-+- W- X- X-  k!-"|#-$ k%--& k'-( k)-* k+-, l--. l/-0 l1-2 l3-6&7-8 K9->?-@ 7A-B GC-DxE-FxG-HxI-JxK-LxM--NY--- ^! 4s  @@Z-=-=-?-?@@@@@@ 1;1A1A1C1C ; :z'7'7'9'9 : : :  7 C     H     (/3/5//// M5 MS M M M M  S #     $ BJJ J!7JJJJ   rW