o
[hE @ s d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
d dlmZ d dlmZmZmZmZ d d lmZ d d
lmZ d d lmZ d Zd
Z g Z dd Z G dd deZ G dd de Z dZ!dZ"dd Z#G dd deZ$dS ) N)run_in_threadpool)FormData) HTTPEndpoint)Request)ResponseRedirectResponse)settings)CSRF_TOKEN_NAMEAUTH_COOKIE_NAMEAUTH_COOKIE_VALUErng)db)Session)renderALWAYS_UNRESTRICTEDUNRESTRICTED_IN_DEMO_MODEc C s t t| |d d S )N)Z
alert_typemsg)_admin_message_queueappenddict)Zbootstrap_alert_typer r v/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/otree/views/cbv.pyenqueue_admin_message s r c @ s e Zd ZU dZeed< dZdZdZdd Z dd Z
d'd d
Z de fd d Z
defd
dZdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zdefd%d&ZdS )( AdminViewFrequestNc C s | j o| S N)_requires_login
_is_logged_inselfr r r _is_unauthorized" s zAdminView._is_unauthorizedc C s | j jttkS r )r sessiongetr
r r r r r r % zAdminView._is_logged_inreturnc sf t | j| jd | _}|j dkr | j I d H | _t| j |I d H }|| j| j| j
I d H d S )Nreceivepost) r scoper& r methodlowerform
_form_datar inner_dispatchsend)r r responser r r dispatch( s zAdminView.dispatchc C s | j S r )r, r r r r
get_post_data0 s zAdminView.get_post_datac C s | r | dS | jdi |j}|r|S |j dkrL| jsBtdsB| }|
t }|j
t }|r<|r<||krBt
dddS | j|fi |jS | j
|fi |jS )NZLoginr' ZOTREE_SKIP_CSRF zCSRF verification failed) status_codecontentr )r redirectintercept_dispatch path_paramsr) r* csrf_exemptosgetenvr1 r" r r! r r' )r r respZ post_dataZ
form_tokencookier r r r- 3 s
zAdminView.inner_dispatchc K s d S r r r kwargsr r r r6 D s z AdminView.intercept_dispatchc K s2 i }| }|r ||d< | jdi |}| |S )Nr+ r )get_formget_context_datarender_to_response)r r r> contextr+ r r r r" G s
z
AdminView.getc C s | j r| S d S r )
form_classr r r r r? O s zAdminView.get_formc O s, | j | d}| r| |S | |S )N)Zformdata)rC r1 validate
form_valid form_invalid)r r argsr> r+ r r r r' S s
zAdminView.postc C s | | S r )r5 get_success_url)r r+ r r r rE Y zAdminView.form_validc C s | j |d}| |S )N)r+ )r@ rA )r r+ rB r r r rF \ s
zAdminView.form_invalidc K sx | j jt}|sttdd}|| j jt< dt d| d}||t< |j| tt j
| j j | j j
d || |S )Ni ʚ;l d( z)viewZadmin_message_queueZis_debugZcurrent_page_nameurl_for)r r! r" r strr randintupdater r DEBUG __class____name__rK vars_for_template)r r> Z
csrf_valueZ token_inputr r r r@ ` s zAdminView.get_context_datac C s i S )z
simpler to use vars_for_template, but need to use get_context_data when:
- you need access to the context produced by the parent class,
such as the form
r r r r r rR u s zAdminView.vars_for_templatec C s t | |S r )r get_template_name)r rB r r r rA } rI z AdminView.render_to_responsec C s | j jS r )r urlr r r r rH s zAdminView.get_success_urlc C s d | jjS )Nz
otree/{}.html)formatrP rQ r r r r rS rI zAdminView.get_template_namec K s t | jj|fi |ddS )Ni. r3 )r r rK )r Zurl_namer> r r r r5 s zAdminView.redirectr$ N) rQ
__module__ __qualname__r8 r __annotations__r rC r, r r r0 r r1 r- r6 r" r? r' rE rF r@ rR rA rH rS r r5 r r r r r s.
r c s> e Zd ZU eed< edd Z fddZ fddZ Z S )AdminSessionPager! c C s
d| j S )Nz
/%s/{code})rQ )clsr r r url_pattern s
z AdminSessionPage.url_patternc s t jdd| ji|S )Nr! r )superr@ r! r= rP r r r@ s z!AdminSessionPage.get_context_datac s" t jt|jd d| _t |S )Ncode)r` )r
Z
get_or_404r r7 r! r^ r- )r r r_ r r r- s z AdminSessionPage.inner_dispatch)
rQ rX rY r rZ classmethodr] r@ r-
__classcell__r r r_ r r[ s
r[ ZOTREE_REST_KEYzotree-rest-keyc C s t | ddS )Nr2 rV r )r r r r _HttpResponseForbidden rd c @ sB e Zd ZdZdddZdd Zdd Zd
d Zd d
Zdd Z dS ) BaseRESTViewz
async so that i can get request.json().
inner_post should also call group_send, at there are complications with doing sync_group_send.
r$ Nc sn t | j| jd }| _| I d H }|rt|}ni }|| _t| j |I d H }|| j| j| j
I d H d S )Nr% ) r r( r& r bodyjsonloads_payloadr r- r. )r r rg payloadr/ r r r r0 s zBaseRESTView.dispatchc
C s t jdv r0tt}|stdt dS |jt}|s$tdt dS ||kr0tdt dS z|j
dkr=| W S | W S t
yY } z tt|W Y d }~S d }~ww )N)ZDEMOZSTUDYzEnv var z must be defined to use REST APIzHTTP Request Header z is missingz
is incorrectr' )r Z
AUTH_LEVELr9 r:
REST_KEY_NAMErd headersr" REST_KEY_HEADERr) r*
outer_post outer_get TypeErrorrL )r r ZREST_KEYZsubmitted_rest_keyexcr r r r- s.
zBaseRESTView.inner_dispatchc C | j di | jS Nr )r" rj r r r r rp s zBaseRESTView.outer_getc C rs rt )r' rj r r r r ro r# zBaseRESTView.outer_postc K t dddS NzMethod not allowedi rV rc r= r r r r" re zBaseRESTView.getc K ru rv rc r= r r r r' re zBaseRESTView.postrW )
rQ rX rY __doc__r0 r- rp ro r" r' r r r r rf s
rf )%asyncior9 rh Zstarlette.concurrencyr Zstarlette.datastructuresr Zstarlette.endpointsr Zstarlette.requestsr Zstarlette.responsesr r Zotreer Z otree.commonr r
r r Zotree.databaser
Z otree.modelsr Zotree.templatingr r r r r r r[ rl rn rd rf r r r r s. q