File: //usr/lib/python3/dist-packages/trac/db/__pycache__/convert.cpython-310.pyc
o
�k�`d � @ sT d dl Z d dlZd dlmZmZ d dlmZ d dlmZ dd� Z dd� Z
d d
� Z dS ) � N)�DatabaseManager�get_column_names)�sqlite_backend)� printfoutc s` t d� |�d�r
d�j_��� �|�d�r%t�j�tjur%td�j ��t t
| �� � �}|�� }t
|�}t |� � �|@ } t |� � �}
t
j�� oLt
j�� �t�|||��|�d�rgt�|d� tjdk�d� nd�d � � �����fd d
�} z1|�� }t| �D ]} t d | dd � | ||| �}
t d
|
� q| |
@ D ]} |�|| � q�|�� W d S |�� � )NzCopying tables:zsqlite:Fzsrc_cursor.cursor is %r�OFF)� � � i� Tc
s �� d��|� � t��}d}�r � r t|t� t|� ��}| �|�}dd�dgt|� � }d}|� d| � ��|�}|sD |S |t|�7 }�rTtd ||d
d � �d urb|d krbt ||��}d
|d�t
| j|��f } �r�|� | d�|gt|� � t |d�� n|� | | |� q:)NzSELECT * FROM �d z(%s)�,z%sr z DELETE FROM Tz%d records
%s table... F��newline�reportzINSERT INTO %s (%s) VALUES � )
�execute�quoter �min�int�len�join� fetchmanyr �replace_report_query�map�sum� executemany)
�db�cursor�table�columns�n_rows� quoted_table�holders�count�rows�query��max_parameters�multirows_insert�progress� replace_cast�
src_cursor�src_dbr �1/usr/lib/python3/dist-packages/trac/db/convert.py�
copy_table1 s<
�� ���z copy_tables.<locals>.copy_tablez %s table... r z %d records.)r �
startswith�cnx�_eagerr �typer �PyFormatCursor�AssertionError�setr �get_table_names�get_sequence_names�sys�stdout�isatty�stderr�get_replace_cast�set_synchronous�sqlite_version�sorted�update_sequence�commit�rollback)�src_env�dst_envr+ �dst_db� src_dburi� dst_dburi�
src_tablesr �dst_dbm�tables� sequencesr- r r" r r% r, � copy_tables sF
�
rK c s� |� dd�|� dd�krd S t�d���fdd��ttdd� ���fdd �d
D �����s/d S t�d d ��� tj�� � �fd
d�}|S )N�:� z AS ([^)]+)c s � � | �d|��}|�d�S )N�namerM )�search�cast�group)r r1 �match)�type_rer r, � cast_typec s
z#get_replace_cast.<locals>.cast_typec S s | d | d � � kS )Nr rM ��lower)�src_dstr r r, �<lambda>g � z"get_replace_cast.<locals>.<lambda>c 3 s( � | ]}� �|�� � � �|�fV qd S �NrU )�.0�t)rT rD r+ r r, � <genexpr>h s � � �z#get_replace_cast.<locals>.<genexpr>)�textr �int64z%\bCAST\(\s*([^\s)]+)\s+AS\s+(%s)\s*\)�|c s �fdd�}� � || �S )Nc s$ | � � \}}d|� �|�� |�f S )NzCAST(%s AS %s))�groups�getrV )rR rN r1 )� type_mapsr r, �replaceq s �z7get_replace_cast.<locals>.replace_cast.<locals>.replace)�sub)r^ rd )�cast_rerc r r, r) p s z&get_replace_cast.<locals>.replace_cast)�split�re�compile�dict�filterr �
IGNORECASE)r+ rD rE rF r) r )rf rT rD r+ rc rS r, r; ^ s$
����r; c s* |� d�� � �fdd���fdd�| D �S )Nr$ c s t | �} �| � �| � < t| �S rZ )�list�tuple)�row)�idxr) r r, rd | s z%replace_report_query.<locals>.replacec s g | ]}� |��qS r r )r[ ro )rd r r, �
<listcomp>� rY z(replace_report_query.<locals>.<listcomp>)�index)r# r r) r )rp rd r) r, r z s
r ) rh r7 � trac.db.apir r �trac.dbr �trac.util.textr rK r; r r r r r, �<module> s
H