File: //usr/share/system-config-printer/__pycache__/cupspk.cpython-39.pyc
a
$x]�m � @ sx d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
edd� dZ dZ dZ
d ZG d
d � d �ZdS ) � N)�
debugprint)�
DBusGMainLoop)�reduceT)Zset_as_defaultz#org.opensuse.CupsPkHelper.Mechanism�/z1org.opensuse.CupsPkHelper.Mechanism.NotPrivilegedc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIS )J�
Connectionc C s` d | _ zt�� | _t�� | _W n" tjjy@ d | _| _Y n0 tj |||d�| _
| � � d S )N)�host�port�
encryption) �_parent�dbusZ
SessionBusZ _session_busZ SystemBus� _system_bus�
exceptions�
DBusException�cupsr � _connection�_hack_subclass)�selfr r r � r �*/usr/share/system-config-printer/cupspk.py�__init__5 s
�zConnection.__init__c C s` t | jj�}t| j�D ]D}|d dkr(qt| j|�}t |�|krBqt| |�st| ||j� qd S )Nr �_)�typer Z getPrinters�dir�getattr�hasattr�setattr�__call__)r Z
methodtype�fname�fnr r r r F s
zConnection._hack_subclassc C s
|| _ d S �N)r
)r �parentr r r �
set_parentT s zConnection.set_parentc C sL z | j �tt�}t�|t�W S tjjy4 Y d S t yF Y d S 0 d S r )
r Z
get_object� CUPS_PK_NAME� CUPS_PK_PATHr Z Interface�
CUPS_PK_IFACEr
r �AttributeError)r �objectr r r � _get_cups_pkX s zConnection._get_cups_pkc
O sN d }|� dd �}|sF| �� } | rFz| �|�}W n tjjyD Y n0 |sN|s\||i |��S d}
zX||d|i�}
t|
�tkr�|
dd � } |
d dkr�t| �dkr�| d W S | W S W �q W q` tjj�y } zD| � � t
kr�t d| � � � t
�t
jd��W Y d } ~ �q W Y d } ~ q`d } ~ 0 0 q`|
dk�rJt d|t|
�f � ||i |��S d S ) NZ signaturez PolicyKit communication issue� r � zDBus exception: %sZpkcancelz%PolicyKit call to %s did not work: %s)�popr' Zget_dbus_methodr r
r r �tuple�lenZ
get_dbus_name�CUPS_PK_NEED_AUTHr Zget_dbus_messager ZIPPErrorZIPP_NOT_AUTHORIZED�repr)
r Z use_fallbackZpk_function_name�pk_argsZfallback_function�args�kwdsZ pk_functionZdbus_args_signatureZcups_pkZ pk_retval�retval�er r r �_call_with_pk_and_fallbackd s>
&
�z%Connection._call_with_pk_and_fallbackc G s� dg}t |�t |�kr0d|d< |�|� t|�S d}tt |��D �]L}t|| �|| k�r�|| tkr�t|| �tkr�|�t|| �� q@n�|| tkr�t|| �tkr�|�t|| �� q@n�|| tkr�t|| �t kr�|�t|| �� q@n�|| tk�r || d u �r |�d� q@nd|| t
k�rRt|| �tk�rR|�t
|| �� q@n.|| t
k�r||| d u �r||�g � q@nd}|�|| � q@||d< t|�S )NFTr r) ) r, �extendr+ �ranger �str�int�append�float�bool�list)r �typesr0 r2 � exception�ir r r �_args_to_tuple� s<
zConnection._args_to_tuplec K s8 g }|D ]&}||v r$|� || � q|� d� qt|�S )Nr) )r9 r+ )r �namesr1 �ret�namer r r �
_kwds_to_vars� s zConnection._kwds_to_varsc s� d}d}g }g }d}t |�dkrF| jttttgg|�R � \}}}}}n@d|v rV|d }d|v rf|d }d|v rv|d }d|v r�|d }||||f}z&| j|d|| jjg|�R i |��} W n� t�yN td � d|v r�d
}t |�dkr�t d d � |�}nd
}t |�dk�rt dd � |�}nd
}|||f}| j|d|| jjg|�R i |��} Y n0 t | �
� �dk�r�t | t | �
� �d �t
k�r�| S i }
| d u�r�| �
� D ]8} t | �tjk�r�t| | �|
t| �< n | | |
| < �q�i } d}
dt|
� � � fdd�|
�
� D �}t |�dk�r�d }i }|D ]>} | d t | �t � � � }|dk�rL|
| ||< n|
| }�q|d u�rj|| |< |
d7 }
dt|
� � � fdd�|
�
� D �}�q�| S )NFr � �timeout�limit�include_schemes�exclude_schemesZ
DevicesGetz-DevicesGet API exception; using old signatureTc S s | d | S �N�,r ��x�yr r r �<lambda>� � z'Connection.getDevices.<locals>.<lambda>r) c S s | d | S rJ r rL r r r rO rP �:c s g | ]}|� � �r|�qS r ��endswith��.0rM �Zpostfixr r �
<listcomp>" rP z)Connection.getDevices.<locals>.<listcomp>z
device-urir( c s g | ]}|� � �r|�qS r rR rT rV r r rW 3 rP )r, r@ r8 r7 r4 r �
getDevices� TypeErrorr r �keysr r< �dictr �String)r r0 r1 �
use_pycupsrG rH rI rF r/ �resultZ
result_strr? Zdevices�nZ device_keysZ
device_uriZ device_dict�keyr rV r rX � s� & ���
��
��� 0
zConnection.getDevicesc O sD | j tgg|�R � \}}|f}| j|d|| jjg|�R i |�� d S )NZ JobCancel)r@ r8 r4 r � cancelJob�r r0 r1 r] �jobidr/ r r r ra ; s ���zConnection.cancelJobc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZJobSetHoldUntil)r@ r8 r7 r4 r �setJobHoldUntil)r r0 r1 r] rc Zjob_hold_untilr/ r r r rd G s ���zConnection.setJobHoldUntilc O sD | j tgg|�R � \}}|f}| j|d|| jjg|�R i |�� d S )NZ
JobRestart)r@ r8 r4 r �
restartJobrb r r r re P s ���zConnection.restartJobc
O s� d}d}t |�dkr2| jttgg|�R � \}}}n^| jtgg|�R � \}}d|v r\|d }n4d|v rn|d }n"d|v r�|d }n|s�t� �nd}|�s�|dus�|du�r�tjdd�\}} t�|� || f}
| j|d|
| j j
g|�R i |�� t� | tj �}t�
|d �} | �d
� |du�r\t�|d
tj� | �� } | d k�r�t�|| �d �� | �� } �q2n8|�d
� | �� } | d k�r�|�| �d �� | �� } �qn| �� t�| � n*||f}
| j|d|
| j j
g|�R i |�� dS )
af Keeping this as an alternative for the code.
We don't use it because it's not possible to know if the call was a
PK-one (and so we push the content of a temporary filename to fd or
file) or a non-PK-one (in which case nothing should be done).
filename = None
fd = None
file = None
if use_pycups:
if len(kwds) != 1:
use_pycups = True
elif kwds.has_key('filename'):
filename = kwds['filename']
elif kwds.has_key('fd'):
fd = kwds['fd']
elif kwds.has_key('file'):
file = kwds['file']
else:
use_pycups = True
if fd or file:
N� �filename�fd�filez/tmp)r ZFileGetZrtr r) zUTF-8)r, r@ r7 rY �tempfile�mkstemp�os�closer4 r �getFile�open�O_RDONLY�fdopen�seek�lseek�SEEK_SET�readline�write�encode�remove)
r r0 r1 � file_objectrh r] �resourcerg �tmpfd�tmpfnamer/ Ztmpfile�liner r r rn Y sf
���
���zConnection.getFilec
O s� t |�dkr*| jttgg|�R � \}}}n^| jtgg|�R � \}}d|v rT|d }n4d|v rf|d }n"d|v rx|d }n|s�t� �nd }|�s�|d us�|d u�r�t�� \}} t�|dtj� |d u�rt�|dtj� t� |d�}
|
dk�rF|
dk�rFt�
||
� t� |d�}
q�n4|� d� |� � } | dk�rFt�
|| � |� � } �q$t�
|� || f} | j|d | | jjg|�R i |�� t�| � n*||f} | j|d | | jjg|�R i |�� d S )
Nrf rg rh ri r i r) rP ZFilePut)r, r@ r7 rY rj rk rl rs rt �readrv rr ru rm r4 r �putFilerx )
r r0 r1 r] rz rg rh ry r{ r| �bufr} r/ r r r r � s\
��� ���zConnection.putFilec O s� | j tgg|�R � \}}| jg d�fi |��\}}}}} }
d} |sr|sr|
rrtjdd�\} }|
�| � t�| � d} |r�|s�|| |||f}
| j|d|
| j j
g|�R i |�� | r�t� |� n0|| |||f}
| j|d|
| j j
g|�R i |�� d S )N)rg �ppdname�info�location�device�ppdFT)�textZPrinterAddWithPpdFileZ
PrinterAdd) r@ r7 rD rj rk ZwriteFdrl rm r4 r �
addPrinter�unlink)r r0 r1 r] rC rg r� r� r� r� r� Z need_unlinkrh r/ r r r r� � s: "
��� ���zConnection.addPrinterc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetDevice)r@ r7 r4 r �setPrinterDevice)r r0 r1 r] rC r� r/ r r r r� � s ���zConnection.setPrinterDevicec O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetInfo)r@ r7 r4 r �setPrinterInfo)r r0 r1 r] rC r� r/ r r r r� s ���zConnection.setPrinterInfoc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetLocation)r@ r7 r4 r �setPrinterLocation)r r0 r1 r] rC r� r/ r r r r� s ���z Connection.setPrinterLocationc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetShared)r@ r7 r; r4 r �setPrinterShared)r r0 r1 r] rC Zsharedr/ r r r r� s ���zConnection.setPrinterSharedc O sP | j tttgg|�R � \}}}}|||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetJobSheets)r@ r7 r4 r �setPrinterJobSheets)r r0 r1 r] rC �start�endr/ r r r r� $ s
���z Connection.setPrinterJobSheetsc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetErrorPolicy)r@ r7 r4 r �setPrinterErrorPolicy�r r0 r1 r] rC Zpolicyr/ r r r r� . s ���z Connection.setPrinterErrorPolicyc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetOpPolicy)r@ r7 r4 r �setPrinterOpPolicyr� r r r r� 8 s ���z Connection.setPrinterOpPolicyc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetUsersAllowed)r@ r7 r< r4 r �setPrinterUsersAllowed�r r0 r1 r] rC Zusersr/ r r r r� B s ���z!Connection.setPrinterUsersAllowedc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterSetUsersDenied)r@ r7 r< r4 r �setPrinterUsersDeniedr� r r r r� L s ���z Connection.setPrinterUsersDeniedc O s| | j tttgg|�R � \}}}}|s,|f}n | j tttgg|�R � \}}}}|||f}| j|d|| jjg|�R i |�� d S )NZPrinterAddOptionDefault)r@ r7 r< r4 r �addPrinterOptionDefault) r r0 r1 r] rC �option�value�valuesr/ r r r r� U s
���z"Connection.addPrinterOptionDefaultc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZPrinterDeleteOptionDefault)r@ r7 r4 r �deletePrinterOptionDefault)r r0 r1 r] rC r� r/ r r r r� g s ���z%Connection.deletePrinterOptionDefaultc O sD | j tgg|�R � \}}|f}| j|d|| jjg|�R i |�� d S )NZ
PrinterDelete)r@ r7 r4 r �
deletePrinter�r r0 r1 r] rC r/ r r r r� q s ���zConnection.deletePrinterc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZClassAddPrinter)r@ r7 r4 r �addPrinterToClass�r r0 r1 r] ZprinterrC r/ r r r r� | s ���z Connection.addPrinterToClassc O sJ | j ttgg|�R � \}}}||f}| j|d|| jjg|�R i |�� d S )NZClassDeletePrinter)r@ r7 r4 r �deletePrinterFromClassr� r r r r� � s ���z!Connection.deletePrinterFromClassc O sD | j tgg|�R � \}}|f}| j|d|| jjg|�R i |�� d S )NZ ClassDelete)r@ r7 r4 r � deleteClassr� r r r r� � s ���zConnection.deleteClassc O sD | j tgg|�R � \}}|f}| j|d|| jjg|�R i |�� d S )NZPrinterSetDefault)r@ r7 r4 r �
setDefaultr� r r r r� � s ���zConnection.setDefaultc O sF | j tgg|�R � \}}|df}| j|d|| jjg|�R i |�� d S )NT�PrinterSetEnabled)r@ r7 r4 r �
enablePrinterr� r r r r� � s ���zConnection.enablePrinterc O sF | j tgg|�R � \}}|df}| j|d|| jjg|�R i |�� d S )NFr� )r@ r7 r4 r �disablePrinterr� r r r r� � s ���zConnection.disablePrinterc O sH | j tgg|�R � \}}|ddf}| j|d|| jjg|�R i |�� d S )NTr) �PrinterSetAcceptJobs)r@ r7 r4 r �
acceptJobsr� r r r r� � s
���zConnection.acceptJobsc O s^ | j tgg|�R � \}}| jdgfi |��\}|d|f}| j|d|| jjg|�R i |�� d S )N�reasonFr� )r@ r7 rD r4 r �
rejectJobs)r r0 r1 r] rC r� r/ r r r r� � s
���zConnection.rejectJobsc O sx d}d}| j |d|| jjg|�R i |��}i }|d urt|�� D ]4}t|�tjkrft|| �|t|�< q>|| ||< q>|S )NFr ZServerGetSettings)r4 r �adminGetServerSettingsrZ r r r\ r7 )r r0 r1 r] r/ r^ �settingsr? r r r r� � s" ��� z!Connection.adminGetServerSettingsc O sD | j tgg|�R � \}}|f}| j|d|| jjg|�R i |�� d S )NZServerSetSettings)r@ r[ r4 r �adminSetServerSettings)r r0 r1 r] r� r/ r r r r� � s ���z!Connection.adminSetServerSettingsN)'�__name__�
__module__� __qualname__r r r! r' r4 r@ rD rX ra rd re rn r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r 4 sH 4.a R6
r )rl �sysrj r r �debugr Zdbus.mainloop.glibr � functoolsr r" r# r$ r- r r r r r �<module> s