o [h@sUddlZddlmZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmmZd Zed ZGd d d eZd ZGdddZdZeed<ddZdefddZdefddZddZ ddZ!dS)N)rng) OrderedDict)Dict)Session)ParticipantBot Submission) make_botsPzotree.test.browser_botsc@s eZdZdS)BadRequestErrorN)__name__ __module__ __qualname__rry/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/otree/bots/browser.pyr sr zBot for Participant {participant_code} not loaded. This can happen for several reasons: (1) You restarted the server after creating the session (2) The bots expired (the server stores bots for only the most recent {prune_limit} sessions).c@sPeZdZddZddZddZddZd efd d Zd e fd d Z ddZ dS) BotWorkercCst|_i|_i|_dSN)rparticipants_by_session browser_botsqueued_post_data)selfrrr__init__'s zBotWorker.__init__c Cs|g|j|<tj|d}|dur)ddlm}||j}|}t t |}t ||dd}|D]}|j| |j ||j|j <q2dS)N)idr) SessionConfigT) session_pk case_numberZuse_browser_bots)prunerrZ objects_getZ otree.sessionrconfigZget_num_bot_casesrchoiceranger appendparticipant_coder) rrrsessionrrZ num_casesZbotsbotrrrinitialize_session,s    zBotWorker.initialize_sessioncCs@t|jtkr|jjdd\}}|D] }|j|dqdSdS)zto avoid memory leaksF)lastN)lenrSESSIONS_PRUNE_LIMITpopitemrpop)r_Zp_codesr!rrrr@s zBotWorker.prunecCs2z|j|WStytj|td}t|w)N)r!Z prune_limit)rKeyError PARTICIPANT_NOT_IN_BOTWORKER_MSGformatr'r )rr!msgrrrget_botGs  zBotWorker.get_botreturncCsD|j}||vr dS||}z |||<WdSty!YdSw)NTF)rr/Zget_next_submit StopIteration)rr!Zqpdr#rrrenqueue_next_post_dataPs  z BotWorker.enqueue_next_post_datacCs|j|}|jSr)rr)Z post_data)rr!Z submissionrrrpop_enqueued_post_datads z BotWorker.pop_enqueued_post_datacCs||}||_||_dSr)r/pathhtml)rr! request_pathr5r#rrrset_attributesjs  zBotWorker.set_attributesN) r r rrr$rr/boolr2rr3r7rrrrr&s  rbrowser_bot_workercKstjdi|dSNr)r9r7kwargsrrrr7usr7r0cKtjdi|Sr:)r9r2r;rrrr2yr2cKr=r:)r9r3r;rrrr3}r>r3cKr=r:)r9r$r;rrrr$r>r$cCs"t|}tj|t|dddS)N)r!)groupdata) channel_utilsZbrowser_bots_launcher_groupZsync_group_senddict)Z session_coder! group_namerrrsend_completion_messages   rD)"loggingZ otree.commonr collectionsrtypingrZ otree.modelsrr#rrrunnerr Zotree.channels.utilsZchannelsutilsrAr' getLoggerlogger Exceptionr r,rr9__annotations__r7r8r2rBr3r$rDrrrrs(       L