B `zL@sddlZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z dd l m Z dd l mZdd lmZddlZGd ddejZdS)N)loads)forms)utils) print_stderr)PY3)urlparse)to_bytes) string_types) binary_type) text_type) BeautifulSoupc@s6eZdZdZdZdZdZeddZeddZ edd Z e d e j e jBZd d Zd dZddZddZd8ddZd9ddZddZd:ddZe edZeddZe dZed d!Zd"d#Zd$d%Zd&d'Zd(d)Z d*d+Z!ed,d-Z"ed.d/Z#ed0d1Z$ed2d3Z%ed4d5Z&d6d7Z'dS); TestResponsez\ Instances of this class are returned by :class:`~webtest.app.TestApp` methods. Nz html.parsercCs|jdkr||jS)a Returns a dictionary containing all the forms in the pages as :class:`~webtest.forms.Form` objects. Indexes are both in order (from zero) and by form id (if the form is given an id). See :doc:`forms` for more info on form objects. N)_forms_indexed _parse_forms)selfrX/home/kop/projects/devel/pgwui/test_venv/lib/python3.7/site-packages/webtest/response.pyrs zTestResponse.formscCs*|j}|stdd|kr"td|dS)z If there is only one form on the page, return it as a :class:`~webtest.forms.Form` object; raise a TypeError is there are no form or multiple forms. z*You used response.form, but no forms existz5You used response.form, but more than one form existsr)r TypeError)rforms_rrrform,szTestResponse.formcCsD||jr6y|jStk r4|j|jdSX|jddS)Nreplaceascii)decode_contentcharsettextUnicodeDecodeErrorbodydecode)rrrrtestbody=szTestResponse.testbodyz<(/?)([:a-z0-9_\-]*)(.*?)>cCs`i}|_dd|dD}xKsz-TestResponse._parse_forms..r)rhtml enumeraterZFormparser_featuresid)rrZ form_textsirrrrrrIs zTestResponse._parse_formscKs*|jd}t|jj|}|jj|f|S)Nlocation)headersrurljoinrequesturltest_appget)rkwr(Z abslocationrrr_followRs zTestResponse._followcKs2d|jkrdks&ntd|j|jf|S)a If this response is a redirect, follow that redirect. It is an error if it is not a redirect response. Any keyword arguments are passed to :class:`webtest.app.TestApp.get`. Returns another :class:`TestResponse` object. i,iz/You can only follow redirect responses (not %s)) status_intAssertionErrorstatusr0)rr/rrrfollowXs zTestResponse.followcKsTd}|}x6d|jkr dkr>nn|r>|jf|}|d8}q W|dkrPtd|S)z Follow all redirects. If this response is not a redirect, do nothing. Any keyword arguments are passed to :class:`webtest.app.TestApp.get`. Returns another :class:`TestResponse` object. di,irrzredirects chain looks infinite)r1r0r2)rr/Zremaining_redirectsresponserrr maybe_followes   zTestResponse.maybe_followFc CsR|jddd|||||d\}}} |p&i}|dt|jj|jt| d|dS)a Click the link as described. Each of ``description``, ``linkid``, and ``url`` are *patterns*, meaning that they are either strings (regular expressions), compiled regular expressions (objects with a ``search`` method), or callables returning true or false. All the given patterns are ANDed together: * ``description`` is a pattern that matches the contents of the anchor (HTML and all -- everything between ```` and ````) * ``linkid`` is a pattern that matches the ``id`` attribute of the anchor. It will receive the empty string if no id is given. * ``href`` is a pattern that matches the ``href`` of the anchor; the literal content of that attribute, not the fully qualified attribute. If more than one link matches, then the ``index`` link is followed. If ``index`` is not given and more than one link matches, or if no link matches, then ``IndexError`` will be raised. If you give ``verbose`` then messages will be printed about each link, and why it does or doesn't match. If you use ``app.click(verbose=True)`` you'll see a list of all the links. You can use multiple criteria to essentially assert multiple aspects about the link, e.g., where the link's destination is. ahrefN)tag href_attr href_extractcontentr& href_patternindexverbose HTTP_REFERERuri) extra_environ) _find_element setdefaultstrr+r,goto) r descriptionZlinkidr9r?r@rC found_html found_desc found_attrsrrrclickws%zTestResponse.clickc CsL|jddtd|||||d\}}}dt|jji} |jt|d| dS)z Like :meth:`~webtest.response.TestResponse.click`, except looks for link-like buttons. This kind of button should look like ``