bs\c@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z ddlZejZdZejZdZdZdefdYZd Zd efd YZdS( i(tprint_functionN(tloadstdumpsis RENPY RPC2scache/bytecode.rpybt ScriptErrorcBseZdZRS(sa Exception that is raised if the script is somehow inconsistent, or otherwise wrong. (t__name__t __module__t__doc__(((srenpy/script.pyR8scCs+g}x|D]}|j|jq W|S(sO Returns a flat list containing every statement in the tree stmts. (t get_childrentappend(tstmtstrvti((srenpy/script.pytcollapse_stmts?s tScriptcBseZdZdZdZdZdZdZdZdZ dZ d d Z e dd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZRS(sk This class represents a Ren'Py script, which is parsed out of a collection of script files. Once parsing and initial analysis is complete, this object can be serialized out and loaded back in, so it shouldn't change at all after that has happened. @ivar namemap: A map from the name of an AST node to the AST node itself. This is used for jumps, calls, and to find the current node when loading back in a save. The names may be strings or integers, strings being explicit names provided by the user, and integers being names synthesised by renpy. @ivar initcode: A list of priority, Node tuples that should be executed in ascending priority order at init time. @ivar all_stmts: A list of all statements, that have been found in every file. Useful for lint, but tossed if lint is not performed to save memory. cCs|tj_tjjtjjdrMttjjddj |_ n d|_ i|_ g|_ g|_g|_g|_t|_i|_i|_t|_tjj|_|j|j|jjd|_tjtj |_!t|_"g|_#g|_$dS(s Loads the script by parsing all of the given files, and then walking the various ASTs to initialize this Script object. s /lock.txttrbiN(%trenpytgametscripttostpathtexiststconfigt renpy_basetfiletreadtkeytNonetnamemapt all_stmtst all_pycodet all_pyexprt need_analysistTruet record_pycodetbytecode_oldcachetbytecode_newcachetFalsetbytecode_dirtyt translationtScriptTranslatort translatort init_bytecodetscan_script_filestchain_translatestserialthashlibtmd5t version_onlytdigestt loaded_rpyt backup_listtduplicate_labels(tself((srenpy/script.pyt__init__cs, %                cCstjr dSx-dddgD]}tjj|rdSqWddl}|jtjjd}|dkrqdSt j j tjj }tj j j tjj|tjj|}tjjd||S(Nsscript_version.txtsscript_version.rpysscript_version.rpycitbackupssBacking up script files to %r:(RtmobileRtloadertloadablet__main__t path_to_savesRtgamedirRRtbasenametbasedirtjointexportstfsencodet write_log(R4R R:R6R=t backupdir((srenpy/script.pytchoose_backupdirs   c Cs|j}g|_tjjdddkr1dS|js>dStjrKdS|j}|dkrgdSx|D] \}}|j tj j sqnntj j |sqnntjj|ttj j d}tj j|\}}tj j||d|d jd|}tj j |r1qnny tjtj j|dWnnXytj||WqnqnXqnWdS( NtRENPY_DISABLE_BACKUPStsI take responsibility for this.it.ithexi(R2RtenvirontgetR1RR7RDRt startswithRR<RRR@RAtlentsplitextR?tencodetmakedirstdirnametshutiltcopy( R4R2RCtfntchecksumtshort_fntbasetextt target_fn((srenpy/script.pyt make_backupss>      & " cCstjj}g|_g|_x|D]\}}|jdrk|dkrUq(n|d }|j}n|jdr|d }|j}n_|jdr|dkrq(n|d }|j}n(|jdr(|d }|j}nq(||f|kr(|j||fq(q(WdS( s8 Scan the directories for script files. s.rpyis.rpycis.rpyms.rpymciN(RR8t listdirfilest script_filest module_filestendswithRR(R4tdirlisttdirRSttarget((srenpy/script.pyR*s.            c Cs|j}|jg}xR|D]J\}}tjrQddl}|jdn|jdd|||q Wgt|D]!\}\}}|||f^q{}|jg|D]\}}}||f^q|_|jj dS(Niis.rpycs.rpy( R[tsortRt emscriptentsleeptload_appropriate_filet enumeratetinitcodeR(R+( R4R[RfRSR_Rbtindextpriotcode((srenpy/script.pyt load_scripts    1 +cCsg|jD]$\}}||kr ||f^q }|sMtd|nt|dkrrtd|n|d\}}g}|jdd|||tjjrtdn|jj |S(NsModule %s could not be loaded.is-Module %s ambiguous, multiple variants exist.is.rpymcs.rpymi( R\t ExceptionRLRdRtparsertreport_parse_errorst SystemExitR(R+(R4tnameRSR_tfilesRf((srenpy/script.pyt load_modules4 cCsit|}ttj}xD|D]<}|jdkr%|||jf|_|jd7_q%q%WdS(Ni(R tintttimeRoRR,(R4R RSRtversionts((srenpy/script.pyt assign_names1s   c Cst|}t|}g|D]}|j^q}g|D]}|j^q>}tjd||}xn|jD]`\}}} xNt| D]@}|||} |||} | jdkr| j| _qqWqxWdS(N(R t diff_infotdifflibtSequenceMatcherRtget_matching_blockstrangeRo( R4t old_stmtst new_stmtsR told_infotnew_infotsmtoldltnewltcounttoldtnew((srenpy/script.pyt merge_names=s  icCsmtjj||d|}|dkr+dS|j|||j|g}|j||t}||fS(s9 Loads Ren'Py script from a string. `filename` The filename that's assigned to the data. `filedata` A unicode string to be loaded. Return the list of statements making up the root block, and a list of init statements that need to be run. t linenumberN(NN(RRltparseRRvtstatic_transformst finish_loadR$(R4tfilenametfiledataRR Rf((srenpy/script.pyt load_stringOs  c s|s |Stjj|dg}x|D]}|j|jq*Wjj||dk rtjj |}|dj j j |j s|ddkr|d7}nx|D]}||_ qWqnfd}j xr|D]j}|j} |||j| <|jrH|j} | rH|j| qHn|jr|jqqWjdk rjj|njj||S(s Given `stmts`, a list of AST nodes comprising the root block, finishes loading it. `initcode` A list we append init statements to. `check_names` If true, produce duplicate name errors. `filename` If given, a filename that overrides the filename found inside the file. Returns a list of statements that corresponds to the top-level block in initcode after transformation. iitccss dStjrdSd}d}d}|j}|jkr|}|}j|}t|tstdt||j |j |j |j fqtj j rdSj jdj||j |j |j |j ndS(Ns-Name %s is defined twice, at %s:%d and %s:%d.uPThe label {} is defined twice, at File "{}", line {} and File "{}", line {}.(RR7RRoRt isinstancet basestringRtreprRRRtallow_duplicate_labelsR3Rtformat(tnodetbad_nametbad_nodetold_nodeRo(t check_namesR4(srenpy/script.pyt check_names,       N(Rtastt chain_blockRRRR(ttake_translatesRltelide_filenameRtlowerR]tupdate_bytecodeRoRtget_initt early_executeRtextendR( R4R RfRRRR RRRotinit((RR4srenpy/script.pyRks<  "  $        cCsG|jtx3tdD]%}|jtjddddqWdS(sU Writes an empty version 2 .rpyc header to the open binary file `f`. itIIIiN(twritet RPYC2_HEADERR{tstructtpack(R4tft_i((srenpy/script.pytwrite_rpyc_headers cCs|jdd|j}tj|d}|j||jttd|dd|jtjd||t||jdddS(s Writes data into `slot` of a .rpyc file. The data should be a binary string, and is compressed before being written. iii i iRN( tseekttelltzlibtcompressRRLRRR(R4Rtslottdatatstart((srenpy/script.pytwrite_rpyc_datas  "%cCs!|jdd|j|dS(s< Writes the md5 to the end of a .rpyc file. iiN(RR(R4RR0((srenpy/script.pytwrite_rpyc_md5sc Cs|jd}|tt tkr[|dkr5dS|jd|j}|jdStt}xZtrtjd|||d!\}}}||krPn|dkrdS|d7}qjW|j||j|}t j |S(s Reads the binary data from `slot` in a .rpyc (v1 or v2) file. Returns the data if the slot exists, or None if the slot does not exist. iiiRRi N( RRLRRRtdecodeR RtunpackRt decompress( R4RRt header_dataRtpost header_slotRtlength((srenpy/script.pytread_rpyc_datas$      &   cCstjj|dS(s This performs transformations on the script that can be performed statically. When possible, these transforms are stored in slot 2 of the rpyc file. N(RR&t restructure(R4R ((srenpy/script.pyR$scCsu|jds|jdr*|s7td|n|d|}|d}tjj|}i}t|d<|jp}d|d<|dkr|gfSzmy_t|_ t |d }|j |d }WdQXt |\} } |j | |~ ~ WnnXWdt|_ X|j|tjj|tjsyBt|d } |j| |j| d t||fd WqqXn|j|tjsys|j| d t||fd t |d "} tj| jj} WdQX|j| | | jWqqXnt|_nA|jdsH|jdrgd}d}tjj|} zxad d gD]O}y2|j | |}|rt |\}}PnWnnX| j dqvWdS|dkrt!d|dSt"|t#sdS|jr%|j$dd|jkr%dS|dtkr9dS|d krU|j|nWd| jXndS||fS(Ns.rpys.rpyms4Cannot load rpy/rpym file %s from inside an archive.t/RRttunlockedRRitwbitrUs.rpycs.rpymcisFailed to load(NN(NN(NN(NN(NN(NN(%R]RkRRlRtscript_versionRRR$R!topenRRRR RvRtmacappRRRRRR-R.RR0RtcloseR1R8tloadRtprintRtdictRJ(R4R_RStfullfntrpycfnR Rtrpycftbindatatold_dataR|Rtfullft rpydigestR((srenpy/script.pyt load_file.s        &  "!   $ cCs|dkr||}||}|j|||\}} |dkr^td|fntjj||} | jtjj d| j tjj } | j n|d||}|d||} tjj |t jj|r/t|d"} tj| j j} WdQXnd} yit jj| rt| d9} | jtjj d| j tjj }WdQXnd}Wn d}nXd} t jj|rt jj| rtjjj}|}d \}} yS| |krK| rK|j|||\}} |dkrKtd| qKnWn)dt jkrxtd| qxnX|dkr|j|||\}} n| } nzt jj| r| }|j|||\}} |} n=t jj|r&|}|j|||\}} | } n| dk rK|jj|| fn|dkrjtd |n|jdkr|d |_n&|j|d krt|d n|j| |d ||jj| dS(NsCould not load from archive %s.iRRRsCould not load tRENPY_RPYC_EXCEPTIONSs While loadingsCould not load file %s.Rs{ does not share a key with at least one .rpyc file. To fix, delete all .rpyc files, or rerun Ren'Py with the --lock option.R(NN(RRRkRR8RRR-R.t digest_sizeRRtadd_autoRRRRR0RtargstcompileRRIR2RRRtupdate(R4tcompiledtsourceR_RSRftrpyfntlastfnRR RR0RRt rpycdigestt force_compile((srenpy/script.pyRdsz     $!  $         cCsWyIttjjtjjd\}}|tkrH||_nWnnXdS(s0 Init/Loads the bytecode cache. RN( RRR8Rt BYTECODE_FILERRtBYTECODE_VERSIONR"(R4Rttcache((srenpy/script.pyR) s - c Cspx2|jD]'}ytjj|dWq q Xq Wg|_x"|jD]}|jt}|jj|d}|dkr@t |_ tj j }d|jd|jdftj _ y|jdkrtjj|jd|jdd|jd}n|jdkr=tjj|jd|jdd|jd}nA|jdkr~tjj|jd|jdd|jd}nWntk r0}|j}|dkrd }ny|jd }Wn|jd }nXtjjd|jd |jd |jd|d|j}tjjj|jqHnX|tj _ n||j |s&