U ڲg @sldZddlmZddlZddlZddlmZmZddl m Z m Z e dZ e ddGd d d Zd e_dS) z" psycopg two-phase commit support ) annotationsN) b64encode b64decode) dataclassreplacez^(\d+)_([^_]*)_([^_]*)$T)frozenc@seZdZUdZded<ded<ded<dZd ed <dZded <dZded <eddd ddZ ddddZ ddddZ dddddZ eddd ddZ edddddddZdddd Zeddd!d"Zedd#dddd$d%d&ZdS)'XidzA two-phase commit transaction identifier. The object can also be unpacked as a 3-item tuple (`format_id`, `gtrid`, `bqual`). z int | None format_idstrgtridz str | NonebqualNzdt.datetime | Nonepreparedownerdatabase)sreturncCs2z ||WStk r,td|dYSXdS)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)clsrr0/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/_tpc.py from_string!s zXid.from_string)rcCs|SN)_as_tidselfrrr__str__,sz Xid.__str__intcCsdS)Nrrrrr__len__/sz Xid.__len__zint | str | None)indexrcCs|j|j|jf|Sr)r r r )rr rrr __getitem__2szXid.__getitem__cCsVt|}|stdt|d}t|d}t|d}||||S)Nzbad Xid formatr)_re_xidmatch ValueErrorrgrouprdecode from_parts)rrmr r r rrrr5s zXid._parse_string)r r r rcCs||dk r`|dkrtdd|kr,dks6ntdt|dkrJtdt|dkrptdn|dkrptdt|||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|jdks|jdkr|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()rZegtridZebqualrrrrQs z Xid._as_tidcCsdS)Nz