Your IP : 3.147.78.249
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� � | S )NZ_slip_polkit_auth_required)�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 s6 ���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, ...):
...c 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_proxyNc s
t � | �S )N)r )r )r# r r �decorate� s z enable_proxy.<locals>.decorate)r )r r" r! r r$ r )r# r r! r" r r @ s
6
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 r) � s
z NotAuthorizedException.__init__)r r r �__doc__r' r) �
__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_namer- � _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)r- r/ r Z SystemBusZadd_signal_receiverr4 r. r0 )r* r r r �_bus� s
z PolKit._busc C s t js| jj� t _t jS )N)r- �_PolKit__bus_namer5 Zget_unique_name)r* 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)
r- r1 r � Interfacer5 �
get_objectr. �
_dbus_path�_dbus_interfacer )r* r r r �
_interface� s
zPolKit._interfacec C s
t | j�S )N)�boolr< )r* 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)r5 r9 r r8 ZGetConnectionUnixUser)r* �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-namer3 r � )r> r< �CheckAuthorizationr7 )r* 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)r* r r �<lambda>� s z4PolKit.AreAuthorizationsObtainable.<locals>.<lambda>)r> �
isinstance�tuple�list�setr )r* �authorizationsZ
obtainabler )r* 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 )�argsrB rC rD )�
reply_handlerr r �reply_cb s
z7PolKit.IsSystemBusNameAuthorizedAsync.<locals>.reply_cbzsystem-bus-namer3 r@ )rO �
error_handlerZtimeout)r> �!_PolKit__dbus_system_bus_name_uidr< rA r ) r* r? r rO rQ � challengerD �flagsrP r )rO r r s
z%PolKit.IsSystemBusNameAuthorizedAsync)r r r r+ r. r: r; r1 r/ r6 r0 � classmethodr4 �propertyr5 r7 r< r> rR rE r r r r r r r- � s"
r- c C s
t j| �S )N)�__polkitr )rM r r r r s Tc C s t j| |||||�S )N)rW r )r? r rO rQ rS rD r r r r s
)r+ Z
__future__r � collectionsr r � functoolsr Z constantsr �__all__r r �objectr r r r
r- rW r r r r r r �<module> s( Wo