a V$c@sddlmZmZmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZddlZddlZddlZddlZddZddZdd Zd d ZGd d d ejZeZddZdS))divisionabsolute_importwith_statementprint_functionunicode_literals) PY2 basestringbchrbordchropenpystrrangeroundstrtobytesunicodeNcCsx|d}g}|D]b}|jtjkrd|jtjkrd|jdkrd|ttj|jd|j|j|j|j }| ||}q|S)z This fixes python-2 style octal numbers. Tokenize seems to report this as two numbers, the first of which has a string of '0'. This merges that with the next token. r0o) typetokenNUMBERstringpoptokenize TokenInfostartendlineappendtokensoldrvnewr%renpy/compat\fixes.pyfix_octal_numbers s"& r'cCsb|d}g}|D]L}|jtjkrN|jtjkrN|ttjd|j|j |j }| ||}q|S)z: This fixes the Python 2 spaceship operator (<>). rz!=) exact_typerLESSGREATERrrrOPrrrrr r%r%r& fix_spaceship5s r,cCsdd}g}g}|D]}|s|jtjkr4||q|jtjkrL||q|jtjkrd||q|jtjkr|||q|jtjkr||q|||jtj krq| ||g}q| |||S)z? This tries to remove Python 2-style print statements. c Sst|dkr|S|djtjkr$|S|djdkr6|S|djtjkrJ|S|djtjkrh|dd}n |dd}|d}|dt tj d|j |j |j |dt tjd|j|j|j |S)Nrprintr,)lenrrNAMErr(LPAR RIGHTSHIFTinsertrrrrrr+r)rnewliner"r%r%r&fix_lineNs  ""zfix_print..fix_line) r(rNLrINDENTDEDENT ENDMARKERENCODINGrNEWLINEextend)r!r7r#rir%r%r& fix_printIs6            r@c CszXtr |WSt|d}tt|j}t|}t|}t |}t | d}|WSt y}z|WYd}~n d}~00dS)z This applies fixes that will help python 2 code run under python 3. Not all problem will be fixed, but this will attempt to handle common issues. These are fixes that apply at the source code level. zutf-8N) rioBytesIOencodelistrreadliner'r,r@ untokenizedecode Exception)sourcebior!r#er%r%r& fix_tokenssrLc@s(eZdZdZddZddZddZdS) ReorderGlobalsz This removes all global statements from functions, and places the variables therein in a new global statement on the first line of the function. cCs t|_dSN)setglobals)selfr%r%r&__init__szReorderGlobals.__init__cCs |jD]}|j|qtSrN)namesrPaddastPass)rQnr?r%r%r& visit_Globals zReorderGlobals.visit_GlobalcCsV|j}zB||}t|j}||r>|jdtj|d|W||_S||_0dS)Nr)rS)rP generic_visitrDsortbodyr5rUGlobal)rQrW old_globals new_globalsr%r%r&visit_FunctionDefs  z ReorderGlobals.visit_FunctionDefN)__name__ __module__ __qualname____doc__rRrXr_r%r%r%r&rMsrMc CsFtr|Szt|}|WSty@}z|WYd}~n d}~00dS)z This applies fixes that will help python 2 code run under python 3. Not all problems will be fixed, but this will attempt to handle common issues. These are fixes that apply at the AST level. N)rreorder_globalsvisitrH)treerKr%r%r&fix_asts rg) __future__rrrrr renpy.compatrrr r r r r rrrrrrrrArUr'r,r@rLNodeTransformerrMrdrgr%r%r%r&s8F!