a V$c.@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 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) add_lines)endswithfilesaddrenpyparserunelide_filenamelist_logical_lines)rfnr r r! ensure_loadedMs  r2cCs6|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)rr2linesrr linenumberr r r! get_line_text_s   r6c 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)rr2r3itemsrrrr) rr5 char_offset line_offset new_lineskeyliner1lnr r r!adjust_line_locationsos r>c 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. rrcountloader auto_lockwriteadd_auto) coderr5old_linemindent line_endingraw_codenew_linefdatar r r!insert_line_befores8    &  *2r]c Cs4|dd}tjjrtdt|t||f}t|j dd}| }Wdn1s^0Y||j |j }|d|j ||j d}t||f=t ||t| |d tjj\t|j dd}||Wdn1s0Ytjj|j d d Wdn1s&0YdS) zi Removes `linenumber` from `filename`. The line must exist and correspond to a logical line. rrr?rArBNr@rCTrD)rr-rFrGrHr2r3rMr rrNrrr>rLrOrPrQrRrS)rr5r<r[r\rTr r r! remove_lines  &  (r^cCs2|dd}t|||ftvr$dSt||fjS)z~ Returns the full text of `linenumber` from `filename`, including any comment or delimiter characters that exist. rrN)rr2r3rr4r r r! get_full_texts   r_cCsHt|g}tjjjD],}|j|kr|j|kr|jdkr||q|S)zC Returns a list of nodes that are found on the given line. never) r2r-gamescript all_stmtsrr5rollbackappend)rr5rvir r r! nodes_on_lines  rhcs8tfddtjjjD}|s*gStt|S)zL Returns a list of nodes that are found at or after the given line. cs0g|](}|jkr|jkr|jdkr|jqS)r`)rr5rd).0rgr4r r! s    z-nodes_on_line_at_or_after..)r2r-rarbrcrhmin)rr5r3r r4r!nodes_on_line_at_or_after s rlcCsg}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)nextrerLrH)nodesfirstslastsrgjr r r!first_and_last_nodess       rscCs6tjjjD]&}|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-rarbrcrr5)rr5offsetrgr r r!adjust_ast_linenumbers>sruc 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. rm)r5Nr)rlrsrurr-rarb load_stringpoprcremove replace_nextast chain_blockcontexts replace_nodelog) rTrr5roold_block_initret_stmtrgr r r!add_to_ast_beforeQs   rcCs2zt||}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)rhrsrH)rr5ror r r!can_add_beforeps   rc 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) rhrsr-rarbrcryrnrenamemaplistru) rr5rofirstlast new_stmtsrgrkr r r!remove_from_asts        rrmcCsXdt}td7atjjtjj}|j}|j}t |||t |||tj dS)Nz'Hello world %f'rm) serialr-rarblookupcontextcurrentrr5rr]exportsrestart_interaction)snoderr5r r r!test_adds  rcCsLtjjtjj}|j}|j}t||t ||tj j dddddS)NrT) checkpointsrEgreedy) r-rarbrrrrr5rr^rrd)rrr5r r r! test_removes   r)+ __future__rrrrr renpy.compatrrr r r r r rrrrrr-rIrMr3setr+objectrr2r6r>r]r^r_rhrlrsrurrrrrrr r r r!s.8#&5 !%