File: //usr/lib/python3/dist-packages/trac/db/__pycache__/postgres_backend.cpython-310.pyc
o
�k�`A � @ sn d dl mZ d dlZd dlZd dlZd dlmZ d dlmZm Z d dl
T d dl m Z d dl
mZmZmZ d dlmZmZ d d lmZmZ d d
lmZ d d lmZ d d lm Z m Z m Z d d
l m Z z d dl!Z"d dl#Z!d dl!m$Z$m%Z% d dl#m&Z&m'Z'm(Z(m)Z)m*Z* W n
e+y� eddg��w e&e'� e(edd� � e(e,e �dd� � ee"��-de"j.�Z/dZ0e1e"d��sd Z2Z3zDe4e"j5j6d��3Z2ej7dkr�e�8d�9d�e2�:� ej;�Z3e3r�e3�<d�Z0ne�8d �9d�e2�:� ej=�r�ed �Z0W d � n1 s�w Y W n
e>�y Y nw [2[3e�?d �Z@d d iZAd!ZBd.d"d#�ZCd$d%� ZDd&d'� ZEd(d)� ZFG d*d+� d+eG�ZHG d,d-� d-ee�ZIdS )/� )� find_libraryN)�DistributionNotFound)�Popen�PIPE)�*)�Option)�ConnectionBase�IDatabaseConnector�parse_connection_uri)�ConnectionWrapper�IterableCursor)� get_pkginfo�lazy)� close_fds)�Markup)�empty�exception_to_unicode�
to_unicode)�_)� DataError�ProgrammingError)�
register_type�UNICODE�register_adapter�AsIs� QuotedStringz psycopg2>=2.0 or psycopg2-binary�Tracc C s t t| ��S �N)r �str)�markup� r �:/usr/lib/python3/dist-packages/trac/db/postgres_backend.py�<lambda>- s r" c C s t d�S )Nz'')r )r r r r! r" . s �version�
libpq_version�rb�ntz�
\0(
(?:/[^/\0]+)*/?
libpq\.(?:so\.[0-9]+|[0-9]+\.dylib)
)\0
zutf-8� z\0libpq\.dll\0�libpqz([/_%])�int64�bigint)� r' r c s4 dd� � | ||||d�}d� � fdd�|�� D ��S )z*Quote the parameters and assemble the DSN.c S s* t | t�s t| �} d| �dd��dd� S )Nz'%s'�\z\\�'z\')�
isinstancer �replace)�valuer r r! �quoteV s
z assemble_pg_dsn.<locals>.quote)�dbname�user�password�host�port� c 3 s( � | ]\}}|rd |� |�f V qdS )z%s=%sNr )�.0�namer0 �r1 r r! � <genexpr>] s � �z"assemble_pg_dsn.<locals>.<genexpr>)�join�items)�pathr3 r4 r5 r6 �dsnr r: r! �assemble_pg_dsnT s ��r@ c C s d| � dd� S )N�"%s"�"z"")r/ )�
identifierr r r! �_quotea s rD c C s> | r t | d�\}}|dkr||fS t |d�\}}|||fS d S )Ni' �
�d )�divmod)�ver�major�minor�patchr r r! �_version_tuplee s
�rL c C s. | r t | t�s t| �} | rd�tt| ��S dS )N�.z (unknown))r. �tuplerL r< �mapr )rH r r r! �_version_stringq s
rP c @ s� e Zd ZdZee� dZedddd�Zdd� Z d d
� Z
d d d d d i fd d
�Z dd� Z d d d d d d i fdd�Z
d d d d d i fdd�Zd d d d d i fdd�Zdd� Zdd� Zdd� Zd d � Zed d � �Zd d!� Zd S )"�PostgreSQLConnectorz�Database connector for PostgreSQL.
Database URLs should be of the form:
{{{
postgres://user[:password]@host[:port]/database[?schema=my_schema]
}}}
F�trac� pg_dump_path�pg_dumpz1Location of pg_dump for Postgres database backupsc C s dt | j� | _d S )Nz#server: (not-connected), client: %s)rP �_client_version�_postgresql_version��selfr r r! �__init__� s ��z PostgreSQLConnector.__init__c c s � dV d S )N)�postgresr' r rW r r r! �get_supported_schemes� s �
z)PostgreSQLConnector.get_supported_schemesNc C sv |� dd� t|||||||�}t|j�} t| j�}
| js9|jtk r/td| tt�d�} t| ��d| |
f | _ d| _|S )N�schema�publiczOPostgreSQL version is %(version)s. Minimum required version is %(min_version)s.)r# � min_versionzserver: %s, client: %sT)
�
setdefault�PostgreSQLConnectionrP �server_versionrU �required�min_postgresql_versionr � TracErrorrV ) rX r> �logr3 r4 r5 r6 �params�cnx�
server_ver�
client_ver�errorr r r! �get_connection� s$ �
�
�z"PostgreSQLConnector.get_connectionc C s t S r )�psycopgrW r r r! �get_exceptions� � z"PostgreSQLConnector.get_exceptionsc
C s� | � |||||||�} | �� }
| jr)| jdkr)|
�dt| j� � |
�d| jf� |d u r3ddlm} |D ]} | �| �D ]} |
�| � q<q5| �� d S )Nr] zCREATE SCHEMA �SET search_path TO %sr )r\ )rk �cursorr\ �executerD �trac.db_default�to_sql�commit)
rX r> r\ re r3 r4 r5 r6 rf rg rp �table�stmtr r r! �init_db� s � � zPostgreSQLConnector.init_dbc
C sh | � |||||||�}|jr |jdkr |�dt|j� � n|�� D ] } |�dt| � � q"|�� d S )Nr] zDROP SCHEMA %s CASCADEz
DROP TABLE %s)rk r\ rq rD �get_table_namesrt )
rX r> re r3 r4 r5 r6 rf rg ru r r r! �
destroy_db� s � z PostgreSQLConnector.destroy_dbc
C s: | � |||||||�}|�� } | �d|jf� | �� d S )NzU
SELECT EXISTS(SELECT 1 FROM pg_namespace WHERE nspname=%s);
r )rk rp rq r\ �fetchone)
rX r> re r3 r4 r5 r6 rf rg rp r r r! � db_exists� s �� z PostgreSQLConnector.db_existsc c s2 � dt |j� g}g }|jD ]-}|j}t�||�}|jr d}t|j�dkr/|j|jv r/|d7 }|� dt |j�|f � qt|j�dkrZ|� dt |jd �d�
d d
� |jD ��f � |� d �
|�d � d
�
|�V |j D ])}|j rtdnd}d|t d|jd�
|j�f �t |j�d�
dd
� |jD ��f V qmd S )NzCREATE TABLE %s (�SERIALr' z PRIMARY KEYz %s %sz" CONSTRAINT %s PRIMARY KEY (%s)�_pk�,c s � � | ]}t |�V qd S r �rD �r8 �colr r r! r; � � � z-PostgreSQLConnector.to_sql.<locals>.<genexpr>z,
z
)�
�UNIQUE� z CREATE %s INDEX %s ON %s (%s)z %s_%s_idxr c s r r r� r� r r r! r; � r� )
rD r9 �columns�type� _type_map�get�auto_increment�len�key�appendr< �indices�unique)rX ru �sql�coldefs�column�ctype�indexr� r r r! rs � s: �
��
���zPostgreSQLConnector.to_sqlc c sx � g }t |�� �D ]\}\}}t�||�}|t�||�kr$|�||f� q |r:dt|�d�dd� |D ��f V dS dS )z�Yield SQL statements altering the type of one or more columns of
a table.
Type changes are specified as a `columns` dict mapping column names
to `(from, to)` SQL type tuples.
zALTER TABLE %s %sz, c s s$ � | ]
\}}d t |�|f V qdS )zALTER COLUMN %s TYPE %sNr� )r8 r9 �type_r r r! r; � s � �z9PostgreSQLConnector.alter_column_types.<locals>.<genexpr>N)�sortedr= r� r� r� rD r< )rX ru r� � alterationsr9 �from_�tor r r! �alter_column_types� s � �
��
��z&PostgreSQLConnector.alter_column_typesc
C s� | j j�dd�}t|�\}}|�di �}|�dd� tj�|d �}| jddd d
d g}d |v r8|� d
|d g� |�d|�d��}|r[|� d|g� d|vr[|� dt
|�dd��g� |� dd|d g� |d7 }|� d||g� tj � � } d|v r�t
|d �| d< z
t
|| ttd�}
W n ty� } z ttd| jt| �d ���d } ~ ww |
�� d } |
jd kr�ttd t| �� �d ���tj�|�s�ttd!���|S )"NrR �databaserf r\ r] r> z-Cz --insertsz-xz-Z�8r3 z-Ur5 z-h�/z-pr6 �5432z-nrA z.gzz-fr4 �
PGPASSWORD)�env�stderrr � Unable to run %(path)s: %(msg)s�r> �msgr' r zpg_dump failed: %(msg)s)r� zNo destination file created)r� �configr� r
r_ �osr> �basenamerS �extendr �environ�copyr r r �OSErrorrd r r � communicate�
returncoder �strip�exists)
rX � dest_file�db_url�scheme�db_prop� db_params�db_name�argsr5 r� �p�e�errmsgr r r! �backup� sH
���
� zPostgreSQLConnector.backupc c s � d| j fV dtfV d S )N�
PostgreSQL�psycopg2)rV �psycopg2_versionrW r r r! �get_system_info s � z#PostgreSQLConnector.get_system_infoc
C s� d }t td�rt�� }t |�S tr@zt�t�}|�� }W t |�S ty? } z| j� dt
|dd�� W Y d }~t |�S d }~ww t |�S )Nr$ z3Exception caught while retrieving libpq's version%sT)� traceback) �hasattrrl r$ �_libpq_pathname�ctypes�CDLL� PQlibVersion� Exceptionre �warningr rL )rX r# �libr� r r r! rU $ s"
�
�
���z#PostgreSQLConnector._client_versionc
C sV z t | jdgttd�}W n ty$ } z ttd| jt|�d���d }~ww |�� d S )Nz --version)�stdoutr r� r� r ) r rS r r r� rd r r r� )rX r� r� r r r! �_pgdump_version3 s
���� z#PostgreSQLConnector._pgdump_version)�__name__�
__module__� __qualname__�__doc__�
implementsr rb r rS rY r[ rk rm rw ry r{ rs r� r� r� r rU r� r r r r! rQ z s: �
�
�
�
�
'
rQ c @ s� e Zd ZdZdZdddddi fdd�Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z d/dd�Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd/d)d*�Zd+d,� Zed-d.� �ZdS )0r` z"Connection wrapper for PostgreSQL.TNc C s� |� d�r |dd � }d|v r|d }t�t|||||��}|�d� |�dd�| _| jdkrOz|�� �d| jf� |� � W n t
t fyN |� � Y nw t
�| ||� d S )Nr� r' r5 r r\ r] ro )�
startswithrl �connectr@ �set_client_encodingr� r\ rp rq rt r r �rollbackr rY ) rX r> re r3 r4 r5 r6 rf rg r r r! rY C s"
�
�z PostgreSQLConnection.__init__c C s t | j�� | j�S r )r rg rp re rW r r r! rp X s zPostgreSQLConnection.cursorc C s d|t �||�f S )NzCAST(%s AS %s))r� r� )rX r� r� r r r! �cast[ s zPostgreSQLConnection.castc G s
d� |�S )Nz||)r< )rX r� r r r! �concat_ s
zPostgreSQLConnection.concatc C s" | � d| �|�| �|�f � d S )NzA
ALTER TABLE %s DROP COLUMN IF EXISTS %s
�rq r1 �rX ru r� r r r! � drop_columnb s
�z PostgreSQLConnection.drop_columnc C s | � d| �|� � d S )NzDROP TABLE IF EXISTS r� )rX ru r r r! �
drop_tableg s z PostgreSQLConnection.drop_tablec C s | � d|f�}dd� |D �S )Nz�
SELECT column_name FROM information_schema.columns
WHERE table_schema=current_schema() AND table_name=%s
ORDER BY ordinal_position
c S � g | ]}|d �qS �r r �r8 �rowr r r! �
<listcomp>p � z9PostgreSQLConnection.get_column_names.<locals>.<listcomp>�rq �rX ru �rowsr r r! �get_column_namesj s �z%PostgreSQLConnection.get_column_names�idc C s( |� d| �| �||��f� |�� d S )NzSELECT CURRVAL(%s)r )rq r1 �_sequence_namerz �rX rp ru r� r r r! � get_last_idr s � z PostgreSQLConnection.get_last_idc s, dd� � � dd�D �}t� fdd�|D ��S )Nc S s g | ] \}|d t d� � �qS )N�_id_seq)r� �r8 r9 r r r! r� x s z;PostgreSQLConnection.get_sequence_names.<locals>.<listcomp>a
SELECT c.relname
FROM pg_class c
INNER JOIN pg_namespace n ON c.relnamespace = n.oid
WHERE n.nspname = ANY (current_schemas(false))
AND c.relkind='S' AND c.relname LIKE %s ESCAPE '!'
)z
%!_id!_seqc 3 s � | ] }|� � � v r|V qd S r )rx r� rW r r! r; s � z:PostgreSQLConnection.get_sequence_names.<locals>.<genexpr>)rq r� )rX �seqsr rW r! �get_sequence_namesw s �z'PostgreSQLConnection.get_sequence_namesc C s | � d�}dd� |D �S )Nzm
SELECT table_name FROM information_schema.tables
WHERE table_schema=current_schema()c S r� r� r r� r r r! r� � r� z8PostgreSQLConnection.get_table_names.<locals>.<listcomp>r� )rX r� r r r! rx � s
z$PostgreSQLConnection.get_table_namesc C s | � d|f�}|d d S )Nz�
SELECT EXISTS (SELECT * FROM information_schema.columns
WHERE table_schema=current_schema()
AND table_name=%s)
r r� r� r r r! � has_table� s � z PostgreSQLConnection.has_tablec C � dS )NzILIKE %s ESCAPE '/'r rW r r r! �like� rn zPostgreSQLConnection.likec C s t �d|�S )Nz/\1)�_like_escape_re�sub)rX �textr r r! � like_escape� � z PostgreSQLConnection.like_escapec C s | j �� }|�d� d S )NzSELECT 1)rg rp rq )rX rp r r r! �ping� s
zPostgreSQLConnection.pingc C r� )NzLIKE %s ESCAPE '/'r rW r r r! � prefix_match� rn z!PostgreSQLConnection.prefix_matchc C s | � |�d S )N�%)r� )rX �prefixr r r! �prefix_match_value� s z'PostgreSQLConnection.prefix_match_valuec C s t |�S r r� )rX rC r r r! r1 � s zPostgreSQLConnection.quotec C s` | � � }|�d| jf� |�� D ]
\}|�d| � q| �� }|D ] }|�d| �|� � q!|S )Nzu
SELECT sequence_name FROM information_schema.sequences
WHERE sequence_schema=%s
z ALTER SEQUENCE %s RESTART WITH 1z DELETE FROM )rp rq r\ �fetchallrx r1 )rX rp �seq� table_namesr9 r r r! � reset_tables� s �z!PostgreSQLConnection.reset_tablesc C s4 |� d| �|�| �|�f | �| �||��f� d S )Nz,SELECT SETVAL(%%s, (SELECT MAX(%s) FROM %s)))rq r1 r� r� r r r! �update_sequence� s
��z$PostgreSQLConnection.update_sequencec C s d||f S )Nz %s_%s_seqr r� r r r! r� � r� z#PostgreSQLConnection._sequence_namec C s t | jj�S r )rL rg ra rW r r r! ra � s z#PostgreSQLConnection.server_version)r� )r� r� r� r� �poolablerY rp r� r� r� r� r� r� r� rx r� r� r� r� r� r� r1 r� r� r� r ra r r r r! r` > s6
�
r` )NNNN)J� ctypes.utilr r� r� �re�
pkg_resourcesr �
subprocessr r � trac.core� trac.configr � trac.db.apir r r
� trac.db.utilr r � trac.utilr
r �trac.util.compatr �trac.util.htmlr �trac.util.textr r r �trac.util.translationr r� rl �psycopg2.extensionsr r r r r r r � ImportErrorr� r� � __version__r� r� r� �_f�_match�open�_psycopg�__file__r9 �search�encode�read�VERBOSE�group�
IGNORECASE�AttributeError�compiler� r� rc r@ rD rL rP � ComponentrQ r` r r r r! �<module> s� �
�
�
��� ���
�
E