File: //usr/share/system-config-printer/__pycache__/asyncipp.cpython-39.pyc
a
��z_�h � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlZe� d� d dl
Z
d dl T d dl Z d dl Z e j
ejejd� G d d
� d
e j�ZG d d � d �ZG d
d� d�ZG dd� de�Zedkr�ed� G dd� d�Ze� e�� dS )� N)�GObject)�GLib)�Gdk)�Gtkz1.9.60)�*)�domain� localedirc @ sP e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zdd d
�Zdd� Z dd� Z
dS )�_IPPConnectionThreadNc
C sn t j�| � | �d� || _|| _|| _|| _| | _|| _ || _
|| _ t �
d�| _|| _d| _td| � d S )NT� F�+%s)� threading�Thread�__init__� setDaemon�_queue�_conn�host�port� _encryption�_reply_handler�_error_handler�
_auth_handler�queue�Queue� _auth_queue�user�
_destroyed�
debugprint)
�selfZmyqueue�conn�
reply_handler�
error_handler� auth_handlerr r r �
encryption� r$ �,/usr/share/system-config-printer/asyncipp.pyr 0 s
z _IPPConnectionThread.__init__c C s t �d| � d S �Nz-%s��debugr �r r$ r$ r% �__del__C s z _IPPConnectionThread.__del__c C s | j �|� d S �N)r �put�r �passwordr$ r$ r% �
set_auth_infoF s z"_IPPConnectionThread.set_auth_infoc
C s� | j d u rt�� | _ | jd u r(t�� | _| jd u r<t�� | _| jrPt�| j� n
t� � | _t�
| j � z$tj | j | j| jd�}| �
d � W n2 ty� } zd }| �|� W Y d }~n
d }~0 0 td� | j�� | _| j�� }tdt|� � |d u �r| j�� �q�n| j�r| j�� q�d| _|\}}}}}} |dk�r@|| _|dk�rP|| _| dk�r`| | _|dk�r|d | _t�| j� td| j � t�
| j � z6tj | j | j| jd�}td� | j�� | �
d � W q� t�y } z(td � | j�� | �|� W Y d }~q�d }~0 0 q�z`td
| � ||g|�R i |��}
|tj jj k�rb|
i k�rbt� tj d ��td � | �
|
� W n@ t �y� } z&td
t|� � | �|� W Y d }~n
d }~0 0 | j�� q�td� | ` | `| `| `| `| `!~t�
d � d S )N)r r r# z Awaiting further instructionsz
Next task: %sFTr z Set user=%s; reconnecting...z...reconnectedz ...failedzCall %s� z
...successz...failure (%s)zThread exiting)"r �cupsZ getServerr ZgetPortr Z
getEncryptionr ZsetUser�getUserZsetPasswordCB2�_auth�
Connection�_reply� RuntimeError�_errorr r �empty�idle�get�reprZ task_doner r r r ZadminGetServerSettings�__call__�IPPError�IPP_NOT_AUTHORIZED� Exceptionr r ) r r �e�item�fn�args�kwdsZrhZehZah�resultr$ r$ r% �runI s�
�
�
z_IPPConnectionThread.runc C s d| _ | j�d � d S )NT)r r r, r) r$ r$ r% �stop� s z_IPPConnectionThread.stopc s: � ���fdd�}�j d u r dS t�||� �j�� }|S )Nc s> t �� � d u r ��| �j� n��| �j��� t �� dS �NF)r �
threads_enterr r �
threads_leave)�prompt�r �method�resourcer r$ r% � prompt_auth� s z/_IPPConnectionThread._auth.<locals>.prompt_authr0 )r r �idle_addr r: )r rK r rM rN rO r. r$ rL r% r3 � s
z_IPPConnectionThread._authc s, � fdd�}� j s(� jr(t�|� j|� d S )Nc s& � j s"t�� | � j|� t�� dS rH �r r rI r rJ )�handlerrE r) r$ r% �
send_reply� s
z/_IPPConnectionThread._reply.<locals>.send_reply)r r r rP )r rE rS r$ r) r% r5 � s z_IPPConnectionThread._replyc s: � fdd�}� j s6� jr6td� j � t�|� j|� d S )Nc s& � j s"t�� | � j|� t�� dS rH rQ )rR �excr) r$ r% �
send_error� s
z/_IPPConnectionThread._error.<locals>.send_errorzAdd %s to idle)r r r r rP )r rT rU r$ r) r% r7 � s z_IPPConnectionThread._error)NNNNNNN)NNN) �__name__�
__module__� __qualname__r r* r/ rF rG r3 r5 r7 r$ r$ r$ r% r / s �
c
r c @ sT e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zdd
d�Z dd� Z
dd� Z dS )�
IPPConnectiona�
This class starts a new thread to handle IPP operations.
Each IPP operation method takes optional reply_handler,
error_handler and auth_handler parameters.
If an operation requires a password to proceed, the auth_handler
function will be called. The operation will continue once
set_auth_info (in this class) is called.
Once the operation has finished either reply_handler or
error_handler will be called.
Nc
C s� t d� || _t�� | _t| j| |||||||d� | _| j�� ttj j
�} g }
t tj �D ]H} | d dkrlqZt tj | �} t| �| kr�qZt
| | | �| �� |
�| � qZ|
| _t d| � d S )NzNew IPPConnection�r r! r" r r r r# r � r )r Z_parentr r r �thread�start�typer1 r4 Z getPrinters�dir�getattr�setattr�
_make_binding�append�bindings)
r r r! r" r r r r# �parentZ
methodtyperd �fnamerB r$ r$ r% r � s.
�
zIPPConnection.__init__c C s t �d| � d S r& r' r) r$ r$ r% r* s zIPPConnection.__del__c C sP t d| � | jD ]}t| |� q| j�� rLt d� | j�� t�d| j� d S )Nz DESTROY: %szStopping worker threadr
) r rd �delattrr\ �is_aliverG r Ztimeout_add_seconds� _reap_thread)r Zbindingr$ r$ r% �destroy s
zIPPConnection.destroyc C s( | j jr| j�� dS td| j � dS )NFz Thread %s still processing tasksT)r\ r9 r �joinr r) r$ r$ r% ri s
zIPPConnection._reap_threadc C s | j �|� dS )z*Call this from your auth_handler function.N)r\ r/ r- r$ r$ r% r/ s zIPPConnection.set_auth_infoc C s& t d� | j�d|fi ||df� d S )Nz Reconnect...TF)r r r, )r r r r! r$ r$ r% � reconnect! s �zIPPConnection.reconnectc s � �fdd�S )Nc s �j � g| �R i |��S r+ )�_call_function)rC rD �rB r r$ r% �<lambda>' � z-IPPConnection._make_binding.<locals>.<lambda>r$ )r rB r$ rn r% rb & s zIPPConnection._make_bindingc O sj d } }}d|v r"|d }|d= d|v r8|d }|d= d|v rN|d }|d= | j �||||||f� d S �NFr r! r" )r r, )r rB rC rD r r! r" r$ r$ r% rm ) s �z IPPConnection._call_function)NNNNNNNN)NN) rV rW rX �__doc__r r* rj ri r/ rl rb rm r$ r$ r$ r% rY � s �
rY c @ sh e Zd Zddd�Zdd� Zdd� Zdd � Zdd
d �Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )�_IPPAuthOperationNc C sh d| _ d| _d| _d| _d| _d| _|| _|| _| jj| _ || _
|| _ || _ || _
|| _td| � d S )NFr0 r )� _auth_called�
_dialog_shown�
_use_password�_cancel�
_reconnect� _reconnected�_userr � try_as_root� _try_as_root�
_client_fn� _client_args� _client_kwds�_client_reply_handler�_client_error_handlerr )r r r! r r rB rC rD r$ r$ r% r A s
z_IPPAuthOperation.__init__c C s t �d| � d S r& r' r) r$ r$ r% r* T s z_IPPAuthOperation.__del__c C s | ` | `| `| `| `| `d S r+ )r r} r~ r r� r� r) r$ r$ r% �_destroyW s z_IPPAuthOperation._destroyc C s | j d u r td� | �||�S | jr:td| � | �|�S | jrxd| _d| _td| | jf � |j| j| j | jd� d S d}t
|�t j kr�|j
\}}|t jks�|t jks�|t jkr�|t jk}n |t jkr�| �||�S | �|�S nRt
|�t jk�r(|j
\}|t jk�s|t jk�r |t jk}n
| �|�S n
| �|�S |�rFtd| � n td| � | j�r�| jd k�r�| jjjd
d k�s�|�r�td � d | _|j| j| j | jd� d S | j�s�| �|�S |jj}|jj}tj j ||d
� d| _ td| | jf � |j| j| j | jd� d S )Nz Connection/reconnection failedz,%s (_error_handler): canceled so chaining upFTz,%s (_error_handler): reconnecting (as %s)...�r r! z %s (_error_handler): forbiddenz#%s (_error_handler): not authorized�rootr �/zAuthentication: Try as root�r r r0 ) r} r �_reconnect_errorrw r7 rx ry rz rl �_reconnect_replyr^ r1 r= rC r> Z
IPP_FORBIDDENZIPP_AUTHENTICATION_CANCELEDZIPP_SERVICE_UNAVAILABLEZ HTTPErrorZHTTP_UNAUTHORIZEDZHTTP_FORBIDDENr| r r\ r rt r �authconn�global_authinfocache�remove_auth_inforv ) r r rT Z forbiddenr@ �m�sr r r$ r$ r% r! _ s�
��
��
�
��� �
���z _IPPAuthOperation.error_handlerc C s� | j dkrj| jd u r t�� | _| jr�|jj}|jj}tjj ||d�}|r�|d | jkrf|d | _
d| _ ~n&|jj}|jj}tjj ||d� d| _
d| _ | j r�t
d� d| _ |�| j
� d S d| _ |js�|�| j
� d S | j�r"tj| jjddtjjtjjtd�d �}|�td
�� |�� |� � d } |j �r8|j � � } | d u �rRtj |jd �}n td �| }
tj |
|jd
�}|� d� | jd u �r�t�� | _|�ddg� |�!d� |�"d� |�#� |�$d| j%� d| _d S )NFr� r r
Tr0 z%Supplying password after reconnectionzNot authorized�re ZmodalZdestroy_with_parentZ message_typeZbuttons�textz The password may be incorrect.)re zAuthentication (%s))�titlere �username�response)&rt rz r1 r2 r\ r r r� r� �lookup_auth_inforv ry r� r r/ �prompt_allowedru r �
MessageDialogr re � MessageType�ERROR� ButtonsTypeZCLOSE�_�format_secondary_textrF rj �semantic�current_operationZ
AuthDialogZ
set_promptZfield_grab_focusZset_keep_above�show_all�connect�_on_auth_dialog_response) r rK r rM rN r r �creds�d�opr� r$ r$ r% r" � st
�
�
�
�
z _IPPAuthOperation.auth_handlerc C s0 d| _ | jj�| j| j| j| j| j| j f� d S rH )
rt r r r, r} r~ r r� r! r" r) r$ r$ r% � submit_task� s �z _IPPAuthOperation.submit_taskc C s� |� � \}}|dkr| j}tjj||f| jjj| jjjd� || _ |�
� |t j j
ksb|t j jkr�d| _| j�d� tjj| jjj| jjjd� td� d S || jkr�|| _| j�|� td� d S || _|| _d| _| j�d� td| j � d S )Nr0 r� Tz
Auth canceledzPassword supplied.zWill try as %s)Z
get_auth_inforz r� r� Zcache_auth_infor r\ r r Z_dialog�hider � ResponseType�CANCELZ DELETE_EVENTrw r/ r� r rv rx )r �dialogr� r r. r$ r$ r% r� � s> ��
� �
z*_IPPAuthOperation._on_auth_dialog_responsec C s$ t d| j � | jd ur | �� d S )NzConnected as %s)r rz r} r� �r r rE r$ r$ r% r� s
z"_IPPAuthOperation._reconnect_replyc C s t d| j � | jjs$| �|� d S d }|jr8|j�� }|d u rJtd�}n td�| }tj | jj
ddtj j tj
j|d�}| jd u r�t|�tkr�d}n"t|�tjkr�|jd }nt|�}|�td| �� |�tjtjjtd �tjj� |� tjj� |� d
| j � t d | ||f � |� � d S ) NzFailed to connect as %szCUPS server errorzCUPS server error (%s)Tr� z!service-error-service-unavailabler
z3There was an error during the CUPS operation: '%s'.ZRetryr� z7%s (_reconnect_error): presenting error dialog (%s; %s)) r rz r r� r7 r� r� r� r r� re r� r� r� ZNONEr} r^ r6 r1 r= rC r; r� Z add_buttonsZ STOCK_CANCELr� r� �OKZset_default_responser� � _on_retry_server_error_responseZshow)r r rT r� �msgr� �messager$ r$ r% r� % sB
� � ��z"_IPPAuthOperation._reconnect_errorc C sn |� � |tjjkrHtd| | jjjf � | jj| jjj| j | j
d� n"td| � | � t �
dtd��� d S )Nz/%s: got retry response, reconnecting (as %s)...r� z%s: got cancel responser zOperation canceled)rj r r� r� r r r\ r rl r� r� r7 r1 r= r� )r r� r� r$ r$ r% r� L s �� z1_IPPAuthOperation._on_retry_server_error_responsec C sT t d| t|�f � | jrDt d| | jf � | �| j|� | �� n t d| � d S )Nz%s (_error): handling %sz%s (_error): calling %sz(%s (_error): no client error handler set)r r; r� r r� )r rT r$ r$ r% r7 X s �
z_IPPAuthOperation._error)NNNN)NN)rV rW rX r r* r� r! r" r� r� r� r� r� r7 r$ r$ r$ r% rs @ s �
S
A
' rs c
@ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�IPPAuthConnectionNTc
C sx || _ | | _|| _|
| _d } tjj||d�} | rJ| d dks@|rH| d } ~ t||| �}
tj | ||
j
|
j | |||d� d S )Nr� r r� rZ ) re r� r{ r� r� r� r� rs rY r r! r" )r r r! r" r r r# re r{ r� r� r r� r� r$ r$ r% r f s$ � �zIPPAuthConnection.__init__c C s d | _ t�| � d S r+ )r� rY rj r) r$ r$ r% rj ~ s zIPPAuthConnection.destroyc O sr d } }}d|v r"|d }|d= d|v r8|d }|d= d|v rN|d }|d= t ||| | jj|||�}|�� d S rq )rs r\ r r� )r rB rC rD r r! r" r� r$ r$ r% rm � s �z IPPAuthConnection._call_function)
NNNNNNNTTN)rV rW rX r rj rm r$ r$ r$ r% r� e s �
r� �__main__Tc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�UIc C s� t �� }|�d| j� t j�d�}|�d| j� t �� }|�|ddd� |� |� t j�d�| _
| j
�d| j � | j
� d� |�| j
ddd� d | _
|�� d S )Nrj ZConnectZclickedFr z Get Devices)r ZWindowr� rj ZButtonZnew_with_label�connect_clickedZVBoxZ
pack_start�add�get_devices_button� get_devices�
set_sensitiver r� )r �w�bZvboxr$ r$ r% r � s
z UI.__init__c C s. z| j �� W n ty Y n0 t�� d S r+ )r rj �AttributeErrorr Z main_quit)r Zwindowr$ r$ r% rj � s
z
UI.destroyc C s& | j r| j �� t| j| jd�| _ d S )Nr� )r rj r� � connected�connect_failed�r Zbuttonr$ r$ r% r� � s
�zUI.connect_clickedc C s t dt|� � | j�d� d S )Nz Success: %sT)r r; r� r� r� r$ r$ r% r� � s z UI.connectedc C s* t dt|� � | j�d� | j�� d S )NzExc %sF)r r; r� r� r rj �r r rT r$ r$ r% r� � s zUI.connect_failedc C s* |� d� td� | jj| j| jd� d S )NFzGetting devicesr� )r� r r Z
getDevices�get_devices_reply�get_devices_errorr� r$ r$ r% r� � s
�zUI.get_devicesc C s6 || j krtd� d S tdt|� � | j�d� d S )NzIgnoring stale replyzGot devices: %sT�r r r; r� r� r� r$ r$ r% r� � s
zUI.get_devices_replyc C s6 || j krtd� d S tdt|� � | j�d� d S )NzIgnoring stale errorzError getting devices: %sTr� r� r$ r$ r% r� � s
zUI.get_devices_errorN) rV rW rX r rj r� r� r� r� r� r� r$ r$ r$ r% r� � s r� )r Zconfigr1 Z
gi.repositoryr r r r r Zrequirer� r( �gettext�installZPACKAGEr r
r rY rs r� rV Z
set_debuggingr� �mainr$ r$ r$ r% �<module> s2
/c '1>