B \ @szdZddlZddlmZmZddlZddlmZmZedej Z edej Z dd d Z d d Z dd dZdddZdS)z Tools to open .py files as Unicode, using the encoding specified within the file, as per PEP 263. Much of the code is taken from the tokenize module in Python 3.2. N) TextIOWrapperBytesIO)opendetect_encodingzcoding[:=]\s*([-\w.]+)z^\s*#.*coding[:=]\s*([-\w.]+)replaceTcCst|tr|St|tr"t|}n|}yt|j\}}Wntk rPd}YnX|dt|||dd}d|_ |rd t |S| SdS)a#Converts a bytes string with python source code to unicode. Unicode strings are passed through unchanged. Byte strings are checked for the python source file encoding cookie to determine encoding. txt can be either a bytes buffer or a string containing the source code. asciirT)errorsline_bufferingrN) isinstancestrbytesrrreadline SyntaxErrorseekrmodejoinstrip_encoding_cookieread)Ztxtrskip_encoding_cookiebufferencoding_textr3lib/python3.7/site-packages/IPython/utils/openpy.pysource_to_unicodes     rccsjt|}y4t|}t|s"|Vt|}t|s:|VWntk rPdSXx|D] }|VqXWdS)z{Generator to pull lines from a text-mode file, skipping the encoding cookie if it is found in the first two lines. N)iternextcookie_comment_rematch StopIteration)Zfilelikeitfirstsecondlinerrrr*s    rc Cs2t| }|rdt|S|SWdQRXdS)aRead a Python file, using the encoding declared inside the file. Parameters ---------- filename : str The path to the file to read. skip_encoding_cookie : bool If True (the default), and the encoding declaration is found in the first two lines, that line will be excluded from the output - compiling a unicode string with an encoding declaration is a SyntaxError in Python 2. Returns ------- A unicode string containing the contents of the file. r N)rrrr)filenamerfrrr read_py_file<s r)cCs.ddlm}||}t|}t|||S)aRead a Python file from a URL, using the encoding declared inside the file. Parameters ---------- url : str The URL from which to fetch the file. errors : str How to handle decoding errors in the file. Options are the same as for bytes.decode(), but here 'replace' is the default. skip_encoding_cookie : bool If True (the default), and the encoding declaration is found in the first two lines, that line will be excluded from the output - compiling a unicode string with an encoding declaration is a SyntaxError in Python 2. Returns ------- A unicode string containing the contents of the file. r)urlopen)Zurllib.requestr*iorrr)Zurlrrr*Zresponserrrr read_py_urlRs r,)rT)T)rT)__doc__r+rrretokenizerrcompileUNICODE cookie_rer rrr)r,rrrrs