Mini Shell
3
�2[}! � @ sb d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlm Z e j
� Z G dd� d�Z dS ) z:Provides the default implementation for bookmarks in Orca.� N� )�messages)�
orca_state)�settings_managerc @ s� e Zd ZdZdd� Zdd� Zdd� Zd#d d
�Zd d � Zd
d� Z dd� Z
dd� Z dd� Z d$dd�Z
d%dd�Zdd� Zdd � Zd d � Zd d � Zd&d!d"�ZdS )'� Bookmarksz&Represents a default bookmark handler.c C s* || _ i | _g | _g | _| j� d | _d S )N)�_script�
_bookmarks�_saveObservers�_loadObservers�_loadBookmarks�_currentbookmarkindex)�selfZscript� r � /usr/lib/python3.6/bookmarks.py�__init__# s zBookmarks.__init__c C s | j j|� d S )N)r �append)r
�observerr r r �addSaveObserver+ s zBookmarks.addSaveObserverc C s | j j|� d S )N)r
r )r
r r r r �addLoadObserver. s zBookmarks.addLoadObserverNc C s� |p|j }yB| jj� }| j| }|j|d |d |d |d � || j|< W n tk
rj | jj� dS X || j_| jj|� || _ dS )z2 Go to the bookmark indexed by inputEvent.hw_code �line�zone�word�charN)
�hw_coder �getFlatReviewContextr �
setCurrent�KeyError�
systemBeepZflatReviewContextZreviewCurrentItemr )r
�
inputEvent�index�context� context_infor r r � goToBookmark1 s
zBookmarks.goToBookmarkc C s. | j j� }| j|�| j|j< | j jtj� dS )z9 Add an in-page accessible object bookmark for this key. N)r r �_contextToBookmarkr r �presentMessager ZBOOKMARK_ENTERED)r
r r r r r � addBookmarkF s
zBookmarks.addBookmarkc C s\ y | j | j� | jjtj� W n" tk
r@ | jjtj� Y nX x| jD ]
}|� qJW dS )z% Save the bookmarks for this script. N) �saveBookmarksToDiskr r r$ r ZBOOKMARKS_SAVED�IOErrorZBOOKMARKS_SAVED_FAILUREr )r
r �or r r �
saveBookmarksL s zBookmarks.saveBookmarksc
C s� t | jj� �}t|�dkr(| jj� dS t|�dks>| jdkrT| jd|d d� dS y&|j| j�}| jd||d d� W n, t t
t fk
r� | jd|d d� Y nX dS )zu Go to the next bookmark location. If no bookmark has yet to be
selected, the first bookmark will be used. r Nr )r ) �sortedr �keys�lenr r r r" r �
ValueErrorr �
IndexError)r
r �hwkeysr r r r �goToNextBookmarkX s
zBookmarks.goToNextBookmarkc
C s� t | jj� �}t|�dkr(| jj� d S t|�dks>| jd krT| jd |d d� d S y&|j| j�}| jd ||d d� W n, t t
t fk
r� | jd |d d� Y nX d S )Nr r )r ) r* r r+ r, r r r r" r r- r r. )r
r r/ r r r r �goToPrevBookmarkp s
zBookmarks.goToPrevBookmarkc C s( | j � p
i | _x| jD ]
}|� qW dS )z# Load this scripts saved bookmarks.N)�readBookmarksFromDiskr r
)r
r( r r r r � s zBookmarks._loadBookmarksc
C s� |p| j jjd�d }tj� }|s&dS tjj|d�}y0ttjj|d| �d�}t j
|j �}|j � |S t
ttfk
r~ dS X dS )zc Read saved bookmarks from disk. Currently an unpickled object
that represents a bookmark � r N� bookmarksz%s.pkl�r)r �name�split�_settingsManager� getPrefsDir�os�path�join�open�pickle�load�buffer�closer' �EOFError�OSError)r
�filename�orcaDir�orcaBookmarksDirZ inputFiler4 r r r r2 � s
z Bookmarks.readBookmarksFromDiskc
C s� |p| j jjd�d }tj� }tjj|d�}ytj|� W n t k
rX tj
|� Y nX t tjj|d| �dtj �}t
j||j� |j� dS )zN Write bookmarks to disk. bookmarksObj must be a pickleable
object. r3 r r4 z%s.pkl�wN)r r6 r7 r8 r9 r: r; r< �statrC �mkdirr= �O_CREATr> �dumpr@ rA )r
Z bookmarksObjrD rE rF �outputr r r r& � s
z Bookmarks.saveBookmarksToDiskc C s0 i }|j |d< |j|d< |j|d< |j|d< |S )z6Converts a flat_review.Context object into a bookmark.r r r r )Z zoneIndexZ charIndexZ wordIndexZ lineIndex)r
r r! r r r r# � s
z Bookmarks._contextToBookmarkc C s. | j j� }|j|d |d |d |d � |S )z6Converts a bookmark into a flat_review.Context object.r r r r )r r r )r
Zbookmarkr r r r �_bookmarkToContext� s
z Bookmarks._bookmarkToContextc C s6 | j jj� }|r.tjj|�}dj|dd� �S dS dS )zpReturns the URI key for a given page as a URI stripped of
parameters?query#fragment as seen in urlparse.� r � N)r � utilitiesZdocumentFrameURI�urllib�parseZurlparser< )r
ZuriZ
parsed_urir r r � getURIKey� s
zBookmarks.getURIKeyc C s@ | j jj� }x.|D ]&}y || }W q tk
r6 dS X qW |S )zPReturn the object with the given path (relative to the
document frame). N)r rP Z
documentFramer. )r
r; Z returnobjZ childnumberr r r � pathToObj� s
zBookmarks.pathToObjc C s� |s| j jj� \}}|s g S | j jj|�r.g S g }|j|j� � |j}x6|r|| j jj|�rf|j� |S |j|j� � |j}qHW g S )z�Given an object, return it's path from the root accessible. If obj
is not provided, the current caret context is used. )r rP ZgetCaretContextZ
isDocumentr ZgetIndexInParent�parent�reverse)r
Z start_objZcharacterOffsetr; �pr r r �
_objToPath� s
zBookmarks._objToPath)N)N)N)N)�__name__�
__module__� __qualname__�__doc__r r r r" r% r) r0 r1 r r2 r& r# rM rS rT rX r r r r r ! s"
r )
r\ r> Zpyatspir: Z urllib.parserQ rN r r r Z
getManagerr8 r r r r r �<module> s