) page_namefieldstags)r )r$ r1 PageHtmlCheckerget_missing_fieldshtmlr0 Zurl_nameMissingHtmlFormFieldErrorHTML_MISSING_FIELD_WARNINGr* join
field_tagssubmit_button_foundMissingHtmlButtonErrorHTML_MISSING_BUTTON_WARNING)r2 r fields_to_checkcheckermissing_fieldsr r r r! r ) s0
z ParticipantBot.assert_html_okc
C s |j j}d| d| jvrHddlm} t| jjdddd }|| j|}tt |j
|j |d}t|j |j |j jd}d| d | d
} t
| d S ) N/r )get_page_lookup )maxsplit)rs r pagezNDiscrepancy between bot code and app code. Bot is trying to submit this page:
z'
But the participant is actually here:
r )r0 r rh Z otree.lookupr intrsplitrc dicttyper r app_namer+ )
r2 r r Z ClassNamer idxlookupexpectedactualr4 r r r! r A s. z"ParticipantBot.assert_correct_pagec C | j S r7 )rf rp r r r! rz [ zParticipantBot.responsec C s2 t |j| _t| jj| _|| _|jd| _d S )Nzutf-8)r re r rh rf contentdecoder )r2 rz r r r! rz _ s c C r r7 )rg rp r r r! r g r zParticipantBot.htmlc C s t t|| _d S r7 )
HtmlStringnormalize_html_whitespacerg r2 r r r r! r k s c C s, t | jsdS | jj| jdd| _t | jS )NFTrx ) is_wait_pagerz rq rU re rp r r r! on_wait_pageo s
zParticipantBot.on_wait_pagec C sr |j }t|}d| j }|r|d|7 }|dr |d7 }|dr'|d7 }t| | jj| j |dd| _
d S ) NzSubmit z, {}r z, SubmissionMustFailr z, timeout_happenedTrx ) r1 bot_prettify_post_datarh r* rU loggerinforq postre rz )r2 r r1 Zpretty_post_dataZ
log_stringr r r! r y s
zParticipantBot.submit)r r r r6 propertyrq ro r{ r} rl r r r# r r rz setterr r r r r r r! r] s,
r] c @ s e Zd Zg ZdededefddZdd Zedd Zed
d Z ed d
Z
edd Z edd Z edd Z
edd ZdS ) PlayerBot case_number player_pk
subsession_pkc C sv t |}|j| _|j| _|j| _|| _|| _|| _ || _
|d kr"d}| j }t |dkr6||t | | _
d S d | _
d S )Nr r )r ZPlayer PlayerClassGroupZ
GroupClassZ
SubsessionSubsessionClass
_player_pk_subsession_pk _session_pk_participant_codecasesrL r ) r2 r r r r Z
session_pkrb Z
models_moduler r r r! r6 s
zPlayerBot.__init__c C s d S r7 r rp r r r! r~ s zPlayerBot.play_roundc C | j j| jdS N)id)r ra r rp r r r! r zPlayerBot.playerc C | j jS )z3can't cache self._group_pk because group can change)r r rp r r r! r s zPlayerBot.groupc C r r )r ra r rp r r r!
subsession r zPlayerBot.subsessionc C r r7 )r r rp r r r! r zPlayerBot.round_numberc C t j| jdS r_ )r
ra r rp r r r! rn zPlayerBot.participantc C r r )r ra r rp r r r! session r zPlayerBot.sessionc C r r7 )rj r rp r r r! r r zPlayerBot.htmlN)r r r r r r6 r~ r r r r r rn r r r r r r! r s0
r c @ r )r Nr r r r r! r r" r c @ r )r Nr r r r r! r r" r c C s> i }t | ttfr| d }t| dkr| d }n| }t||S )Nr r? r )r, listtuplerL r# )Z
yielded_valuer1 r3 r r r! r s
r c C s& | dd dd} tdd| } | S )Nr r
z\s+)replaceresub)r r r r! r s r c @ s$ e Zd Zdd Zdd Zdd ZdS )r c C s& d}| |t| }d| |d S )z
Make output more readable by truncating everything before the
{% content %} block. I also considered indenting the HTML,
but minidom had a parse error, and BS4 modifies a lot of tags,
didn't seem optimal.
z z...N)indexrL )r2 Zdiv_strir r r! truncated s zHtmlString.truncatedc C | S r7 r rp r r r! __str__ zHtmlString.__str__c C r r7 r rp r r r! __repr__ r zHtmlString.__repr__N)r r r r r r r r r r! r s r c s< e Zd Z fddZdd Zdd Zdd Zd d
Z ZS ) r c s( t t|| _h d| _d| _d S )N> selectinputbuttonZtextareaF)r9 r6 ri r r r )r2 r r: r r! r6 s
zPageHtmlChecker.__init__c C s | | | jS r7 )feedr r r r r! r s
z"PageHtmlChecker.get_missing_fieldsc C s4 || j v r|D ]\}}|dkr| j| qd S d S )Nname)r r discardr2 r attrsZ attr_nameZ
attr_valuer r r! check_if_field s
z PageHtmlChecker.check_if_fieldc C sp | j s2|dkr |D ]\}}|dkr|dkr d S q d| _ |dkr4|D ]\}}|dkr1|dkr1d| _ q"d S d S d S )Nr r Tr r )r r r r r! check_if_button
s z PageHtmlChecker.check_if_buttonc C s | || | || d S r7 )r r )r2 r r r r r! handle_starttag s z PageHtmlChecker.handle_starttag) r r r r6 r r r r r= r r r: r! r s r c C s | j tjjtjjkS r7 )headersrU r& r' Zwait_page_http_headerZget_param_truth_value)rz r r r! r s r c C s$ t | dr | } dd | D S )Nr c S s i | ] \}}|t vr||qS r r )r kvr r r! r . s z*bot_prettify_post_data..)r) r items)r1 r r r! r % s
r ) r\ r] r r r r r r- r objectr r r r r r r! sb
+' PC"