a 8^i.@sddlmZmZmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlZddlZddlZiaeZGdddeZddZdd Zd d Zd d ZddZddZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&ddZ'd a(d!d"Z)d#d$Z*dS)%)divisionabsolute_importwith_statementprint_functionunicode_literals) PY2 basestringbchrbordchropenpystrrangeroundstrtobytesunicodeNc@s eZdZdZddZddZdS)Linez. Represents a logical line in a file. cCs:|dd}||_||_||_||_||_d|_d|_dS)N\/)replacefilenamenumberstartend end_delimtext full_text)selfrrrr TC:\Program Files (x86)\Steam\steamapps\common\Selene ~Apoptosis~\renpy\scriptedit.py__init__/s z Line.__init__cCsd|j|j|jS)Nz)formatrrr)rr r r!__repr__Isz Line.__repr__N)__name__ __module__ __qualname____doc__r"r$r r r r!r*srcCsN|ds|dsdS|tvr$dSt|tj|}tjj|dddS)z} Ensures that the given filename and linenumber are loaded. Doesn't do anything if the filename can't be loaded. z.rpyz.rpycNT)Z add_lines)endswithfilesaddrenpyparserZunelide_filenameZlist_logical_lines)rfnr r r! ensure_loadedMs  r/cCs6|dd}t|||ftvr.t||fjSdSdS)zq Gets the text of the line with `filename` and `linenumber`, or the None if the line does not exist. rrN)rr/linesrr linenumberr r r! get_line_text_s   r3c Cs|dd}t|i}tD]l\}}|\}}||kr||kr||7}|j|7_|j|7_|j|7_|j|7_||||f<q |adS)a8 Adjusts the locations in the line data structure. `filename`, `linenumber` The filename and first line number to adjust. `char_offset` The number of characters in the file to offset the code by,. `line_offset` The number of line in the file to offset the code by. rrN)rr/r0itemsrrrr) rr2Z char_offset line_offsetZ new_lineskeyliner.lnr r r!adjust_line_locationsos r9c Cs|dd}tjjrtdt|t||f}td|j }| d}|sPd}|j dsh|j dsnd}nd}||}|||}t |j |j|j}||_ ||_|jt||_|jt||_t|j d d } | } Wd n1s0Y| d |j|| |jd } t||t||dtjj^t|j d d } | | Wd n1sp0Ytjj|j d dWd n1s0Y|t||f<d S)z Adds `code` immediately before `filename` and `linenumber`. Those must correspond to an existing line, and the code is inserted with the same indentation as that line. rrTr?)rr,rArBrCr/r0rHr rrIrrr9rGrJrKrLrMrN)rr2r7rRrSrOr r r! remove_lines  &  (rUcCs2|dd}t|||ftvr$dSt||fjS)z~ Returns the full text of `linenumber` from `filename`, including any comment or delimiter characters that exist. rrN)rr/r0rr1r r r! get_full_texts   rVcCsHt|g}tjjjD],}|j|kr|j|kr|jdkr||q|S)zC Returns a list of nodes that are found on the given line. never) r/r,gamescript all_stmtsrr2rollbackappend)rr2rvir r r! nodes_on_lines  r_cs8tfddtjjjD}|s*gStt|S)zL Returns a list of nodes that are found at or after the given line. cs0g|](}|jkr|jkr|jdkr|jqS)rW)rr2r[).0r^r1r r! s    z-nodes_on_line_at_or_after..)r/r,rXrYrZr_min)rr2r0r r1r!nodes_on_line_at_or_after s rccCsg}g}|D]H}|D]}|j|urq2q|||D]}|j|ur6q q6||q t|dkrjtdt|dkr~td|d|dfS)z Finds the first and last nodes in `nodes`, a list of nodes. This assumes that all the nodes are "simple", with no control flow, and that all of the relevant nodes are in `nodes`. z%Could not find unique first AST node.z$Could not find unique last AST node.r)nextr\rGrC)nodesZfirstsZlastsr^jr r r!first_and_last_nodess       rhcCs6tjjjD]&}|j|kr |j|kr |j|7_q dS)z This adjusts the line numbers in the the ast. `filename` The filename to adjust. `linenumber` The first line to adjust. `offset` The amount to adjust by. Positive numbers increase the line N)r,rXrYrZrr2)rr2offsetr^r r r!adjust_ast_linenumbers>srjc Cst||}t|\}}t|j|dtjjj|j||d\}}|}tjjj ||s^dStjjj D]} | ||dqhtj ||tjjD]} | ||dqtjj||ddS)zr Adds `code`, which must be a textual line of Ren'Py code, before the given filename and line number. rd)r2Nr)rcrhrjrr,rXrY load_stringpoprZremove replace_nextast chain_blockcontexts replace_nodelog) rOrr2rfold_block_initZret_stmtr^r r r!add_to_ast_beforeQs   rxcCs2zt||}t|WdSty,YdS0dS)z Returns True if it's possible to add a line before the given filename and linenumber, and False if it's not possible. TFN)r_rhrC)rr2rfr r r!can_add_beforeps   ryc Cst||}t|\}}g}tjjjD]&}||vr2q$|||j||q$|tjj_tjjj }t |D]}|||vrh|j||<qht ||ddS)z Removes from the AST all statements that happen to be at `filename` and `linenumber`, then adjusts the line numbers appropriately. There's an assumption that the statement(s) on the line are "simple", not involving control flow. N) r_rhr,rXrYrZrnrer\namemaplistrj) rr2rffirstlastZ new_stmtsr^r{kr r r!remove_from_asts        rrdcCsXdt}td7atjjtjj}|j}|j}t |||t |||tj dS)Nz'Hello world %f'rd) serialr,rXrYlookupcontextcurrentrr2rxrTexportsrestart_interaction)snoderr2r r r!test_adds  rcCsLtjjtjj}|j}|j}t||t ||tj j dddddS)NrT) checkpointsr@greedy) r,rXrYrrrrr2rrUrr[)rrr2r r r! test_removes   r)+ __future__rrrrr renpy.compatrrr r r r r rrrrrr,rDrHr0setr*objectrr/r3r9rTrUrVr_rcrhrjrxryrrrrr r r r!s.8#&5 !%