o [h"@sddlZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZeeZeZd ZGd d d eZdedefddZdZ dZ!dZ"defddZ#GdddZ$dZ%dZ&de e$fddZ'dS)N)Path)TemporaryDirectory)sleep)Optional)unzip)send_termination_notice) BaseCommand)check_update_neededZ8000c@seZdZddZddZdS)CommandcCs|jddddS)Nzipfile?)nargs) add_argument)selfparserrz/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/otree/cli/zipserver.py add_argumentsszCommand.add_argumentscKsd|d}z"|r t|}nt}|dkr tt| WdSt|WdSty1YdSw)Nr r) getrun_single_zipfileautoreload_for_new_zipfilesoskillgetpidsysexitKeyboardInterrupt)roptionsr exit_coderrrhandles   zCommand.handleN)__name__ __module__ __qualname__rr rrrrr s r fnreturncCs$tt|}|||SN)Projectrunzip_to_tempdirstartwait)r$projectrrrr.s rz7No *.otreezip file found in this folder yet, waiting...zNewer project foundz Running {}c Csd}t}d}|stt t}|rntdqg}zY |r!|}tt|||r8||dj | |j |  | }|dkr[|W||D]}|qSStdt}|rs||krs|}tt|nqCq||D]}|q{w)NTr)get_newest_project stdout_writeMSG_NO_OTREEZIP_YETrMSG_RUNNING_OTREEZIP_NAMEformatzipnamer(take_db_from_previousnameappendtmpdirr)pollr*cleanupMSG_FOUND_NEWER_OTREEZIP terminate)rr+Z newer_projectZtempdirstdZlatest_projectrrrr<sV    rc@seZdZUdZeed<ejed<defddZ ddZ d d Z d d Z d dZ ddZddZddZdefddZddZdefddZddZdS)r'Nr6_procotreezipcCs ||_dSr& _otreezip)rr=rrr__init__r zProject.__init__cCs|jjSr&)r?r4rrrrr2uszProject.zipnamecCs |jjSr&)r?statst_mtimerBrrrmtimex z Project.mtimecCs |j|jkSr&r>)rotherrrr__eq__{rFzProject.__eq__cCs"t|_tt|j|jjdSr&)rr6rstrr?r4rBrrrr(~szProject.unzip_to_tempdircCs.|tjddtg|jjtjd|_ dS)NZotreeZdevserver_inner)cwdenv) r subprocessPopenPORTr6r4renvironcopyr<rBrrrr)s z Project.startcCs|jdSr&)r?unlinkrBrrrdelete_otreezipszProject.delete_otreezipcC |jSr&)r<r7rBrrrr7rAz Project.pollr%cCrSr&)r<r*rBrrrr*rAz Project.waitcCs"tt}|jt|ddS)N )rrNr<r:rr)rZ child_pidrrrr:s zProject.terminate other_tmpdircCs6dD]}t||}|rtt||jjqdS)N)z db.sqlite3)rexistsshutilmoverIr6r4)rrUitemZ item_pathrrrr3s  zProject.take_db_from_previouscCs,tt|jjd}|rt|dSdS)a< The main need to check if requirements.txt matches the current version is for oTree Studio users, since they have no way to control what version is installed on the server. we instead need the otreezip file to tell their local installation what version to use. We used to check if an update was needed for any otree command (devserver etc), but i think putting it here is more targeted with a clearer scenario. other cases are not really essential and there are already other ways to handle those. zrequirements.txtN)r rr6r4joinpathloggerwarning)rr\rrrr s  zProject.check_update_needed)r!r"r#r6r__annotations__rLrMrr@r2rErHr(r)rRr7intr*r:rIr3r rrrrr'ns    r' zDeleting old file: {}cCsjddtddD}|sdSt|dddd}|d }|tdD]}tt||q#|S) NcSsg|]}t|qSr)r').0pathrrr sz&get_newest_project...z *.otreezipcSs|Sr&)rE)Zprojrrrsz$get_newest_project..T)keyreverser) rglobsortedMAX_OTREEZIP_FILESr.MSG_DELETING_OLD_OTREEZIPr1r2rR)ZprojectsZsorted_projectsZnewest_projectZold_projrrrr-s r-)(loggingrZos.pathrWrLrpathlibrtempfilertimertypingrrZ otree.mainrbaser Z otree.updater getLoggerr!r[printr.rNr rIr^rr/r9r0rr'rirjr-rrrrs8          2H