��Yf= � @ s� d Z d d l Z d d l m Z m Z m Z d d l j Z d d � Z Gd d � d � Z
d d � Z d
d � Z d d
� Z
e d k r� d d l Z e j d d d d d �d S)z'Define SearchEngine for search dialogs.� N)� StringVar�
BooleanVar�TclErrorc C s% t | d � s t | � | _ | j S)z�Return the singleton SearchEngine instance for the process.
The single SearchEngine saves settings between dialog instances.
If there is not a SearchEngine already, make one.
�
_searchengine)�hasattr� SearchEnginer )�root� r �9/opt/alt/python35/lib64/python3.5/idlelib/SearchEngine.py�get s r c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d# d d � Z d d d d � Z d d d � Z d d! d" � Z d S)$r z<Handles searching a text widget for Find, Replace, and Grep.c C sy | | _ t | d � | _ t | d � | _ t | d � | _ t | d � | _ t | d � | _ t | d � | _ d S)z�Initialize Variables that save search state.
The dialogs bind these to the UI elements present in the dialogs.
� FTN) r r �patvarr �revar�casevar�wordvar�wrapvar�backvar)�selfr r r r
�__init__ s zSearchEngine.__init__c C s
| j j � S)N)r
r )r r r r
�getpat# s zSearchEngine.getpatc C s | j j | � d S)N)r
�set)r �patr r r
�setpat& s zSearchEngine.setpatc C s
| j j � S)N)r r )r r r r
�isre) s zSearchEngine.isrec C s
| j j � S)N)r r )r r r r
�iscase, s zSearchEngine.iscasec C s
| j j � S)N)r r )r r r r
�isword/ s zSearchEngine.iswordc C s
| j j � S)N)r r )r r r r
�iswrap2 s zSearchEngine.iswrapc C s
| j j � S)N)r r )r r r r
�isback5 s zSearchEngine.isbackc C s, | j � r t j | � } | j | � d S)z!Set pattern after escaping if re.N)r �re�escaper )r r r r r
� setcookedpat: s zSearchEngine.setcookedpatc C sA | j � } | j � s' t j | � } | j � r= d | } | S)Nz\b%s\b)r r r r r )r r r r r
� getcookedpatA s
zSearchEngine.getcookedpatc C s� | j � } | s& | j | d � d S| j � } d } | j � sQ | t j B} y t j | | � } Wnt t j k
r� } zQ | j } | d } t | � d k r� | d n d } | j | | | � d SWYd d } ~ Xn X| S)z&Return compiled cooked search pattern.zEmpty regular expressionNr � � ���)
r � report_errorr! r r �
IGNORECASE�compile�error�args�len)r r �flags�progZwhatr) �msg�colr r r
�getprogI s"
"zSearchEngine.getprogr# c C sg d t | � } | r* | d t | � } | d k rJ | d t | � } t j d | d | j �d S)NzError: z
Pattern: r z
Offset: zRegular expression errorZmaster)�str� tkMessageBoxZ showerrorr )r r r- r. r r r
r% ] s zSearchEngine.report_errorNr c C s� | s | j � } | s d S| j j � } t | � \ } } | j � r� | rX | } n | } t | � \ } } | j | | | | | | � }
nE | r� | } n | } t | � \ } } | j | | | | | | � }
|
S)a� Return (lineno, matchobj) or None for forward/backward search.
This function calls the right function with the right arguments.
It directly return the result of that call.
Text is a text widget. Prog is a precompiled pattern.
The ok parameter is a bit complicated as it has two effects.
If there is a selection, the search begin at either end,
depending on the direction setting and ok, with ok meaning that
the search starts with the selection. Otherwise, search begins
at the insert mark.
To aid progress, the search functions do not return an empty
match at the starting position unless ok is True.
N)r/ r r �
get_selectionr � get_line_col�search_backward�search_forward) r �textr, �ok�wrap�first�last�start�liner. �resr r r
� search_textg s$ ! zSearchEngine.search_textc C s� d } | } | j d | d | d � } x� | r� | j | d d � | � }
|
rw | sm |
j � | k rw | |
f S| d } | r� | | k r� Pd } d } | j d | d | d � } | r- | r- d } d } d } | j d d � } q- Wd S)Nr z%d.0r# z1.0z2.0r$ )r �search�end) r r6 r, r< r. r8 r7 �wrapped� startline�chars�mr r r
r5 � s(
zSearchEngine.search_forwardc C s d } | } | j d | d | d � } x� t | | d d � | � }
|
rq | sg |
j � | k rq | |
f S| d } | r� | | k r� Pd } | d k r� | s� Pd } d } | j d � } t t | j d � � \ } } | j d | d | d � } t | � d } q- Wd S)Nr z%d.0r# zend-1c�.r$ )r �search_reverser; �index�map�int�splitr* ) r r6 r, r< r. r8 r7 rA rB rC rD �posr r r
r4 � s,
z SearchEngine.search_backwardr$ )�__name__�
__module__� __qualname__�__doc__r r r r r r r r r r! r/ r% r> r5 r4 r r r r
r s
(r c C s� | j | � } | s d Sd } | j � \ } } xc | | k r� | | k r� | } | | k rh | d } | j | | � } | s� P| j � \ } } q4 W| S)a& Search backwards and return an re match object or None.
This is done by searching forwards until there is no match.
Prog: compiled re object with a search method returning a match.
Chars: line of text, without \n.
Col: stop index for the search; the limit for match.end().
Nr# )r? �span)r, rC r. rD �found�i�jr r r
rF � s
rF c C sl y" | j d � } | j d � } Wn t k
r@ d } } Yn X| sV | j d � } | sb | } | | f S)zFReturn tuple of 'line.col' indexes from selection or insert mark.
z sel.firstzsel.lastN�insert)rG r )r6 r9 r: r r r
r2 � s
r2 c C s( t t | j d � � \ } } | | f S)z8Return (line, col) tuple of ints from 'line.col' string.rE )rH rI rJ )rG r< r. r r r
r3 � s r3 �__main__z#idlelib.idle_test.test_searchengine� verbosityr" �exitF)rO r Ztkinterr r r Ztkinter.messageboxZ
messageboxr1 r r rF r2 r3 rL Zunittest�mainr r r r
�<module> s �