Your IP : 18.223.238.150
3
�uAcI$ � @ s� d Z ddlmZ ddlZddlZddlmZ ddlmZ ddlm Z dd d
d d d
gZ
dd� Z dZ G dd
� d
e
�Zdeddfdd �ZG dd � d ej�ZG dd� de
�Ze� Zdd � Zdi fdd
�ZdS )zmThis module contains convenience decorators and functions for using
PolicyKit with dbus services and clients.� )�absolute_importN)� decorator)�reduce� )�method_call_no_timeout� require_auth� enable_proxy�AUTHFAIL_DONTCATCH�NotAuthorizedException�AreAuthorizationsObtainable� IsSystemBusNameAuthorizedAsyncc s � fdd�}|S )u� Decorator for DBus service methods.
Specify that a user needs a specific PolicyKit authorization `polkit_auth´
to execute it.c s t | d�st�t| d� � | S )NZ_dbus_is_methodZ_slip_polkit_auth_required)�hasattr�AssertionError�setattr)�method)� polkit_auth� � /usr/lib/python3.6/polkit.py�require_auth_decorator/ s z,require_auth.<locals>.require_auth_decoratorr )r r r )r r r ) s zBorg.fedoraproject.slip.dbus.service.PolKit.NotAuthorizedException.c @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__r r r r r < s c s� | dkst | tj�st��dtfks0�dks0t��dksHt �tj�sHt��dks^t�t�s^t����fdd�� | dk r�t� | �S � fdd�}|S dS )u� Decorator for DBus proxy methods.
Let's you (optionally) specify either a result value or an exception type
and a callback which is returned, thrown or called respectively if a
PolicyKit authorization doesn't exist or can't be obtained in the DBus
mechanism, i.e. an appropriate DBus exception is thrown.
An exception constructor may and a callback must accept an `action_id´
parameter which will be set to the id of the PolicyKit action for which
authorization could not be obtained.
Examples:
1) Return `False´ in the event of an authorization problem, and call
`error_handler´:
def error_handler(action_id=None):
print "Authorization problem:", action_id
class MyProxy(object):
@polkit.enable_proxy(authfail_result=False,
authfail_callback=error_handler)
def some_method(self, ...):
...
2) Throw a `MyAuthError´ instance in the event of an authorization problem:
class MyAuthError(Exception):
def __init__(self, *args, **kwargs):
action_id = kwargs.pop("action_id")
super(MyAuthError, self).__init__(*args, **kwargs)
self.action_id = action_id
class MyProxy(object):
@polkit.enable_proxy(authfail_exception=MyAuthError)
def some_method(self, ...):
...Nc s� y
| ||�S t jk
r� } zr|j� }|jt�s2� |tt�d � }� d k rT� |d� �d k r�y�|d�}W n �� }Y nX |��tkr�� �S d }~X nX d S )N)� action_id)�dbus�
DBusExceptionZ
get_dbus_name�
startswith�AUTH_EXC_PREFIX�lenr )�func�p�k�eZexc_namer Zaf_exc)�authfail_callback�authfail_exception�authfail_resultr r �
_enable_proxyv s$
z#enable_proxy.<locals>._enable_proxyc s
t � | �S )N)r )r )r% r r �decorate� s z enable_proxy.<locals>.decorate)�
isinstance� collections�Callabler r �
issubclass� Exceptionr )r r$ r# r" r& r )r% r" r# r$ r r @ s *
c s$ e Zd ZdZdZ� fdd�Z� ZS )r
zqException which a DBus service method throws if an authorization
required for executing it can't be obtained.zAorg.fedoraproject.slip.dbus.service.PolKit.NotAuthorizedExceptionc s( | j jd | | _tt| �j||� d S )N�.)� __class__�_dbus_error_name�superr
�__init__)�selfr r r )r- r r r0 � s
z NotAuthorizedException.__init__)r r r �__doc__r. r0 �
__classcell__r r )r- r r
� s c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
e dd� �Z e
dd � �Ze
d
d � �Ze
d d
� �Ze
dd� �Zdd� Zdd� Zdd� Zdi fdd�ZdS )�PolKitz"Convenience wrapper around polkit.zorg.freedesktop.PolicyKit1z%/org/freedesktop/PolicyKit1/Authorityz$org.freedesktop.PolicyKit1.AuthorityNc C s4 || j kr0tjr0tjjtj� d t_d t_d t_d S )N)�
_dbus_namer4 � _PolKit__busZremove_signal_receiver�_PolKit__signal_receiver�_PolKit__interface)�cls�nameZ old_ownerZ new_ownerr r r �_on_name_owner_changed� s
z PolKit._on_name_owner_changedc C s0 t js*tj� t _t jj| jdd| jd�t _t jS )NZNameOwnerChangedzorg.freedesktop.DBus)Zhandler_functionZ signal_nameZdbus_interfaceZarg0)r4 r6 r Z SystemBusZadd_signal_receiverr; r5 r7 )r1 r r r �_bus� s
z PolKit._busc C s t js| jj� t _t jS )N)r4 �_PolKit__bus_namer< Zget_unique_name)r1 r r r � _bus_name� s zPolKit._bus_namec C sF t js@y"tj| jj| j| j�| j�t _W n tj k
r> Y nX t jS )N)
r4 r8 r � Interfacer< �
get_objectr5 �
_dbus_path�_dbus_interfacer )r1 r r r �
_interface� s
zPolKit._interfacec C s
t | j�S )N)�boolrC )r1 r r r �_polkit_present� s zPolKit._polkit_presentc C s> | j jdd�}tj|d�}y|j|�}W n d }Y nX |S )Nzorg.freedesktop.DBusz/org/freedesktop/DBus)r< r@ r r? ZGetConnectionUnixUser)r1 �system_bus_nameZ
bus_objectZ
bus_interfaceZuidr r r Z__dbus_system_bus_name_uid� s
z!PolKit.__dbus_system_bus_name_uidc C s6 | j s
dS | jjdd| jif|i dd�\}}}|p4|S )NTzsystem-bus-namer: r � )rE rC �CheckAuthorizationr> )r1 Z
authorization�
is_authorized� is_challenge�detailsr r r Z __authorization_is_obtainable� s z$PolKit.__authorization_is_obtainablec s8 � j s
dS t|tttf�s |f}t� fdd�|d�}|S )NTc s | o � j |�S )N)�$_PolKit__authorization_is_obtainable)�x�y)r1 r r �<lambda>� s z4PolKit.AreAuthorizationsObtainable.<locals>.<lambda>)rE r' �tuple�list�setr )r1 �authorizationsZ
obtainabler )r1 r r � s
z"PolKit.AreAuthorizationsObtainableTc
sd | j s � |d kp | j|�dk�S d}|r0|dO }� fdd�}| jjdd|if|||d||td� d S ) Nr r c s | \}}}� |� d S )Nr )�argsrI rJ rK )�
reply_handlerr r �reply_cb s
z7PolKit.IsSystemBusNameAuthorizedAsync.<locals>.reply_cbzsystem-bus-namer: rG )rU �
error_handlerZtimeout)rE �!_PolKit__dbus_system_bus_name_uidrC rH r ) r1 rF r rU rW � challengerK �flagsrV r )rU r r s
z%PolKit.IsSystemBusNameAuthorizedAsync)r r r r2 r5 rA rB r8 r6 r= r7 � classmethodr; �propertyr< r> rC rE rX rL r r r r r r r4 � s"
r4 c C s
t j| �S )N)�__polkitr )rS r r r r s Tc C s t j| |||||�S )N)r] r )rF r rU rW rY rK r r r r s
)r2 Z
__future__r r( r r � functoolsr Z constantsr �__all__r r �objectr r r r
r4 r] r r r r r r �<module> s( Wo