o 6 f @spdZddlZddlZddlmZmZddlmZm Z ddl m Z m Z e dZe ddGd d d Zd e_dS) z" psycopg two-phase commit support N) b64encode b64decode)OptionalUnion) dataclassreplacez^(\d+)_([^_]*)_([^_]*)$T)frozenc @s2eZdZUdZeeed<eed<eeed<dZee j ed<dZ eeed<dZ eeed<e d ed dfd d Zd efd dZd efddZded eeedffddZe d ed dfddZe deededeed dfddZd efddZe d efddZe dede j deded df ddZdS)XidzA two-phase commit transaction identifier. The object can also be unpacked as a 3-item tuple (`format_id`, `gtrid`, `bqual`). format_idgtridbqualNpreparedownerdatabasesreturncCs,z||WStytd|dYSw)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 )clsrrE/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg/_tpc.py from_string s   zXid.from_stringcCs|SN)_as_tidselfrrr__str__+sz Xid.__str__cCdS)Nrrrrr__len__.sz Xid.__len__indexcCs|j|j|jf|Sr)r r r )rr rrr __getitem__1szXid.__getitem__cCsVt|}|s tdt|d}t|d}t|d}||||S)Nzbad Xid formatr)_re_xidmatch ValueErrorintgrouprdecode from_parts)rrmr r r rrrr4s zXid._parse_stringcCs|dur3|dur tdd|krdkstdtdt|dkr(tdt|dkr2tdn|dur;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 r rrrr*?s   zXid.from_partscCsT|jdus |jdur |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 r r rencoder))regtridebqualrrrrRs  z Xid._as_tidcCr)Nz