Mini Shell
3
��^�' � @ s� d dl Z d dlZd dlmZmZ d dlmZmZ d dlm Z d dl
m Z d dl m
Z
d dlZejdd� d d lmZ e je�Zd
d d d
dgZG dd
� d
e�ZG dd� ded�ZG dd � d e�ZG dd � d e�ZG dd
� d
e�ZG dd� de�Z dS )� N)�ABCMeta�abstractmethod)� DBUS_NAME_FLAG_ALLOW_REPLACEMENT�%DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER)� ObjectProxy)� ErrorMapper)�ServerObjectHandler�Gioz2.0)r �GLibConnection�
MessageBus�SystemMessageBus�SessionMessageBus�AddressedMessageBusc @ sP e Zd ZdZejjejjB Ze d dd��Z
e d
dd��Z e eddfdd��Z dS ) r
z4The low-level DBus connection library based on GLib.Nc C s t jd� tjtjj| �S )z Get a system bus connection.z Connecting to the system bus.)�log�infor � bus_get_sync�BusTypeZSYSTEM)� cancellable� r � /usr/lib/python3.6/connection.py�get_system_bus_connection6 s
z(GLibConnection.get_system_bus_connectionc C s t jd� tjtjj| �S )z Get a session bus connection.z Connecting to the session bus.)r r r r r ZSESSION)r r r r �get_session_bus_connection? s
z)GLibConnection.get_session_bus_connectionc C s t jj| |||�S )z3Get a connection to a bus at the specified address.)r ZDBusConnectionZnew_for_address_sync)Z bus_address�flagsZobserverr r r r � get_addressed_bus_connectionH s
z+GLibConnection.get_addressed_bus_connection)N)N)
�__name__�
__module__� __qualname__�__doc__r ZDBusConnectionFlagsZAUTHENTICATION_CLIENTZMESSAGE_BUS_CONNECTIONZ
DEFAULT_FLAGS� staticmethodr r r r r r r r
. s
c @ sX e Zd ZdZeedd� ��Zdd� Zedd� �Zedd � �Z ed
d � �Z
ed d
� �Z dS )�AbstractMessageBusa
Abstract representation of a message bus.
The property connection represents a connection to the bus. You can
register a service name with register_service, or publish an object
with publish_object and get a proxy of a remote object with get_proxy.
c C s dS )zThe DBus connection.Nr )�selfr r r �
connection\ s z AbstractMessageBus.connectionc C s< y
| j dk S tk
r6 } ztjd|� dS d}~X nX dS )znCheck if the connection is set up.
:return: True if the connection is set up otherwise False
Nz Connection can't be created:
%sF)r! � Exceptionr Zwarning)r �er r r �check_connectionb s
z#AbstractMessageBus.check_connectionc K s dS )z�Returns a proxy of a remote DBus object.
:param service_name: a DBus name of a service
:param object_path: a DBus path an object
:return: a proxy object
Nr )r � service_name� object_path�kwargsr r r � get_proxym s z AbstractMessageBus.get_proxyc K s dS )a Register a service on DBus.
A service can be registered by requesting its name on DBus.
This method should be called only after all of the required
objects of the service are published on DBus.
:param service_name: a DBus name of a service
Nr )r r% r' r r r �register_servicew s
z#AbstractMessageBus.register_servicec K s dS )z�Publish an object on DBus.
:param object_path: a DBus path of an object
:param obj: an instance of @dbus_interface or @dbus_class
Nr )r r& �objr' r r r �publish_object� s z!AbstractMessageBus.publish_objectc C s dS )zDisconnect from DBus.Nr )r r r r �
disconnect� s z AbstractMessageBus.disconnectN) r r r r �propertyr r! r$ r( r) r+ r, r r r r r T s
r )� metaclassc s~ e Zd ZdZdef� fdd� Zedd� �Zedd� �Z ed d
� �Z
e fd d �Z d
d� Z
efdd�Zefdd�Zdd� Z� ZS )r z/Representation of a message bus based on D-Bus.Nc s: t � j� || _|pt� | _d| _d| _g | _t� | _ dS )z�Create a new message bus.
:param error_mapper: a DBus error mapper
:param provider: a provider of DBus connections
N)
�super�__init__� _providerr �
_error_mapper� _connection�_proxy�_registrations�set�_requested_names)r � error_mapperZprovider)� __class__r r r0 � s
zMessageBus.__init__c C s | j s| j� | _ | j S )zThe DBus connection.)r3 �_get_connection)r r r r r! � s
zMessageBus.connectionc C s dS )zReturn a DBus connection.Nr )r r r r r: � s zMessageBus._get_connectionc C s | j s| jdd�| _ | j S )zThe proxy of DBus.zorg.freedesktop.DBusz/org/freedesktop/DBus)r4 r( )r r r r �proxy� s
zMessageBus.proxyc K s$ | j |� || ||fd| ji|��S )aD Returns a proxy of a remote DBus object.
:param service_name: a DBus name of a service
:param object_path: a DBus path an object
:param proxy_factory: a factory of a DBus object proxy
:param proxy_arguments: additional arguments for the proxy factory
:return: a proxy object
r8 )�_check_service_accessr2 )r r% r& Z
proxy_factoryZproxy_argumentsr r r r( � s
zMessageBus.get_proxyc C s4 || j krdS tj� tj� k r"dS tdj|���dS )a� Check if we can access a DBus service.
FIXME: This is a temporary check that should be later removed.
This is useful during the transition of the Anaconda code from
UI to DBus modules. This check prevents a deadlock in case that
a DBus module tries to access a service, that it provides, from
the main thread.
:param service_name: a DBus name of a service
:raises: RuntimeError if the service cannot be accessed
Nz4Can't access DBus service '{}' from the main thread.)r7 � threadingZcurrent_threadZ main_thread� RuntimeError�format)r r% r r r r< � s
z MessageBus._check_service_accessc sV t jd�� � jj�|�}|tkr0tdj|���� jj�� � j j
� �fdd�� dS )z�Register a service on DBus.
:param service_name: a DBus name of a service
:param flags: the flags argument of the RequestName DBus method
z Registering a service name %s.zName request has failed: {}c s � j j��S )N)r; Z ReleaseNamer )r r% r r �<lambda>� s z-MessageBus.register_service.<locals>.<lambda>N) r �debugr; Z RequestNamer �ConnectionErrorr? r7 �addr5 �append)r r% r �resultr )r r% r r) � s zMessageBus.register_servicec C s8 t jd|� || ||| jd�}|j� | jj|j� dS )z�Publish an object on DBus.
:param object_path: a DBus path of an object
:param obj: an instance of @dbus_interface or @dbus_class
:param server_factory: a factory of a DBus server object handler
zPublishing an object at %s.)r8 N)r rA r2 Zconnect_objectr5 rD Zdisconnect_object)r r& r* Zserver_factoryZobject_handlerr r r r+ � s
zMessageBus.publish_objectc C s8 t jd� x| jr$| jj� }|� q W d| _t� | _dS )zDisconnect from DBus.zDisconnecting from the bus.N)r rA r5 �popr3 r6 r7 )r �callbackr r r r, s
zMessageBus.disconnect)r r r r r
r0 r- r! r r: r; r r( r< r r) r r+ r, �
__classcell__r r )r9 r r � s c @ s e Zd ZdZdd� ZdS )r z*Representation of a system bus connection.c C s t jd� | jj� S )z Get a system DBus connection.z Connecting to the system bus.)r r r1 r )r r r r r: ! s
z SystemMessageBus._get_connectionN)r r r r r: r r r r r s c @ s e Zd ZdZdd� ZdS )r
z+Representation of a session bus connection.c C s t jd� | jj� S )z Get a session DBus connection.z Connecting to the session bus.)r r r1 r )r r r r r: * s
z!SessionMessageBus._get_connectionN)r r r r r: r r r r r
' s c s4 e Zd ZdZ� fdd�Zedd� �Zdd� Z� ZS )r z9Representation of a connection for the specified address.c s t � j||� || _dS )z\Create a new representation of a connection.
:param address: a bus address
N)r/ r0 �_address)r �address�argsr' )r9 r r r0 3 s z AddressedMessageBus.__init__c C s | j S )zThe bus address.)rI )r r r r rJ ; s zAddressedMessageBus.addressc C s t jd| j� | jj| j�S )z3Get a connection to a bus at the specified address.zConnecting to a bus at %s.)r r rI r1 r )r r r r r: @ s z#AddressedMessageBus._get_connection) r r r r r0 r- rJ r: rH r r )r9 r r 0 s ) Zloggingr= �abcr r Zdasbus.constantsr r Zdasbus.client.proxyr Z dasbus.errorr Zdasbus.server.handlerr ZgiZrequire_versionZ
gi.repositoryr Z getLoggerr r �__all__�objectr
r r r r
r r r r r �<module> s,
&>