Mini Shell
3
V]�_�= � @ s� d dl mZmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZmZmZ d d lZejd
d � ejd d � d d
lmZmZ dddgZG dd� de�ZG dd� ded�ZG dd� de�Z d S )� )�ABCMeta�abstractmethod)�partial)�
PropertyProxy)� ErrorMapper)�Signal)�DBUS_FLAG_NONE)�DBusSpecification)� get_variant�get_variant_type�unwrap_variantN�Gioz2.0�GLib)r
r �
GLibClient�AbstractClientObjectHandler�ClientObjectHandlerc @ s� e Zd ZdZejZeeefdd��Z ef eefdd��Z
edd� �Z ef efdd ��Z ed
d � �Z
ed d
� �Zedd� �Zedd� �Zedd� �ZdS )r z0The low-level DBus client library based on GLib.c
C s |j |||||||| d� S )zVSynchronously call a DBus method.
:return: a result of the DBus call
N)Z call_sync)
�cls�
connection� service_name� object_path�interface_name� method_name�
parameters�
reply_type�flags�timeout� r � /usr/lib/python3.6/handler.py� sync_call1 s zGLibClient.sync_callc C s( |j |||||||
| | j|| fd�
dS )z"Asynchronously call a DBus method.)�callback� user_dataN)�call�_async_call_finish) r r r r r r r r r �
callback_argsr r r r r �
async_callE s zGLibClient.async_callc s$ |\}}|� �fdd�f|�� dS )z(Finish an asynchronous DBus method call.c s
�j � �S )N)Z call_finishr )�
result_object�
source_objectr r �<lambda>` s z/GLibClient._async_call_finish.<locals>.<lambda>Nr )r r& r% r r r# r )r% r& r r" X s z GLibClient._async_call_finishc
C s. |j ||||d|| j||fd�} t| j|| �S )zKSubscribe to a signal.
:return: a callback to unsubscribe
N)r r )Zsignal_subscribe�_signal_callbackr �_unsubscribe_signal)
r r r r r � signal_namer r# r �subscription_idr r r �subscribe_signald s zGLibClient.subscribe_signalc
C s |\}} ||f| �� dS )z8A callback that is called when a DBus signal is emitted.Nr )
r r Z sender_namer r r* r r r r# r r r r( } s zGLibClient._signal_callbackc C s |j |� dS )zUnsubscribe from a signal.N)Zsignal_unsubscribe)r r r+ r r r r) � s z GLibClient._unsubscribe_signalc C s t |tj�otjj|�S )zIs it a remote DBus error?)�
isinstancer �Errorr
� DBusError�is_remote_error)r �errorr r r r0 � s zGLibClient.is_remote_errorc C s t jj|�S )z)Get a DBus name of the remote DBus error.)r
r/ Zget_remote_error)r r1 r r r �get_remote_error_name� s z GLibClient.get_remote_error_namec C s: | j |�}|j}djd|�}|j|�r6|t|�d� S |S )z'Get a message of the remote DBus error.z{}:{}: z GDBus.ErrorN)r2 �message�format�
startswith�len)r r1 �namer3 �prefixr r r �get_remote_error_message� s
z#GLibClient.get_remote_error_messageN)�__name__�
__module__� __qualname__�__doc__r ZMAXINTZDBUS_TIMEOUT_NONE� classmethodr r r$ r" r, r( r) r0 r2 r9 r r r r r + s
c @ s� e Zd ZdZddddgZdd� Zedd � �Zed
d � �Zed d
� �Z e
dd� �Z dd� Z dd� Z
dd� Ze
dd� �Ze
dd� �Ze
dd� �Ze
d d � �Zd S ) r z-The abstract handler of a remote DBus object.� _message_bus�
_service_name� _object_path�_specificationc C s || _ || _|| _d| _dS )z�Create a new handler.
:param message_bus: a message bus
:param service_name: a DBus name of the service
:param object_path: a DBus path the object
N)r? r@ rA rB )�self� message_busr r r r r �__init__� s z$AbstractClientObjectHandler.__init__c C s | j S )z9DBus service name.
:return: a DBus name
)r@ )rC r r r r � s z(AbstractClientObjectHandler.service_namec C s | j S )z8DBus object path.
:return: a DBus path
)rA )rC r r r r � s z'AbstractClientObjectHandler.object_pathc C s | j s| j� | _ | j S )zDBus specification.)rB �_get_specification)rC r r r �
specification� s
z)AbstractClientObjectHandler.specificationc C s t � S )zKIntrospect the DBus object.
:return: a DBus specification
)r )rC r r r rF � s z.AbstractClientObjectHandler._get_specificationc C s" | j ||�}| jt|��}||�S )z�Create a member of the DBus object.
:param interface_name: a name of the interface
:param member_name: a name of the member
:return: a signal, a method or a property
)�_find_member_spec�
_find_handler�type)rC r � member_name�specZhandlerr r r �
create_member� s z)AbstractClientObjectHandler.create_memberc C s | j j||�S )z�Find a specification of the DBus member.
:param interface_name: a name of the interface
:param member_name: a name of the member
:return: a specification of the member
)rG Z
get_member)rC r rK r r r rH � s z-AbstractClientObjectHandler._find_member_specc C sD |t jkr| jS |t jkr | jS |t jkr0| jS tdj|j ���dS )z�Find a handler for the given member type.
:param member_type: a type of the member
:return: a callback
zUnsupported type: {}N)
r ZProperty�
_get_propertyZMethod� _get_methodr � _get_signal� TypeErrorr4 r: )rC � member_typer r r rI � s
z)AbstractClientObjectHandler._find_handlerc C s dS )z�Get a proxy of the DBus property.
:param property_spec: a property_specification
:return: a property object
Nr )rC �
property_specr r r rN s z)AbstractClientObjectHandler._get_propertyc C s dS )zGet a proxy of the DBus method.
:param method_spec: a method specification
:return: a callable object
Nr )rC � method_specr r r rO s z'AbstractClientObjectHandler._get_methodc C s dS )z}Get a proxy of the DBus signal.
:param signal_spec: a signal specification
:return: a signal object
Nr )rC � signal_specr r r rP s z'AbstractClientObjectHandler._get_signalc C s dS )z�Disconnect members of the DBus object.
Unsubscribe from DBus signals and disconnect all
registered callbacks of the proxy signals.
Nr )rC r r r �disconnect_members s z.AbstractClientObjectHandler.disconnect_membersN)r: r; r< r= � __slots__rE �propertyr r rG r rF rM rH rI rN rO rP rV r r r r r � s" )� metaclassc s� e Zd ZdZddddgZdeef� fdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Z� ZS )#r z$The client handler of a DBus object.�_client�_signal_factory�
_error_mapper�_subscriptionsNc s2 t � j|||� || _|| _|p$t� | _g | _dS )aB Create a new handler.
:param message_bus: a message bus
:param service_name: a DBus name of the service
:param object_path: a DBus path the object
:param error_mapper: a DBus error mapper
:param client: a DBus client library
:param signal_factory: a signal factory
N)�superrE rZ r[ r r\ r] )rC rD r r Z error_mapperZclientZsignal_factory)� __class__r r rE 2 s
z ClientObjectHandler.__init__c C s | j dddd�}tj|�S )zIntrospect the DBus object.z#org.freedesktop.DBus.IntrospectableZ
IntrospectNz(s))� _call_methodr Zfrom_xml)rC Zxmlr r r rF D s z&ClientObjectHandler._get_specificationc C sR | j � }| jj| jj| j| j|j|j| j |j
fd�}| j j |� | j j |j
� |S )z Get a proxy of the DBus signal.)r r# )r[ rZ r, r? r r@ rA r r7 r( �emitr] �appendZ
disconnect)rC rU �signalZ unsubscriber r r rP O s z ClientObjectHandler._get_signalc C s |t |�� dS )z8A callback that is called when a DBus signal is emitted.N)r )rC r r r r r r( e s z$ClientObjectHandler._signal_callbackc C s6 d}d}|j rt| j|�}|jr,t| j|�}t||�S )z!Get a proxy of the DBus property.N)�readabler �_get_property_value�writable�_set_property_valuer )rC rS �getter�setterr r r rN i s z!ClientObjectHandler._get_propertyc C s | j dddd|j|j�}t|�S )z!Get a value of the DBus property.z org.freedesktop.DBus.PropertiesZGetz(ss)z(v))r` r r7 r )rC rS Zvariantr r r re v s z'ClientObjectHandler._get_property_valuec
C s" | j dddd|j|jt|j|��S )z!Set a value of the DBus property.z org.freedesktop.DBus.Properties�Setz(ssv)N)r` r r7 r
rJ )rC rS Zproperty_valuer r r rg � s z'ClientObjectHandler._set_property_valuec C s t | j|j|j|j|j�S )z(Get a callable proxy of the DBus method.)r r` r r7 �in_type�out_type)rC rT r r r rO � s z ClientObjectHandler._get_methodc O s� |sd}|dk rt ||�}d}|dk r.t|�}| jj| j| j||||f}|jdd�} |jdt� �}
| s|| j| j j
f|�|�S | j j ||| j | |
fd���S dS )zKCall a DBus method.
:return: a result of the call or None
Nr r# )r r# )
r
r r? r r@ rA �pop�tuple�_get_method_replyrZ r r$ �_method_callback) rC r r rk rl r �kwargsr �argsr r# r r r r` � s4
z ClientObjectHandler._call_methodc s |� �fdd�f|�� dS )z/A callback of an asynchronous DBus method call.c s
�j � �S )N)ro r )rh rC r r r' � s z6ClientObjectHandler._method_callback.<locals>.<lambda>Nr )rC rh r r# r )rh rC r rp � s z$ClientObjectHandler._method_callbackc O s@ y|||�}| j |�S tk
r: } z
| j|�S d}~X nX dS )a Get a result of a DBus call.
:param call: a callback
:param args: arguments of the callback
:param kwargs: keyword arguments of the callback
:return: a result of the callback
:raise: an exception raised by the callback
N)�_handle_method_result� Exception�_handle_method_error)rC r! rr rq �resultr1 r r r ro � s
z%ClientObjectHandler._get_method_replyc C sL | j j|�s|�| j j|�}| jj|�}| j j|�}||�}||_|d�dS )zcHandle an error of a DBus call.
:param error: an exception raised during the call
N)rZ r0 r2 r\ Zget_exception_typer9 Z dbus_name)rC r1 r7 r r3 Z exceptionr r r ru � s z(ClientObjectHandler._handle_method_errorc C s( t |�}|sdS t|�dkr$|d S |S )zPHandle a result of a DBus call.
:param result: a variant tuple
N� r )r r6 )rC rv �valuesr r r rs � s z)ClientObjectHandler._handle_method_resultc C s x| j r| j j� }|� qW dS )z&Disconnect members of the DBus object.N)r] rm )rC r r r r rV s
z&ClientObjectHandler.disconnect_members)r: r; r< r= rW r r rE rF rP r( rN re rg rO r` rp ro ru rs rV �
__classcell__r r )r_ r r ( s(
1) �abcr r � functoolsr Zdasbus.client.propertyr Z dasbus.errorr Z
dasbus.signalr Zdasbus.constantsr Zdasbus.specificationr Z
dasbus.typingr
r r ZgiZrequire_versionZ
gi.repositoryr
r �__all__�objectr r r r r r r �<module> s$ y