�
U�*Wc @ s� d Z d d l m Z d d l Z d d l Z d d l Z y d d l m Z Wn! e k
rq d d l
m Z n Xd d d d d d
d g Z y
e Z
Wn e k
r� e e f Z
n Xe j d d
k Z e j d � Z e j d � Z e j d � Z e j Z d � Z d � Z e j d e d e � Z d � Z e j d � Z e j d � Z! d e f d � � YZ" d e" f d � � YZ# e d � Z$ e e% d � Z& d e' f d � � YZ( d � Z) i d d 6Z* e+ d k r�d d l Z e j, � n d S( s�
lxml-based doctest output comparison.
Note: normally, you should just import the `lxml.usedoctest` and
`lxml.html.usedoctest` modules from within a doctest, instead of this
one::
>>> import lxml.usedoctest # for XML output
>>> import lxml.html.usedoctest # for HTML output
To use this module directly, you must call ``lxmldoctest.install()``,
which will cause doctest to use this in all subsequent calls.
This changes the way output is checked and comparisons are made for
XML or HTML-like content.
XML or HTML content is noticed because the example starts with ``<``
(it's HTML if it starts with ``<html``). You can also use the
``PARSE_HTML`` and ``PARSE_XML`` flags to force parsing.
Some rough wildcard-like things are allowed. Whitespace is generally
ignored (except in attributes). In text (attributes and text in the
body) you can use ``...`` as a wildcard. In an example it also
matches any trailing tags in the element, though it does not match
leading tags. You may create a tag ``<any>`` or include an ``any``
attribute in the tag. An ``any`` tag matches any tag, while the
attribute matches any and all attributes.
When a match fails, the reformatted example and gotten text is
displayed (indented), and a rough diff-like output is given. Anything
marked with ``+`` is in the output but wasn't supposed to be, and
similarly ``-`` means its in the example but wasn't in the output.
You can disable parsing on one line with ``# doctest:+NOPARSE_MARKUP``
i����( t etreeN( t escapet
PARSE_HTMLt PARSE_XMLt NOPARSE_MARKUPt LXMLOutputCheckert LHTMLOutputCheckert installt temp_installi i c C s | d k r d S| j � Sd S( N( t Nonet strip( t v( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR
? s c C s t j d | � S( Nt ( t _norm_whitespace_ret sub( R ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt norm_whitespaceE s t recovert remove_blank_textc C s t j | t � S( N( R t
fromstringt _html_parser( t html( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt html_fromstringJ s s ^<[^>]+ (at|object) s [ \t\n][ \t\n]+c
B s� e Z d Z d
� Z d � Z d � Z d
� Z d � Z d � Z d � Z d � Z
e d � Z d d � Z
e d � Z d � Z d � Z d � Z d � Z d � Z e d � Z RS( t paramt imgt areat brt basefontt inputt baset metat linkt colc C s t j S( N( R t XML( t self( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt get_default_parserW s c C s� t | d d � } | d k r0 | j } | } n t j } | j | | | � } | sg | | | | | � Sy | | � } Wn t j k
r� t SXy | | � } Wn t j k
r� t SX| j | | � S( Nt _temp_override_self(
t getattrR t _temp_call_super_check_outputt
OutputCheckert check_outputt
get_parserR t XMLSyntaxErrort Falset compare_docs( R! t wantt gott optionflagst alt_selft super_methodt parsert want_doct got_doc( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR' Z s$ c C s� d } t | @r d St | @r' t } n| t | @r= t j } nf | j � j � j d � rv | j � j d � rv t } n- | j
| � r� | j
| � r� | j � } n | S( Ns <html( R R R R R R R R
t lowert
startswitht _looks_like_markupR" ( R! R, R- R. R1 ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR( o s
c C s) | j � } | j d � o( t j | � S( Nt <( R
R5 t _repr_ret search( R! t s( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR6 s c
C s� | j | j | j � s t S| j | j | j t � s; t S| j | j | j t � sZ t Sd | j k r� t | j j � � } t | j j � � } | | k r� t Sx8 | D]- } | j | j | | j | t � s� t Sq� Wn | j d k s� t
| � r�t | � } t | � } xs | s | r�| s.| r2t S| j d � } | j d � } | j
| | � sft S| r| j d k rPqqWn t S( Nt anys ...i ( t tag_comparet tagR* t text_comparet textt Truet tailt attribt sortedt keyst lent listt popR+ (
R! R, R- t want_keyst got_keyst keyt
want_childrent got_childrent
want_firstt got_first( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR+ � s6
# c C s� | p d } | p d } | rE t | � j � } t | � j � } n d t j | � } | j d d � } t j | | � r� t St Sd S( Nt s ^%s$s \.\.\.s .*( R R
t reR t replaceR9 R@ R* ( R! R, R- R
( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR> � s c C s� | d k r t St | t � s0 t | t � r: | | k S| pC d } | pO d } | j d � r� | j d � d | j d � d k S| | k Sd S( NR; RO s {...}t }i����( R@ t
isinstancet _basestringR5 t split( R! R, R- ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR< � s
$c
C s� | j } | j | | | � } g } | d k r� y | | � } Wn5 t j k
rw t j � d } | j d | � n Xy | | � } Wq� t j k
r� t j � d } | j d | � q� Xn | d k s� | rt j | | | | � }
| r
| j |
� d j
| � S|
Sn | t k } g } | j d � | j | j | | d � � | j d � | j | j | | d � � | j d � | j | j
| | | d � � d j
| � S( Ni s In example: %ss In actual output: %ss
s Expected:i s Got:s Diff:( R, R( R R R) t syst exc_infot appendR& t output_differencet joinR t
format_doct collect_diff(
R! t exampleR- R. R, R1 t errorsR2 t eR3 t valueR t
diff_parts( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRY � s<
c C s= | s
t S| j | j k r t S| j s5 t | � r9 t St S( N( R* R= t
empty_tagsR? RE R@ ( R! t elR ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt html_empty_tag� s RO c C s; g } t | � s� | j d | � | j | � | j | j | � � | j | | � s� t | j � r� | j | j | j � � n | j | j | � � n t | j � r� | j | j | j � � n | j d � d j | � S| j d | � | j | � | j | j | � � | j | | � s�| j d � t | j � r}| j d | � | j | j | j � � | j d � n x. | D]& } | j | j
| | | d � � q�W| j d | � | j | j | � � | j d � n t | j � r.| j d | � | j | j | j � � | j d � n d j | � S( NR s
RO i ( RE RX t
format_tagRd R
R? t format_textt format_end_tagRA RZ R[ ( R! t docR t indentt prefixt partsRc ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR[ � s@
$c C s2 | d k r d S| r% | j � } n t | d � S( NRO i ( R R
t html_escape( R! R? R
( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRf
s
c C s� g } t | t j � r d SxF t | j j � � D]/ \ } } | j d | | j | t � f � q2 W| sv d | j Sd | j d j
| � f S( Ns <!--s %s="%s"s <%s>s <%s %s>R ( RS R t CommentBaseRC RB t itemsRX Rf R* R= RZ ( R! Rc t attrst nameR` ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRe s "' c C s! t | t j � r d Sd | j S( Ns -->s </%s>( RS R Rm R= ( R! Rc ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRg s c C s� g } t | � r� t | � r� | j d | � | j | j | | � � | j | | � s� | j | j | j | j � � | j | j | | � � n | j | j | j | j � � | j d � d j | � S| j d | � | j | j | | � � | j d � t | j � s%t | j � re| j d | � | j | j | j | j � � | j d � n t
| � } t
| � } x� | s�| r4| s�| j | j | j d � | | d d � � q�n | s�| j | j | j d � | | d d � � q�n | j | j
| j d � | j d � | | d � � q�W| j d | � | j | j | | � � | j d � t | j � s�t | j � r�| j d | � | j | j | j | j � � | j d � n d j | � S( NR s
RO i i t +t -( RE RX t collect_diff_tagRd t collect_diff_textR? t collect_diff_end_tagRA RZ R
RF R[ RG R\ ( R! R, R- R Ri Rk RK RL ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR\ " sH
,, -
c C s� | j | j | j � s1 d | j | j f } n | j } g } | j d k p[ d | j k } x� t | j j � � D]� \ } } | | j k r� | r� | j d | | j | t � f � qt | | j k r� | j | j | | t � } n | j | t � } | j d | | f � qt W| s~x^ t | j j � � D]D \ } } | | j k rTq3n | j d | | j | t � f � q3Wn | r�d | d j | � f } n
d | } | S( Ns %s (got: %s)R; s +%s="%s"s %s="%s"s -%s="%s"s <%s %s>R s <%s>(
R< R= RB RC Rn RX Rf R* Rt RZ ( R! R, R- R= Ro R; Rp R` R? ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRs H s* "& "*
c C s< | j | j k r+ d | j | j f } n | j } d | S( Ns %s (got: %s)s </%s>( R= ( R! R, R- R= ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRu c s c C sO | j | | | � r/ | s d S| j | | � Sd | | f } | j | | � S( NRO s %s (got: %s)( R> Rf ( R! R, R- R
R? ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRt j s (
R R R R R R R R R R ( t __name__t
__module__Rb R" R' R( R6 R+ R> R< RY R@ Rd R[ Rf Re Rg R\ Rs Ru Rt ( ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR Q s&
!
# & c B s e Z d � Z RS( c C s t S( N( R ( R! ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR" s s ( Rv Rw R" ( ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR r s c C s | r t t _ n t t _ d S( s�
Install doctestcompare for all future doctests.
If html is true, then by default the HTML parser will be used;
otherwise the XML parser is used.
N( R t doctestR& R ( R ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR v s c C s� | r t } n t } t � } | j d } | � } | j } | | _ t rk | j d j } | j j } n | j d j } | j j } t t
_ t | | | | | | � d S( s�
Use this *inside* a doctest to enable this checker for this
doctest only.
If html is true, then by default the HTML parser will be used;
otherwise the XML parser is used.
R! t checkN( R R t _find_doctest_framet f_localst _checkert _IS_PYTHON_3t __func__R' t im_funcR Rx t _RestoreChecker( R t
del_modulet Checkert framet dt_selft checkert old_checkert
check_funct checker_check_func( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR � s"
R� c B sP e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C s` | | _ | | _ | j | j _ | | j _ | | _ | | _ | | _ | j � | j � d S( N(
R� R� t
call_superR% R# R� t
clone_funcR� t
install_clonet install_dt_self( R! R� R� t new_checkerR� R� R� ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt __init__� s
c C sm t r9 | j j | _ | j j | _ | j j | j _ n0 | j j | _ | j j | _ | j j | j _ d S( N( R} R� t __code__t func_codet __globals__t func_globalsR� ( R! ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR� � s c C s+ t r | j | j _ n | j | j _ d S( N( R} R� R� R� ( R! ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt uninstall_clone� s c C s | j j | _ | | j _ d S( N( R� t _DocTestRunner__record_outcomet prev_func( R! ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR� � s c C s | j | j _ d S( N( R� R� R� ( R! ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt uninstall_dt_self� s c C sp | j rl d d l } | j | j =d | j k rl | j j d d � \ } } | j | } t | | � ql n d S( Ni����t .i ( R� RV t modulest rsplitt delattr( R! RV t packaget modulet package_mod( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt uninstall_module� s
c O sF | j � | j � | j ` | j ` | j | | � } | j � | S( N( R� R� R� R# R% R� R� ( R! t argst kwt result( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt __call__� s
c O s0 | j � z | j | | � SWd | j � Xd S( N( R� R� R� ( R! R� R� ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR� � s
(
Rv Rw R� R� R� R� R� R� R� R� ( ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyR� � s c C sZ d d l } | j d � } x, | rI | j } d | k r= | S| j } q Wt d � � d S( Ni����i t BOOMsB Could not find doctest (only use this function *inside* a doctest)( RV t _getframeR{ t f_backt LookupError( RV R� t l( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyRz � s
si
>>> temp_install()
>>> print """<xml a="1" b="2">stuff</xml>"""
<xml b="2" a="1">...</xml>
>>> print """<xml xmlns="http://example.com"><tag attr="bar" /></xml>"""
<xml xmlns="...">
<tag attr="..." />
</xml>
>>> print """<xml>blahblahblah<foo /></xml>""" # doctest: +NOPARSE_MARKUP, +ELLIPSIS
<xml>...foo /></xml>
t basict __main__(- t __doc__t lxmlR RV RP Rx R R Rl t ImportErrort cgit __all__t
basestringRT t NameErrort strt bytest version_infoR} t register_optionflagR R R R& R
R t
HTMLParserR* R@ R R t compileR8 R
R R R R R t objectR� Rz t __test__Rv t testmod( ( ( s9 /usr/lib64/python2.7/site-packages/lxml/doctestcompare.pyt <module>$ sJ
� " '6