File: //usr/lib/python3/dist-packages/trac/ticket/__pycache__/batch.cpython-310.pyc
o
�k�`�* � @ s� d dl Z d dlT d dlmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZ d d lmZmZ d d
lmZ d d lmZmZ d d lmZmZ d d
lmZm Z m Z d dl m Z m Z m!Z! G dd� de"�Z#dS )� N)�*)�NotificationSystem)�IPermissionRequestor)�ITicketManipulator� TicketSystem)�Ticket)�BatchTicketChangeEvent)�to_list)� datetime_now�utc)�tag)�exception_to_unicode�
to_unicode)�_�tag_)�HTTPBadRequest�IRequestFilter�IRequestHandler)�Chrome�add_script_data� add_warningc @ s� e Zd ZdZeeee� dZe e
�Z e �
d�ZdZdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd S ) �BatchModifyModuleac Ticket batch modification module.
This component allows multiple tickets to be modified in one request from
the custom query page. For users with the TICKET_BATCH_MODIFY permission
it will add a [TracBatchModify batch modify] section underneath custom
query results. Users can choose which tickets and fields they wish to
modify.
Fz[;\s,]+z, c C s
|j dkS )Nz /batchmodify)� path_info)�self�req� r �3/usr/lib/python3/dist-packages/trac/ticket/batch.py�
match_request5 s
z BatchModifyModule.match_requestc C s� |j dkr ttd���|j�d� |j�dd�}|j�d�}i }t| j�� � D ]*}|d }|dd |jvr6d
nd vrQ|d d
krQd| }||jv rQ|j�|�||< q't
|j�dd��}| � |||||� |� |j�d�pp|j
�� � d S )N�POSTzInvalid request arguments.�TICKET_BATCH_MODIFY�batchmod_value_comment� �action�name)�id�
resolution�status�owner�time�
changetime�summary� description� TICKET_ADMIN)�reporterr �type�textarea�batchmod_value_�selected_tickets�
query_href)�methodr r �perm�require�args�getr �env�get_ticket_fieldsr �_save_ticket_changes�redirect�href�query) r r �commentr" �
new_values�fieldr# �arg_namer1 r r r �process_request8 s,
�
� � z!BatchModifyModule.process_requestc C s |S )Nr )r r �handlerr r r �pre_process_requestV s z%BatchModifyModule.pre_process_requestc C sZ |j dkr(|d ur(d|�d�v r(| �|||d � t| j�}|�|� |�|� |||fS )Nz/queryr �ticket�tickets)r r4 �add_template_datar r8 �add_auto_preview�add_wiki_toolbars)r r �template�data�metadata�chromer r r �post_process_requestY s
z&BatchModifyModule.post_process_requestc C s dddgfddgfgS )Nr �
TICKET_MODIFYr, r �r r r r �get_permission_actionsd s �z(BatchModifyModule.get_permission_actionsc C sz d|d< |j d p
|j�� |d< | �||�|d< td�dd�td�d d�td
�d d�td �d
d�g}t||| �� d� d S )NT� batch_modifyr2 �action_controls�add�+)r# �value�remove�-z add / remove�+-zset to�=)�batch_list_modes�batch_list_properties)�sessionr< r= �_get_action_controlsr r �_get_list_fields)r r rK rF r[ r r r rG i s �
�z#BatchModifyModule.add_template_datac C s dd� t | j��� D �S )Nc S s. g | ]}|d dkr|� d�dkr|d �qS )r. �text�format�listr# )r7 )�.0�fr r r �
<listcomp>w s
�z6BatchModifyModule._get_list_fields.<locals>.<listcomp>)r r8 r9 rP r r r r_ v s �z"BatchModifyModule._get_list_fieldsc s4 � fdd�|D �}i }i }|D ]6}t � j�jD ]-}|�||�p g D ]"\}} | |v r:t|| |�|| < || �|� q!||| < |g|| < q!qqdd� tt|�� �dd� dd�D �}
g } |
D ];} d } g }
g }|| d }� � ||| �D ]}|�
||| �\}}}| s�|} |�|� |
�|� qq| �| | t |�|
f� q\| S ) Nc s g | ]
}t � j|d ��qS )r$ )r r8 )rc �trP r r re | s z:BatchModifyModule._get_action_controls.<locals>.<listcomp>c S s g | ]\}}|�qS r r )rc �a�wr r r re � � c S s | d | d fS )N� r r )�itemr r r �<lambda>� s z8BatchModifyModule._get_action_controls.<locals>.<lambda>T)�key�reverser ) r r8 �action_controllers�get_ticket_actions�max�append�sorted�iter�items�_get_action_controllers� render_ticket_action_controlr )r r � ticket_datarF �action_weights�action_ticketsrf �ctrl�weightr" �sorted_actionsrS � first_label�hints�widgetsrE �
controller�label�widget�hintr rP r r^ { sN � ��
�� �
�
z&BatchModifyModule._get_action_controlsc c s@ � t | j�jD ]}dd� |�||�pg D �}||v r |V qdS )z>Generator yielding the controllers handling the given `action`c S s g | ]\}}|�qS r r )rc rh rg r r r re � ri z=BatchModifyModule._get_action_controllers.<locals>.<listcomp>N)r r8 ro rp )r r rE r" r� �actionsr r r rv � s ����z)BatchModifyModule._get_action_controllersc C s~ | � � }|�� }|D ]2}|j�d| �}|r<||v r || nd}|j�d| d�} |j�d| d�}
| �|| |
|�||< q
|S )N�batchmod_mode_r! �batchmod_primary_�batchmod_secondary_)r_ �copyr6 r7 � _change_list) r r rE r? � list_fields�_valuesr@ �mode�old�new�new2r r r �_get_updated_ticket_values� s �z,BatchModifyModule._get_updated_ticket_valuesc
C s� d}| j D ]}t|d�r |�||�D ]
}d}t|td|d�� qqg } |D ]U}
t| j|
�} | �|| |�} | �|| |�D ] }
| � |
�
|| |�� q;| � | � | j D ]$}|� || �D ]\}}d}|rnt|tdt
�|�|d�� qXt||� qXqP| �| � q%|sdS tt�}| jj�' | D ] } | j|j||d � | �|| |�D ] }
|
�|| |� q�q�W d � n1 s�w Y t|||j|||�}z t| j��|� W dS ty� } z| j�d
t |�� t|td t |�d�� W Y d}~dS d}~ww ) zSave changes to tickets.T�validate_commentFz*The ticket comment is invalid: %(message)s)�messagez2The ticket field %(field)s is invalid: %(message)s)r@ r� N)�whenz6Failure sending notification on ticket batchchange: %sz[The changes have been saved, but an error occurred while sending notifications: %(message)s) �ticket_manipulators�hasattrr� r r r r8 r� rv �update�get_ticket_changes�populate�validate_ticketr �strongrr r
r �db_transaction� save_changes�authname�apply_action_side_effectsr r �notify� Exception�log�errorr
r )r r r1 r? r> r" �valid� manipulatorr� rF �id_rf �values�ctlrr@ r� �event�er r r r: � sn
��
� �
�� �������z&BatchModifyModule._save_ticket_changesc C s� t || j�}t || j�}t || j�}|dkr|}nN|dkr,|D ] }||vr*|�|� q n;|dkrC|D ]}||v rA|�|� ||v s8q2n$|dkrg|D ] }||vrT|�|� qI|D ]}||v rf|�|� ||v s]qW| j�|�S )NrZ rU rX rY )r �list_separator_rerr rW �list_connector_string�join)r �old_list�new_list� new_list2r� � changed_list�entryr r r r� � s:
��
���
�
�� z BatchModifyModule._change_listN) �__name__�
__module__� __qualname__�__doc__�
implementsr r r �is_valid_default_handler�ExtensionPointr r� �re�compiler� r� r rB rD rN rQ rG r_ r^ rv r� r: r� r r r r r ! s&
% 6r )$r� � trac.core�trac.notification.apir � trac.permr �trac.ticket.apir r �trac.ticket.modelr �trac.ticket.notificationr � trac.utilr �trac.util.datefmtr
r �trac.util.htmlr �trac.util.textr
r �trac.util.translationr r � trac.web.apir r r �trac.web.chromer r r � Componentr r r r r �<module> s