o [h@sdZddlmZddlZddlZddlZddlmZddlZddl Z ddl m Z e eZedZdZd Zd Zd ejfd d ZGdddeZdefddZdedefddZddZGdddeZdZdede dZ e dZ!defdd Z"dS)!z Even though this command doesn't require Django to be setup, it should run after django.setup() just to make sure it doesn't crash when pushed to Heroku ) BaseCommandN)Path) __version__.zi~ .git db.sqlite3 .pyo .pyc .pyd .idea .DS_Store .otreezip venv _static_root staticfiles __pycache__ .envzoTree-may-overwrite-this-filez!oTree-may-not-overwrite-this-filetar_infocCsf|j}tD] }||rdSqd|vrdS|jd?}|dkr*td|d||jdO_|S)NZ__temp izAdding large file (z KB): )pathEXCLUDED_PATH_ENDINGSendswithsizeloggerinfomode)rr endingkbrt/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/otree/cli/zip.py filter_funcs  rc@s eZdZdZddZddZdS)CommandzZip into an archivecKs ttdS)N) zip_project PROJECT_PATH)selfoptionsrrrhandle8s zCommand.handlecCsB||d|d}||dd}t|}|jdi|dS)z2 copy-pasted from 'unzip' command rrNr)Z create_parser parse_argsvarsr)rargvparserrZ cmd_optionsrrr run_from_argv;szCommand.run_from_argvN)__name__ __module__ __qualname__helprr!rrrrr5s r project_pathc Cs|jd}|d}|sd}t|tdt|D] \}}|| |q zt |Wnt yQ}ztt |tdWYd}~nd}~ww|d}| dt |d}|j|dtd Wdn1stwY|td |d dS) Nz .otreezipz settings.pyzjCannot find oTree settings. You must run this command from the folder that contains your settings.py file.rz.python-versionz3.12.5zw:gz)arcnamefilterzSaved your code into file "")nameexistsrerrorsysexitfix_reqs_filesitemsjoinpath write_textvalidate_reqs_filesRequirementsErrorstrtarfileopenaddrunlinkr) r& archive_nameZ settings_filemsgfnnew_textexcZ runtime_txttarrrrrFs.     rreturncCs|d}|d}|d}|r|dnd}d}t|vr#d}n(t|vrKtd}|dkr4d}n|d krC|jd td |iStj d d}|r]|jt i}|r[t ||j<|SiS) Nrequirements.txtrequirements_base.txtutf8r'FTzDo you want oTree to automatically keep your requirements files up to date? (Enter 'n' if you have custom requirements in requirements.txt or requirements_base.txt) (y/n): yn#  z Answer not recognized; skipping ) r2 read_textr,OVERWRITE_TOKENDONT_OVERWRITE_TOKENinputlowerr+r.stdoutwrite REQS_DEFAULTREQS_BASE_DEFAULT)r&rpathrbpathZ original_rtxtZoriginal_rbtxtZ can_overwriteZansdrrrr0ns2      r0cCs2g}|D]}|}|r|ds||q|S)N#)strip startswithappend)flineslinerrrget_non_comment_liness r\c@s eZdZdS)r5N)r"r#r$rrrrr5sr5z\# You should put your requirements in requirements.txt instead. # You can delete this file. rGzp # IF YOU MODIFY THIS FILE, remove these comments. # otherwise, oTree will automatically overwrite it. otree%s==z# psycopg2>=2.8.4 sentry-sdk>=0.7.9 r'c Cs|d}|d}|jdd }t|}Wdn1swY|rk|jdd }t|}Wdn1s;wYt}||D]"}td|}|ri|d} | |vrdd| d} t| || qGnd } | |vr{d | d } t| d | dvrd } t| dS)NrBrCrD)encodingz (^[\w-]+).*?rr*zM" is listed more than once in your requirements_base.txt & requirements.txt. z-r requirements_base.txtz,your requirements.txt has a line that says "z". You should remove that line.Zpsycopg2zBYou should add a line to your requirements.txt that says: psycopg2) r8r\r,setrematchgroupr5r9rI) r&rRrSrYZrlinesZrblines already_seenlnmpackager<ZREFERENCE_TO_REQS_BASErrrr4s:         r4)#__doc__baserr7osloggingpathlibrr.r_ZotreerZ otree_version getLoggerr"rresolversplitr rJrKTarInforrrdictr0r\ Exceptionr5rQZ_REQS_DEFAULT_FMTrPr4rrrrs8     (