*Jf dZddlZddlZddlmZmZddlmZm Z ddl m Z m Z ej dZe dGd d Zd e_dS) z" psycopg two-phase commit support N) b64encode b64decode)OptionalUnion) dataclassreplacez^(\d+)_([^_]*)_([^_]*)$T)frozenc eZdZUdZeeed<eed<eeed<dZee j ed<dZ eeed<dZ eeed<e d ed dfd Zd efd Zd efd Zded eeedffdZe d ed dfdZe deededeed dfdZd efdZe d efdZe dede j deded df dZdS)XidzA two-phase commit transaction identifier. The object can also be unpacked as a 3-item tuple (`format_id`, `gtrid`, `bqual`). format_idgtridbqualNpreparedownerdatabasesreturncp ||S#t$rtd|dcYSwxYw)zTry to parse an XA triple from the string. This may fail for several reasons. In such case return an unparsed Xid. N) _parse_string Exceptionr )clsrs O/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/_tpc.py from_stringzXid.from_string sN  &$$Q'' ' & & &tQ%% % % % &s 55c*|SN)_as_tidselfs r__str__z Xid.__str__+s||~~cdS)Nrs r__len__z Xid.__len__.sqr indexc6|j|j|jf|Sr)r r r)rr%s r __getitem__zXid.__getitem__1s DJ7>>r ct|}|stdt|d}t |d}t |d}||||S)Nzbad Xid formatr")_re_xidmatch ValueErrorintgrouprdecode from_parts)rrmr r rs rrzXid._parse_string4s MM!   /-.. . OO !''!**%%,,..!''!**%%,,..~~i666r c2|t|tdd|cxkrdksntdt|dkrtdt|dkrtdn|tdt|||S) Nz,if format_id is specified, bqual must be toorlz/format_id must be a non-negative 32-bit integer@z&bqual must be not longer than 64 charsz>rid must be not longer than 64 charsz,if format_id is None, bqual must be None too) TypeErrorr-lenr )rr r rs rr1zXid.from_parts?s  } NOOO ....J.... !RSSS5zzB !IJJJ5zzB !IJJJ]JKK K9eU+++r c*|j|j|jSt|j}t|j}|jd|d|S)a Return the PostgreSQL transaction_id for this XA xid. PostgreSQL wants just a string, while the DBAPI supports the XA standard and thus a triple. We use the same conversion algorithm implemented by JDBC in order to allow some form of interoperation. see also: the pgjdbc implementation http://cvs.pgfoundry.org/cgi-bin/cvsweb.cgi/jdbc/pgjdbc/org/ postgresql/xa/RecoveredXid.java?rev=1.2 N_)r rr rencoder0)regtridebquals rrz Xid._as_tidRs > !TZ%7: 4:,,..//66884:,,..//6688.44644F444r cdS)Nz