Mini Shell
3
�2[ 4 � @ sl d Z dZdZdZdZddlZddlZddlmZ dd lm Z dd
l
m Z m Z G d d � d �Z
e
� Zd
d� ZdS )z$Id$z
$Revision$z$Date$z Copyright (c) 2011. Orca Team.ZLGPL� N� )�debug)�
orca_state)�apps�toolkitsc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z d dd�Z
d dd�Z dd� Z d!dd�Z
d"dd�Zdd � Zd d � ZdS )#�
ScriptManagerc C s� t jt jdd� i | _i | _i | _tj| _t j| _
d | _ ddddg| _ ddd d
d d d
d�| _
| jd d� tjjd�| _d| _t jt jdd� d S )Nz SCRIPT MANAGER: InitializingTz orca-scriptsz orca.scriptszorca.scripts.appszorca.scripts.toolkitsZMozillaZ ThunderbirdZempathyZ gcalctoolZmetacityZBansheeznotification-daemon)ZFirefoxZIcedovez empathy-chatzgnome-calculatorZmarcoZNereidzmate-notification-daemon�__init__r FzSCRIPT MANAGER: Initialized)r �println�
LEVEL_INFO�
appScripts�toolkitScripts�
customScriptsr �__all__� _appModulesr �_toolkitModules�_defaultScript�_scriptPackages� _appNames�setActiveScript�pyatspiZRegistryZ
getDesktop�_desktop�_active)�self� r �$/usr/lib/python3.6/script_manager.pyr $ s,
zScriptManager.__init__c C sN t jt jdd� | jd�| _| jj� | j| jd� d| _t jt jdd� dS )z-Called when this script manager is activated.zSCRIPT MANAGER: ActivatingTN�activatezSCRIPT MANAGER: Activated)r r r
� getScriptr ZregisterEventListenersr r )r r r r r @ s
zScriptManager.activatec C s^ t jt jdd� | jr | jj� d| _| jdd� i | _i | _i | _d| _ t jt jdd� dS )z/Called when this script manager is deactivated.zSCRIPT MANAGER: DectivatingTN�
deactivateFzSCRIPT MANAGER: Deactivated)
r r r
r ZderegisterEventListenersr r r r
r )r r r r r J s
zScriptManager.deactivatec s� y|dk o|j dk}W n4 ttfk
rJ d}d| }tjtj|d� Y nX |sTdS |j jd�d � t| jj � �}� fdd �|D �}|r�| jj
|d �� n6x4| j | j fD ]$}� fd
d �|D �}|r�|d � P q�W d |j � f }tjtj|d� � S ) zAReturns the module name of the script to use for application app.N� FzERROR: %s no longer existsT�.r c s g | ]}|j � � j � kr|�qS r )�lower)�.0�n)�namer r �
<listcomp>j s z/ScriptManager.getModuleName.<locals>.<listcomp>c s g | ]}|j � � j � kr|�qS r )r )r! r" )r# r r r$ o s z SCRIPT MANAGER: mapped %s to %s)
r# � LookupError� RuntimeErrorr r r
�split�listr �keys�getr r )r �appZappAndNameExist�msgZaltNames�namesZnameListr )r# r �
getModuleNameX s* zScriptManager.getModuleNamec C sP d}|rLy |j � }W n ttfk
r, Y n X tdd� |D ��}|jdd�}|S )z4Returns the name of the toolkit associated with obj.r c S s g | ]}|j d d��qS )�:r )r' )r! �attrr r r r$ � s z3ScriptManager._toolkitForObject.<locals>.<listcomp>Ztoolkit)Z
getAttributesr% r& �dictr* )r �objr# Z
attributes�attrsr r r �_toolkitForObjectx s z ScriptManager._toolkitForObjectc C s* y |j � }W n
dS |tjkr&dS dS )Nr Zterminal)�getRoler Z
ROLE_TERMINAL)r r2 �roler r r �_scriptForRole� s
z ScriptManager._scriptForRolec C s� |o|s dS d}x�| j D ]�}dj||f�}ytj|�}W n0 tk
rN wY n tk
rh tj� Y nX tjtj d| d� y&t
|d�r�|j |�}n
|j |�}P W q tj
tj � d| }tjtj |d� Y qX qW |S )zkAttempts to locate and load the named module. If successful, returns
a script based on this module.Nr zSCRIPT MANAGER: Found %sTr zERROR: Could not load %s)r �join� importlib�
import_module� ImportError�OSErrorr ZexamineProcessesr r
�hasattrr �Script�printException)r r+ r# �script�package�
moduleName�moduler, r r r �_newNamedScript� s,
z ScriptManager._newNamedScriptNc C s� | j |�}| j||�}|r |S | j|�}| j||�}|r<|S yt|dd�}W n0 ttfk
r| d| }tjtj|d� Y nX |r�|r�| j||�}|s�| j |�}d}tjtj|d� |S )z8For the given application, create a new script instance.� toolkitNameNz,ERROR: Exception getting toolkitName for: %sTz&SCRIPT MANAGER: Default script created)
r. rD r4 �getattrr% r& r r r
�getDefaultScript)r r+ r2 rB r@ �
objToolkitrE r, r r r �
_createScript� s(
zScriptManager._createScriptc C s6 | r| j r| j S ddlm} |j|�}|s2|| _ |S )Nr )�default)r �scriptsrJ r> )r r+ rJ r@ r r r rG � s
z ScriptManager.getDefaultScriptc
C s� | j s
|S y|j| jk}W n d}Y nX |r4|S d|j }tjtj|d� | j|j�}|rzd| }tjtj|d� |S d|j }tjtj|d� |S )NFz,WARNING: %s is not in the registry's desktopTz2SCRIPT MANAGER: Script for app replicant found: %sz2WARNING: Failed to get a replacement script for %s)r r+ r r r r
�_getScriptForAppReplicant)r r@ Z appInDesktopr, � newScriptr r r �sanityCheckScript� s$
z ScriptManager.sanityCheckScriptFc C s� d}d}d}| j |�}|rT| jj|i �}|j|�}|sJ| j||�}|||< || j|< | j|�} | r�| jj|i �}
|
j| �}|s�| j||�}||
| < |
| j|< y>|s�| j� }n,|| jkr�| j| }n| j|d�}|| j|< W n4 d} t j
t j � t j t j
| d� | j� }Y nX |�r|S y |j� } W n d}
Y nX | tjtjgk}
|�rl|
�rlt|j|j� �rl|S |�r�|�r�| j|�}|S )z�Get a script for an app (and make it if necessary). This is used
instead of a simple calls to Script's constructor.
Arguments:
- app: the Python app
Returns an instance of a Script.
Nz&WARNING: Exception getting app script.TF)r7 r
r* rD r4 r rI rG r r r? Z LEVEL_ALLr Z
LEVEL_WARNINGr5 r Z
ROLE_FRAMEZROLE_STATUS_BAR�
issubclass� __class__rN )r r+ r2 Z sanityCheck� customScript� appScript�
toolkitScriptZroleNamer
rH r r, r6 ZforceAppScriptr r r r � sX
zScriptManager.getScriptc C sV t j|krdS t jr t jj� |t _|s,dS |j� d|j|f }tjtj|d� dS )znSet the new active script.
Arguments:
- newScript: the new script to be made active.
Nz5SCRIPT MANAGER: Setting active script: %s (reason=%s)T)r Z activeScriptr r r# r r r
)r rM �reasonr, r r r r 6 s
z ScriptManager.setActiveScriptc C sh | j s
d S dd� }||�}|dkr&d S | jj� }x2|D ]*\}}||kr6||�|kr6|| jkr6|S q6W d S )Nc S s y| j � S dS d S )Nr ���)Zget_process_id)r+ r r r �_pidP s z5ScriptManager._getScriptForAppReplicant.<locals>._pidr rU )r r �itemsr )r r+ rV �pidrW �ar@ r r r rL L s
z'ScriptManager._getScriptForAppReplicantc
sb t � jj� �}y� fdd�|D �}W n tjtj� dS �x |D �]}d| }tjtj|d� � jj|�}� j |�}|r�d| }tjtj|d� |j
� }x<|j � D ]0\}}d||f }tjtj|d� t |||� q�W ~y� j
j|�} W n tk
�r Y nX x| j� D ]}
~
�qW y� jj|�} W n tk
�r@ Y nX x| j� D ]} ~ �qLW ~qDW dS )zqCompares the list of known scripts to the list of known apps,
deleting any scripts as necessary.
c s" g | ]}|d k r|� j kr|�qS )N)r )r! rY )r r r r$ h s z0ScriptManager.reclaimScripts.<locals>.<listcomp>Nz5SCRIPT MANAGER: %s is no longer in registry's desktopTz2SCRIPT MANAGER: Script for app replicant found: %sz SCRIPT MANAGER: Setting %s to %s)r( r r) r r? Z LEVEL_FINESTr r
�poprL ZgetTransferableAttributesrW �setattrr �KeyError�valuesr
)
r ZappListr+ r, rR rM r3 r0 �valuer rS r
rQ r )r r �reclaimScriptsa sB
z ScriptManager.reclaimScripts)N)N)NF)N)�__name__�
__module__� __qualname__r r r r. r4 r7 rD rI rG rN r r rL r_ r r r r r " s
C
r c C s t S )N)�_managerr r r r �
getManager� s rd )Z__id__� __version__Z__date__Z
__copyright__Z __license__r9 r r r r rK r r r rc rd r r r r �<module> s q