File: //usr/lib/python3/dist-packages/trac/db/__pycache__/sqlite_backend.cpython-310.pyc
o
�k�`�C � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m
Z
m Z m Z d dl
mZmZ d dlmZmZmZ d dlmZmZ d dlmZmZmZ d d lm Z d d
l m Z m Z e� d �Z!e� d �Z"zd dl#m$Z% W n
e&yy d dl'Z%Y nw e%j(Z)e%j)Z*e%j+Z,ee%��-d
de, �Z.dZ/dZ0G dd� de%j1�Z2G dd� de2�Z3ddd�Z4dd� Z5G dd� de
�Z6G dd � d ee�Z7d d � Z8d d � Z9d!d"� Z:dS )#� N)�closing)�ConfigurationError�
ListOption)� Component� TracError�
implements)�ConnectionBase�IDatabaseConnector)�Table�Column�Index)�ConnectionWrapper�IterableCursor)� get_pkginfo�getuser�lazy)�tag)�_�tag_z([/_%])z[*?\[]�versionz%d.%d.%s)� r r )� � r c @ s& e Zd Zdd� Zddd�Zdd� ZdS ) �PyFormatCursorc O s6 z || g|�R i |��W S t jy | j�� � w �N)�sqlite�
DatabaseError�cnx�rollback)�self�function�args�kwargs� r# �8/usr/lib/python3/dist-packages/trac/db/sqlite_backend.py�_rollback_on_error4 s
�z!PyFormatCursor._rollback_on_errorNc C s* |r
|dt |� }| �tjj||pg �S )N��?)�lenr% r �Cursor�execute�r �sqlr! r# r# r$ r* ; s
�zPyFormatCursor.executec C s. |sd S |dt |d � }| �tjj||�S )Nr& r )r( r% r r) � executemanyr+ r# r# r$ r- A s �zPyFormatCursor.executemanyr )�__name__�
__module__� __qualname__r% r* r- r# r# r# r$ r 3 s
r c @ s6 e Zd Zdd� Zdd� Zdd� Zd dd �Zd
d � ZdS )
� EagerCursorc C s t �| |� g | _d| _d S �Nr )r �__init__�rows�pos)r �conr# r# r$ r3 Q s
zEagerCursor.__init__c G s( t j| g|�R � }t �| �| _d| _|S r2 )r r* �fetchallr4 r5 )r r! �resultr# r# r$ r* V s zEagerCursor.executec C s8 z| j | j }W n
ty Y d S w | jd7 _|S )N� )r4 r5 �
IndexError)r �rowr# r# r$ �fetchone\ s �zEagerCursor.fetchoneNc C s6 |d u r| j }| j| j| j| � }| j|7 _|S r )� arraysizer4 r5 )r �numr8 r# r# r$ � fetchmanye s
zEagerCursor.fetchmanyc C s | j | jd � }t| j �| _|S r )r4 r5 r( )r r8 r# r# r$ r7 m s zEagerCursor.fetchallr )r. r/ r0 r3 r* r< r? r7 r# r# r# r$ r1 P s
r1 �integer)�int�int64c c s � d| j g}g }| jD ].}|j�� }t�||�}|jr d}nt| j�dkr0|j | jv r0|d7 }|� d|j |f � q t| j�dkrM|� dd�
| j� � |� d�
|�d � d
�
|�V | j D ] }|j rgd nd }d
|| j d�
|j�| j d�
|j�f V q`d S )NzCREATE TABLE %s (zinteger PRIMARY KEYr9 z PRIMARY KEYz %s %sz UNIQUE (%s)�,z,
z
);�
�UNIQUE� z%CREATE %s INDEX %s_%s_idx ON %s (%s);r )
�name�columns�type�lower� _type_map�get�auto_incrementr( �key�append�join�indices�unique)�tabler, �coldefs�column�ctype�indexrR r# r# r$ �_to_sqlz s* �
��rX c @ s� e Zd ZdZee� dZedddd�ZdZ dd � Z
d
d � Z di fd d
�Z dd� Z
ddi fdd�Zdi fdd�Zdi fdd�Zdd� Zdd� Zdd� Zd d � Zed d � �ZdS ) �SQLiteConnectorzxDatabase connector for SQLite.
Database URLs should be of the form:
{{{
sqlite:path/to/trac.db
}}}
Fr �
extensionsz�Paths to [https://sqlite.org/loadext.html sqlite extensions].
The paths may be absolute or relative to the Trac environment.
)�docNc C s
d | _ d S r )�error�r r# r# r$ r3 � �
zSQLiteConnector.__init__c c sR � t tk rtdtdt d�| _n
ttk r tddt d�| _d| jr#dndfV d S ) NzKSQLite version is %(version)s. Minimum required version is %(min_version)s.z%d.%d.%d)r � min_versionz,Need at least PySqlite %(version)s or higher�r r ���r9 )�sqlite_version�min_sqlite_versionr �sqlite_version_stringr\ �pysqlite_version�min_pysqlite_versionr] r# r# r$ �get_supported_schemes� s �
��z%SQLiteConnector.get_supported_schemesc C sb d| _ | j|d< |dkr+z | j�� W | jS ttjfy* t|||�| _Y | jS w t|||�S )NTrZ �:memory:)�required� _extensions�
memory_cnx�cursor�AttributeErrorr r �SQLiteConnection�r �path�log�paramsr# r# r$ �get_connection� s
�� z SQLiteConnector.get_connectionc C s t S r )r r] r# r# r$ �get_exceptions� � z SQLiteConnector.get_exceptionsc s � fdd�}|dkre� � |�rttd|d���tj�|�}tj�|�s(t�|� tj |d t
|� dd��d�}t |�
� �� }t||� d �� t||� d
�� |||� W d � n1 s\w Y d |_n"� �|||�}t |�
� ��
}|||� W d � n1 s�w Y |�� d S ) Nc s< |d u r
ddl m} |D ]}� �|�D ]}| �|� qq d S )Nr )�schema)�trac.db_defaultrv �to_sqlr* )rl rv rS �stmtr] r# r$ �
insert_schema� s ��z.SQLiteConnector.init_db.<locals>.insert_schemarh z#Database already exists at %(path)s�rp �timeouti' )�isolation_levelr| � journal_mode� synchronous�DEFERRED)� db_existsr r �osrp �dirname�exists�makedirsr �connectrA rL r rl �_set_journal_mode�set_synchronousr} rs �commit) r rp rv rq rr rz �dirr rl r# r] r$ �init_db� s.
�
� � � zSQLiteConnector.init_dbc
C sp |dkr6t j�|�st j�| jj|�}zt �|� W d S ty5 } z|jtjkr*� W Y d }~d S d }~ww d S )Nrh ) r� rp �isabsrP �env�remove�OSError�errno�ENOENT)r rp rq rr �er# r# r$ �
destroy_db� s ����zSQLiteConnector.destroy_dbc C s t j�|�S r )r� rp r� ro r# r# r$ r� � s zSQLiteConnector.db_existsc C � t |�S r )rX �r rS r# r# r$ rx � � zSQLiteConnector.to_sqlc C sF t |�� �D ]\}\}}t�||�t�||�kr td||f ��qdS )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.
z+Conversion from %s to %s is not implementedr# )�sorted�itemsrK rL �NotImplementedError)r rS rH rG �from_�tor# r# r$ �alter_column_types� s ��z"SQLiteConnector.alter_column_typesc C s� ddl }| j�dd�}z |d|�d�� }W n ty Y nw tj�| jj|dd� �}|� ||� tj�
|�s?t t d���|S )zlSimple SQLite-specific backup of the database.
:param dest_file: Destination file basename
r N�trac�databaser' � zNo destination file created)
�shutil�configrL rW �
ValueErrorr� rp rP r� �copyr� r r )r � dest_filer� �db_str�db_namer# r# r$ �backup� s � zSQLiteConnector.backupc c s � dt fV dtfV d S )N�SQLite�pysqlite)rd �pysqlite_version_stringr] r# r# r$ �get_system_info s �
z SQLiteConnector.get_system_infoc C s< g }| j D ]}tj�|�stj�| jj|�}|�|� q|S r )rZ r� rp r� rP r� rO )r rj �extpathr# r# r$ rj s
zSQLiteConnector._extensions)r. r/ r0 �__doc__r r ri r rZ rk r3 rg rs rt r� r� r� rx r� r� r� r rj r# r# r# r$ rY � s* �
rY c @ s� e Zd ZdZddgZedkZdi fdd�Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zd1dd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd1d+d,�Zd-d.� Zd/d0� ZdS )2rn z Connection wrapper for SQLite.�_active_cursors�_eager)r r � Nc
C sj d | _ |dkr@t�|tj�sttd|d���tj�|�}t�|tjtj �r0t�|tjtj �s@tt
dt � t
� �t � |�d���t�� | _t|�dd��}|�dd �d k| _tj|tjd td
k |d �}|�d g �}t|�d
kr�|�d� |D ]}|�|� qv|�d� t|� � ��} t | |�d�� t | |�d�� W d � n1 s�w Y d|_ t �!| ||� d S )Nrh z Database "%(path)s" not found.r{ zyThe user %(user)s requires read _and_ write permissions to the database file %(path)s and the directory it is located in.)�userrp r| g $@rl �eager)r r r9 )� detect_typesr} �check_same_threadr| rZ r TFr~ r r� )"r r� �access�F_OKr r rp r� �R_OK�W_OKr r �coder �weakref�WeakKeyDictionaryr� rA rL r� r r� �PARSE_DECLTYPESrb r( �enable_load_extension�load_extensionr rl r� r� r} r
r3 )
r rp rq rr �dbdirr| r rZ �extrl r# r# r$ r3 $ sF � �
�
�
�zSQLiteConnection.__init__c C s2 | j �ttf| j �}d| j|< | |_ t|| j�S )NT)r rl r r1 r� r� r rq �r rl r# r# r$ rl J s
zSQLiteConnection.cursorc C s" | j D ]}|�� q| j�� d S r )r� �closer r r� r# r# r$ r P s
zSQLiteConnection.rollbackc C s0 t dkrd|t�||�f S |dkrd| S |S )N)r r r zCAST(%s AS %s)rA z1*)rb rK rL )r rU rI r# r# r$ �castU s
zSQLiteConnection.castc G s
d� |�S )Nz||)rP )r r! r# r# r$ �concat^ r^ zSQLiteConnection.concatc s� � � |�}||v rc� �|�}|�|g� |d }� �|�}� �|�}|�|� d�� fdd�|D ��}� �� } | �d||f � � �|� t |�D ]}
| �|
� qI| �d||||f � � �|� d S d S )N�_oldrC c 3 s � | ]}� � |�V qd S r )�quote)�.0�colr] r# r$ � <genexpr>j s � z/SQLiteConnection.drop_column.<locals>.<genexpr>zO
CREATE TEMPORARY TABLE %s AS SELECT * FROM %s
zG
INSERT INTO %s (%s) SELECT %s FROM %s
)
�get_column_names�_get_table_schema�remove_columnsr� r� rP rl r* �
drop_tablerX ) r rS rU � column_names� table_schema�
temp_table�
table_name�temp_table_name� cols_to_copyrl r, r# r] r$ � drop_columna s.
�
���z SQLiteConnection.drop_columnc C sZ | � � }tdk r!z
|�d| �|� � W d S tjy Y d S w |�d| �|� � d S )N)r r� r z DROP TABLE zDROP TABLE IF EXISTS )rl rb r* r� r �OperationalError�r rS rl r# r# r$ r� x s �zSQLiteConnection.drop_tablec C s dd� | � |�D �S )Nc S � g | ]}|d �qS )r9 r# �r� r; r# r# r$ �
<listcomp>� � z5SQLiteConnection.get_column_names.<locals>.<listcomp>)�_get_table_infor� r# r# r$ r� � � z!SQLiteConnection.get_column_names�idc C s |j S r )� lastrowid�r rl rS rU r# r# r$ � get_last_id� s z SQLiteConnection.get_last_idc C s g S r r# r] r# r# r$ �get_sequence_names� ru z#SQLiteConnection.get_sequence_namesc C s | � d�}dd� |D �S )NzK
SELECT name FROM sqlite_master WHERE type='table'
c S r� )r r# r� r# r# r$ r� � r� z4SQLiteConnection.get_table_names.<locals>.<listcomp>)r* )r r4 r# r# r$ �get_table_names� s
z SQLiteConnection.get_table_namesc C s t | �|��S r )�boolr� r� r# r# r$ � has_table� s zSQLiteConnection.has_tablec C s t dkrdS dS )N�r r9 r zLIKE %s ESCAPE '/'zLIKE %s)rb r] r# r# r$ �like� s zSQLiteConnection.likec C s t dkr
t�d|�S |S )Nr� z/\1)rb �_like_escape_re�sub)r �textr# r# r$ � like_escape� s z SQLiteConnection.like_escapec C s dS )NzGLOB %sr# r] r# r# r$ � prefix_match� ru z SQLiteConnection.prefix_matchc C s t �dd� |�d S )Nc S s d| � d� S )Nz[%s]r )�group)�mr# r# r$ �<lambda>� s z5SQLiteConnection.prefix_match_value.<locals>.<lambda>�*)�_glob_escape_rer� )r �prefixr# r# r$ �prefix_match_value� r� z#SQLiteConnection.prefix_match_valuec C r� r )�_quote)r �
identifierr# r# r$ r� � r� zSQLiteConnection.quotec C s, | � � }| �� }|D ] }|�d| � q
|S )NzDELETE FROM %s)rl r� r* )r rl � table_namesrG r# r# r$ � reset_tables� s
z SQLiteConnection.reset_tablesc C s d S r r# r� r# r# r$ �update_sequence� s z SQLiteConnection.update_sequencec C s$ | � � }|�d| �|� � t|�S )NzPRAGMA table_info(%s))rl r* r� �listr� r# r# r$ r� � s z SQLiteConnection._get_table_infoc
C s d }g }| � |�D ](}|d }|d }|d }|dkr%|dkr%|g}d}nd}|�t|||d�� q | �� } | �d| �|� � | �� D ]2}|d }
|d } | �d | �|
� � d
d � | D �} |d u rk|
�d �rk| }qD|�t| t | �d
�� qDt
||p|g d�| S )Nr9 r � r@ TF)rI rM zPRAGMA index_list(%s)zPRAGMA index_info(%s)c S r� )r r# r� r# r# r$ r� � r� z6SQLiteConnection._get_table_schema.<locals>.<listcomp>�sqlite_autoindex_)rR )rN ) r� rO r rl r* r� r7 �
startswithr r� r
)
r rS rN r� r; rU �type_�pkrM rl rW rR rH r# r# r$ r� � s0
� z"SQLiteConnection._get_table_schema)r� ) r. r/ r0 r� � __slots__rb �poolabler3 rl r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r# r# r# r$ rn s2 &
rn c C s d| � dd� S )Nz`%s`�`z``)�replace)r� r# r# r$ r� � s r� c C s� |sd S |� � }|dkrttd|d���| �dt|� � | �� }|s+ttdtd���|d p0d� � |kr>ttd |td
���d S ) N�OFFz8PRAGMA journal_mode `%(value)s` cannot be used in SQLite)�valuezPRAGMA journal_mode = %sz9PRAGMA journal_mode isn't supported by SQLite %(version)sr` r rF zEPRAGMA journal_mode `%(value)s` isn't supported by SQLite %(version)s)r r )�upperr r r* r� r< rd )rl r r; r# r# r$ r� � s$ ����r� c C s2 |sd S |� � rtt|��}| �dt|� � d S )NzPRAGMA synchronous = %s)�isdigit�strrA r* r� )rl r r# r# r$ r� � s
r� );r� r� �rer� �
contextlibr � trac.configr r � trac.corer r r � trac.db.apir r �trac.db.schemar
r r � trac.db.utilr
r � trac.utilr r r �trac.util.htmlr �trac.util.translationr r �compiler� r� �pysqlite2.dbapi2�dbapi2r � ImportError�sqlite3�sqlite_version_inforb rd � version_infore rL r� rc rf r) r r1 rK rX rY rn r� r� r� r# r# r# r$ �<module> sV
�
�� %� <