Mini Shell
3
�2[�S � @ s� d dl Z d dlZd dlZd dlZd dlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z dd lm Z dd
lm
Z
e
j� ZdZd ZdZd Zd Zd
ZdZG dd� d�ZG dd� d�ZdS )� N)�GLib� )�cmdnames)�chnames)� keybindings)�messages)� input_event)�
orca_state)�settings_manager� � �- � c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�
PriorityQueuez� This class represents a thread **UNSAFE** priority queue where priority
is determined by the given integer priority. The entries are also
maintained in chronological order.
TODO: experiment with Queue.Queue to make thread safe
c C s
g | _ d S )N)�queue)�self� r �!/usr/lib/python3.6/liveregions.py�__init__& s zPriorityQueue.__init__c C s t j| j|tj� ||f� dS )zS Add a new element to the queue according to 1) priority and
2) timestamp. N)�bisectZ insort_leftr �time)r �data�priority�objr r r �enqueue) s zPriorityQueue.enqueuec C s | j jd�S )z2get the highest priority element from the queue. r )r �pop)r r r r �dequeue. s zPriorityQueue.dequeuec C s
g | _ dS )z Clear the queue N)r )r r r r �clear2 s zPriorityQueue.clearc s* t j � � � fdd�}tt|| j��| _dS )zL Purge items from the queue that are older than the keepalive
time c s | d t � kS )Nr )�MSG_KEEPALIVE_TIME)�item)� currenttimer r �<lambda>: s z0PriorityQueue.purgeByKeepAlive.<locals>.<lambda>N)r �list�filterr )r �myfilterr )r r �purgeByKeepAlive6 s z PriorityQueue.purgeByKeepAlivec s" � fdd�}t t|| j��| _dS )ze Purge items from the queue that have a lower than or equal priority
than the given argument c s | d � kS )Nr r )r )r r r r! @ s z/PriorityQueue.purgeByPriority.<locals>.<lambda>N)r" r# r )r r r$ r )r r �purgeByPriority= s z PriorityQueue.purgeByPriorityc C s
t | j�S )z Return the length of the queue )�lenr )r r r r �__len__C s zPriorityQueue.__len__N) �__name__�
__module__� __qualname__�__doc__r r r r r% r& r( r r r r r s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd4d*d+�Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd)S )5�LiveRegionManagerc C s~ || _ t� | _| j� | _| j� | _| j| j| j | j
g| _ g | _ d | _
d | _d | _d| _| j� |jj| j� |jj| j� d S )NT)�_scriptr � msg_queue�_getInputEventHandlers�inputEventHandlers�_getKeyBindings� keyBindings�advancePoliteness�setLivePolitenessOff�toggleMonitoring�reviewLiveAnnouncementZ functions� msg_cache�_politenessOverrides�_restoreOverrides� lastliveobj�
monitoring�bookmarkLoadHandler� bookmarksZaddSaveObserver�bookmarkSaveHandlerZaddLoadObserver)r �scriptr r r r I s
zLiveRegionManager.__init__c C sX i }t j| jtj�|d< t j| jtj�|d< t j| jtj�|d< t j| j tj
�|d< |S )N�advanceLivePolitenessr5 �monitorLiveRegionsr7 ) r ZInputEventHandlerr4 r Z LIVE_REGIONS_ADVANCE_POLITENESSr5 Z LIVE_REGIONS_SET_POLITENESS_OFFr6 ZLIVE_REGIONS_MONITORr7 ZLIVE_REGIONS_REVIEW)r Zhandlersr r r r0 r s z(LiveRegionManager._getInputEventHandlersc C s� t j� }|jt jdt jt j| jjd��� |jt jdt jt j| jjd��� |jt jdt jt j | jjd��� x.dD ]&}|jt j|t jt j
| jjd��� qtW |S )NZ backslashrA r5 rB �F1�F2�F3�F4�F5�F6�F7�F8�F9r7 ) rC rD rE rF rG rH rI rJ rK ) r Z KeyBindings�addZ
KeyBindingZdefaultModifierMaskZNO_MODIFIER_MASKr1 �getZSHIFT_MODIFIER_MASKZORCA_SHIFT_MODIFIER_MASKZORCA_MODIFIER_MASK)r r3 �keyr r r r2 � s6
z!LiveRegionManager._getKeyBindingsc C sN i }| j jj� }x2| jj� D ]$\}}|d |ks8|tkr |||< q W || _d S )Nr )r. r> � getURIKeyr9 �items� LIVE_NONE)r Z
newpoliteness�
currenturirN �valuer r r �reset� s zLiveRegionManager.resetc C s | j jj| jdd� dS )zBookmark save callback�
politeness)�filenameN)r. r> ZsaveBookmarksToDiskr9 )r r r r r? � s z%LiveRegionManager.bookmarkSaveHandlerc C s | j jjdd�pi | _dS )zBookmark load callbackrU )rV N)r. r> ZreadBookmarksFromDiskr9 )r r r r r= � s z%LiveRegionManager.bookmarkLoadHandlerc C s� | j |j�}|tkrdS |tkr,| js`dS n4|tkr6n*|tkrL| jjt� n|t kr`| jjt� | j
|�}|r�t | j�dkr�t j
d| j� | jj|||j� dS )z Main live region event handlerNr �d )� _getLiveType�source�LIVE_OFFrQ r< � LIVE_POLITE�LIVE_ASSERTIVEr/ r& � LIVE_RUDE� _getMessager' r Z timeout_add� pumpMessagesr )r �eventrU �messager r r � handleEvent� s"
z LiveRegionManager.handleEventc C s� t | j�dkrp| jj� | jj� \}}}}|d |d krD|d }n|d |d }| jj|� || _| j|� | js�| jj� t | j�dkS )z� Main gobject callback for live region support. Handles both
purging the message queue and outputting any queued messages that
were queued up in the handleEvent() method.
r �labels�content) r' r/ r% r r. �presentMessager; �
_cacheMessager< )r rU Z timestampra r �uttsr r r r_ � s
z LiveRegionManager.pumpMessagesc C sN g }| j jj� }x8| jD ].\}}||krt|t�r|j| j jj|�� qW |S )zXReturn the live objects that are registered and have a politeness
of LIVE_NONE. )r. r> rO r9 �
isinstance�tuple�appendZ pathToObj)r ZretvalrR �uri�objectidr r r �getLiveNoneObjects� s z$LiveRegionManager.getLiveNoneObjectsc C s t jd�s | jjtj� dS tj}| j|�}| jj j
� }y| j ||f }W n t k
rh | j
|�}Y nX |tksz|tkr�t| j ||f< | jjtj� nr|tkr�t| j ||f< | jjtj� nL|tkr�t| j ||f< | jjtj� n&|tk�r
t| j ||f< | jjtj� dS )z0Advance the politeness level of the given object�inferLiveRegionsN)�_settingsManager�
getSettingr. re r �LIVE_REGIONS_OFFr Z locusOfFocus� _getObjectIdr> rO r9 �KeyError�_liveStringToTyperZ rQ r[ ZLIVE_REGIONS_LEVEL_POLITEr\ Z LIVE_REGIONS_LEVEL_ASSERTIVEr] ZLIVE_REGIONS_LEVEL_RUDEZLIVE_REGIONS_LEVEL_OFF)r r@ �
inputEventr rl rk Z cur_priorityr r r r4 s,
z#LiveRegionManager.advancePolitenessc C s6 | j r2| jjj| j d� | jj| jjj| j d�� dS )z_Move the caret to the last announced live region and speak the
contents of that objectr N)r; r. � utilitiesZsetCaretPositionZ
speakContentsZgetObjectContentsAtOffset)r r r r �goLastLiveRegion( s z"LiveRegionManager.goLastLiveRegionc C sd t |jdd� �}tjd�s.| jjtj� dS |t| j �krL| jjtj
� n| jj| j | � dS )z%Speak the given number cached messager Nrn ) �intZ event_stringro rp r. re r rq r' r8 ZLIVE_REGIONS_NO_MESSAGE)r r@ ru Zmsgnumr r r r7 0 s
z(LiveRegionManager.reviewLiveAnnouncementc C s� t jd�s | jjtj� dS | jjj� }| jjj � }| j
r�| jjtj � | j j
� tj| j�| _x | jj� D ]}t| j|< qlW tj|| j�}x$|D ] }| j|�}t| j||f< q�W d| _
n6x | jj� D ]\} }
|
| j| < q�W | jjtj� d| _
dS )z]User toggle to set all live regions to LIVE_OFF or back to their
original politeness.rn NFT)ro rp r. re r rq rv �
documentFramer> rO r< ZLIVE_REGIONS_ALL_OFFr/ r �copyr9 r: �keysrZ �pyatspiZfindAllDescendants�matchLiveRegionrr rP ZLIVE_REGIONS_ALL_RESTORED) r r@ ru �docframerk �overrideZmatches�matchrl rN rS r r r r5 = s(
z&LiveRegionManager.setLivePolitenessOffc K s� | j |�}| jjj� }g }xr|j� D ]f}|j� }|tjkr$|jd�}y0|j � j
dd�} | j � |j j � krr|j
| � W q$ tk
r� Y q$X q$W y | j||f }
| j|
�} W n tk
r� d} Y nX |s�| dkr�|j
tj| � |S )zoUsed in conjunction with whereAmI to output description and
politeness of the given live region objectr r �none���)rr r. r> rO ZgetRelationSetZgetRelationTyper| ZRELATION_DESCRIBED_BYZ getTargetZ queryText�getText�strip� descriptionrj �NotImplementedr9 �_liveTypeToStringrs r ZLIVE_REGIONS_LEVEL) r r �argsrl rk �resultsZrelationZ relationtypeZ targetobjr� Z livepriorityZliveprioritystrr r r � generateLiveRegionDescriptionj s*
z/LiveRegionManager.generateLiveRegionDescriptionc C s | j |�}d|kS )z$Predicate used to find a live regionzcontainer-live)�_getAttrDictionary)r r �attrsr r r r} � s
z!LiveRegionManager.matchLiveRegionc C s | j |j�}d}d}|jjd�rR|jd�dkr@| jjj|j�}q�| jjj|j�}nB|jjd�r�|jd�dkrt|j}n | jjj |j�}|r�|j
dd �}|s�dS |j � }t |�dkr�t
j|�}|jjp�|jjj � }|r�||kr�|}|jd �d
k�r|| }| jj� | jj|� dS |g|gd �S )
z4Gets the message associated with a given live event.� zobject:children-changed:addzcontainer-atomic�truezobject:text-changed:insertr r NZchannelZnotify)rd rc r� )r� rY �type�
startswithrM r. rv Z
expandEOCsZany_dataZqueryNonEmptyTextr� r� r' r ZgetCharacterName�namer� ZpresentationInterruptre )r r` r� rd rc �textr� rg r r r r^ � s6
z LiveRegionManager._getMessagec C s | j j� d S )N)r/ r )r r r r �
flushMessages� s z LiveRegionManager.flushMessagesc C s* | j j|� t| j �tkr&| j jd� dS )z6Cache a message in our cache list of length CACHE_SIZEr N)r8 rj r' �
CACHE_SIZEr )r rg r r r rf � s z LiveRegionManager._cacheMessagec C s` | j |�}| jjj� }||f| jkr2| j||f S | j|�}|tkrX| jrX|| j||f< |S dS )z�Returns the live politeness setting for a given object. Also,
registers LIVE_NONE objects in politeness overrides when monitoring.N)rr r. r> rO r9 rt rQ r< )r r rl rk Zlivetyper r r rX � s
z LiveRegionManager._getLiveTypec C sD | j |�}|dkr | j|�S y|d S tk
r> | j|�S X dS )zRReturns the HTML 'id' or a path to the object is an HTML id is
unavailableN�id)r� �_getPathrs )r r r� r r r rr � s
z LiveRegionManager._getObjectIdNc C sp |p | j |�}yH|d dkr tS |d dkr0tS |d dkr@tS |d dkrPtS tS W n tk
rj tS X dS )z.Returns the politeness enum for a given objectzcontainer-live�off�polite� assertive�rudeN)r� rZ r[ r\ r] rQ rs )r r Z
attributesr� r r r rt � s z#LiveRegionManager._liveStringToTypec C sD |t kr dS |tkrdS |tkr$dS |tkr0dS |tkr<dS dS dS )z@Returns the politeness level as a string given a politeness enumr� r� r� r� r� �unknownN)rZ r[ r\ r] rQ )r rU r r r r� � s z#LiveRegionManager._liveTypeToStringc C s& yt dd� |j� D ��S i S d S )Nc S s g | ]}|j d d��qS )�:r )�split)�.0�attrr r r �
<listcomp> s z8LiveRegionManager._getAttrDictionary.<locals>.<listcomp>)�dictZ
getAttributes)r r r r r r� s z$LiveRegionManager._getAttrDictionaryc C sn | j jj� }g }xX|jdks$||kr4|j� t|�S y|j|j� � W n tk
r^ t �Y nX |j}qW dS )z` Returns, as a tuple of integers, the path from the given object
to the document frame.N)
r. rv ry �parent�reverseri rj ZgetIndexInParent� Exception� LookupError)r r r~ �pathr r r r�
s
zLiveRegionManager._getPathc C sL t jd�s&t jdd� | jjtj� n"t jdd� | j� | jjtj� d S )Nrn TF) ro rp Z
setSettingr. re r ZLIVE_REGIONS_MONITORING_ONr� ZLIVE_REGIONS_MONITORING_OFF)r r@ ru r r r r6 s
z"LiveRegionManager.toggleMonitoring)N) r) r* r+ r r0 r2 rT r? r= rb r_ rm r4 rw r7 r5 r� r} r^ r� rf rX rr rt r� r� r� r6 r r r r r- H s2 )"
"
-'.
r- r� )r rz r| r Z
gi.repositoryr r� r r r r r r r
Z
getManagerro rZ rQ r[ r\ r] r r� r r- r r r r �<module> s* )