|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/trac/db/__pycache__/ |
Upload File : |
o
�k�`E � @ sN d dl Z d dlmZ e �de j�Zdd� ZG dd� de�ZG dd � d e�Z dS )
� N)�closingzA
'(?:[^']+|'')*' |
`(?:[^`]+|``)*` |
"(?:[^"]+|"")*" c C s dd� }t �|| �S )Nc S s | � d��dd�S )Nr �%z%%)�group�replace)�match� r �./usr/lib/python3/dist-packages/trac/db/util.py�repl s z sql_escape_percent.<locals>.repl)�_sql_escape_percent_re�sub)�sqlr r r r �sql_escape_percent s r
c @ sD e Zd ZdZddgZddd�Zdd� Zd d
� Zdd d �Zd
d� Z dS )�IterableCursorz�Wrapper for DB-API cursor objects that makes the cursor iterable
and escapes all "%"s used inside literal strings with parameterized
queries.
Iteration will generate the rows of a SELECT query one by one.
�cursor�logNc C s || _ || _d S �N)r r )�selfr r r r r �__init__* s
zIterableCursor.__init__c C s t | j|�S r )�getattrr �r �namer r r � __getattr__. s zIterableCursor.__getattr__c c s � | j �� }|s d S |V qr )r �fetchone)r �rowr r r �__iter__1 s �
�zIterableCursor.__iter__c
C s� | j rO| j �d|� z0|r | j �d|� | j�t|�|�}n| j�|�}t| jdd �}|d ur8| j �dt|�� |W S tyN } z| j �d|� � d }~ww |rZ| j�t|�|�S | j�|�S )NzSQL: %s�args: %r�rowszprefetch: %d rowszexecute exception: %r)r �debugr �executer
r �len� Exception)r r �args�rr �er r r r 8 s& �� zIterableCursor.executec
C s� | j r@| j �d|� | j �d|� |sd S z|d r$| j�t|�|�W S | j�||�W S ty? } z| j �d|� � d }~ww |sDd S |d rQ| j�t|�|�S | j�||�S )NzSQL: %rr r zexecutemany exception: %r)r r r � executemanyr
r )r r r! r# r r r r$ L s* ���zIterableCursor.executemanyr )
�__name__�
__module__� __qualname__�__doc__� __slots__r r r r r$ r r r r r ! s
r c @ sF e Zd ZdZdZddd�Zdd� Zdd d
�ZeZdd d �Z d
d� Z
dS )�ConnectionWrappera! Generic wrapper around connection objects.
:since 0.12: This wrapper no longer makes cursors produced by the
connection iterable using `IterableCursor`.
:since 1.0: added a 'readonly' flag preventing the forwarding of
`commit` and `rollback`
��cnxr �readonlyNFc C s || _ || _|| _d S r r+ )r r, r r- r r r r l s
zConnectionWrapper.__init__c C s | j r |dv r t�t| j|�S )N)�commit�rollback)r- �AttributeErrorr r, r r r r r q s z ConnectionWrapper.__getattr__c C sf | � |�}t| j�� �� }|�||dur|ng � |r |�� nd}W d � |S 1 s,w Y |S )a Execute an SQL `query`
The optional `params` is a tuple containing the parameter
values expected by the query.
If the query is a SELECT, return all the rows ("fetchall").
When more control is needed, use `cursor()`.
N)� check_selectr r, r r �fetchall�r �query�params�dqlr r r r r r v s
��zConnectionWrapper.executec C sZ | � |�}t| j�� ��}|�||� |r|�� nd}W d � |S 1 s&w Y |S )a9 Execute an SQL `query`, on a sequence of tuples ("executemany").
The optional `params` is a sequence of tuples containing the
parameter values expected by the query.
If the query is a SELECT, return all the rows ("fetchall").
When more control is needed, use `cursor()`.
N)r1 r r, r r$ r2 r3 r r r r$ � s
��z ConnectionWrapper.executemanyc C s$ |� � �d�}| jr|std��|S )a Verify if the query is compatible according to the readonly nature
of the wrapped Connection.
:return: `True` if this is a SELECT
:raise: `ValueError` if this is not a SELECT and the wrapped
Connection is read-only.
�SELECTz,a 'readonly' connection can only do a SELECT)�lstrip�
startswithr- �
ValueError)r r4 r6 r r r r1 � s
z ConnectionWrapper.check_select)NFr ) r% r& r' r( r) r r r �__call__r$ r1 r r r r r* a s
r* )
�re�
contextlibr �compile�VERBOSEr
r
�objectr r* r r r r �<module> s �@