U
e5dd @ sH d Z dZdZddlmZ ddlmZmZ ddl Z ddl
m Z ddl Z
ddlZddlZddlT dd lmZ ed
ejZed ejZddlZejd d
dddg Z[G dd de ddZdd Zdd Zdd ZdZ dZ e ede ee Z d Z d Z d Z!d Z"d Z#ee e!e"e#Z$d!Z%ed"d#ee% Z&d$e% Z'ee&e'Z(ed%e(d& Z)ee)e(e$Z*d'd( Z+d)d* Z,ee+ Z-d+Z.d,Z/d-Z0d.Z1ee-d/ e-d0 Z2ee-d1 e-d2 Z3ee4ej5e6ed3d4 Z7ed5e7Z8ee*e8e3e Z9e e9 Z:ee-d6 ed7d e-d8 ed9d Z;ed:e e2Ze+ D ]6Z?e.e>e?d7 < e/e>e?d9 < e0e>e?d/ < e1e>e?d0 < q,e@ ZAe@ ZBe+ D ]JZCeCd9 eCd7 fD ]ZDeAEeD qeCd0 eCd/ fD ]ZDeBEeD qqvd;ZFG dd? d?eGZIG d@dA dAZJdBd ZKdCdD ZLdEd ZMdFdG ZdHd ZNdIdJ ZOdKd
ZPdLdM ZQeRdNkrDeQ dS )Oao Tokenization help for Python programs.
tokenize(readline) is a generator that breaks a stream of bytes into
Python tokens. It decodes the bytes according to PEP-0263 for
determining source file encoding.
It accepts a readline-like method which is called repeatedly to get the
next line of input (or b"" for EOF). It generates 5-tuples with these
members:
the token type (see token.py)
the token (a string)
the starting (row, column) indices of the token (a 2-tuple of ints)
the ending (row, column) indices of the token (a 2-tuple of ints)
the original line (string)
It is designed to match the working of the Python tokenizer exactly, except
that it produces COMMENT tokens for comments and gives type OP for all
operators. Additionally, all token lists start with an ENCODING token
which tells you which encoding was used to decode the bytes stream.
zKa-Ping Yee zpGvR, ESR, Tim Peters, Thomas Wouters, Fred Drake, Skip Montanaro, Raymond Hettinger, Trent Nelson, Michael Foord )open)lookupBOM_UTF8N)
TextIOWrapper)*)EXACT_TOKEN_TYPESz&^[ \t\f]*#.*?coding[:=][ \t]*([-\w.]+)s ^[ \t\f]*(?:[#\r\n]|$)tokenizegenerate_tokensdetect_encoding
untokenize TokenInfoc @ s e Zd Zdd Zedd ZdS )r c C s$ d| j t| j f }d| j|d S )Nz%d (%s)z8TokenInfo(type=%s, string=%r, start=%r, end=%r, line=%r))type)r
tok_name_replace)selfannotated_type r /usr/lib64/python3.8/tokenize.py__repr__. s
zTokenInfo.__repr__c C s( | j tkr | jtkr t| j S | j S d S N)r
OPstringr r r r r
exact_type3 s
zTokenInfo.exact_typeN)__name__
__module__ __qualname__r propertyr r r r r r - s ztype string start end linec G s dd | d S )N(|))joinchoicesr r r group: r$ c G s t | d S )Nr r$ r" r r r any; r% r' c G s t | d S )N?r& r" r r r maybe< r% r) z[ \f\t]*z #[^\r\n]*z\\\r?\nz\w+z0[xX](?:_?[0-9a-fA-F])+z0[bB](?:_?[01])+z0[oO](?:_?[0-7])+z (?:0(?:_?0)*|[1-9](?:_?[0-9])*)z[eE][-+]?[0-9](?:_?[0-9])*z)[0-9](?:_?[0-9])*\.(?:[0-9](?:_?[0-9])*)?z\.[0-9](?:_?[0-9])*z[0-9](?:_?[0-9])*z[0-9](?:_?[0-9])*[jJ]z[jJ]c C s^ ddddddg} dh}| D ]>}t |D ].}t jdd |D D ]}|d| q@q(q|S )
Nbrufbrfr c S s g | ]}|| fqS r )upper).0cr r r
^ s z(_all_string_prefixes..)
_itertools permutationsproductaddr! )_valid_string_prefixesresultprefixtr, r r r _all_string_prefixesS s r= c C s t | t jS r )recompileUNICODE)exprr r r _compileb s rB z[^'\\]*(?:\\.[^'\\]*)*'z[^"\\]*(?:\\.[^"\\]*)*"z%[^'\\]*(?:(?:\\.|'(?!''))[^'\\]*)*'''z%[^"\\]*(?:(?:\\.|"(?!""))[^"\\]*)*"""z'''z"""z '[^\n'\\]*(?:\\.[^\n'\\]*)*'z "[^\n"\\]*(?:\\.[^\n"\\]*)*"T)reversez\r?\nz'[^\n'\\]*(?:\\.[^\n'\\]*)*'z"[^\n"\\]*(?:\\.[^\n"\\]*)*"z
\\\r?\n|\Z c @ s e Zd ZdS )
TokenErrorNr r r r r r r rG s rG c @ s e Zd ZdS )StopTokenizingNrH r r r r rI s rI c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
Untokenizerc C s g | _ d| _d| _d | _d S )N r )tokensprev_rowprev_colencodingr r r r __init__ s zUntokenizer.__init__c C s |\}}|| j k s&|| j kr>|| jk r>td||| j | j|| j }|rb| jd| d| _|| j }|r| jd| d S )Nz+start ({},{}) precedes previous end ({},{})\
r )rM rN
ValueErrorformatrL append)r startrowcol
row_offset
col_offsetr r r add_whitespace s
zUntokenizer.add_whitespacec C s6 t |}g }d}|D ]}t|dkr8| || q*|\}}}} }
|tkrV|| _q|tkrd q*|tkrz|| qnl|tkr| | \| _
| _ qnL|t t
fkrd}n:|r|r|d } |d t| kr| j| t| | _ d}| | | j| | \| _
| _ |t t
fkr| j
d7 _
d| _ qd| jS )NF TrK r r0 )iterlencompatENCODINGrO ENDMARKERINDENTrU DEDENTpoprM rN NEWLINENLrL r[ r! ) r iterableitindents startliner< tok_typetokenrV endlineindentr r r r sF
zUntokenizer.untokenizec
C s g }| j j}|d ttfk}d}t|g|D ]}|d d \}} |tkrR| | _q.|tt fkrf| d7 } |t
kr|rzd| } d}nd}|t kr|| q.n>|t kr|
q.n*|ttfkrd}n|r|r||d d}|| q.d S )Nr Fr\ rR Tr] )rL rU rf rg r5 chainra rO NAMENUMBERSTRINGrc rd re )
r rm rh rj toks_appendrk
prevstringtoktoknumtokvalr r r r` s8
zUntokenizer.compatN)r r r rP r[ r r` r r r r rJ s
%rJ c C s* t }|| }|jdk r&||j}|S )a Transform tokens back into Python source code.
It returns a bytes object, encoded using the ENCODING
token, which is the first token sequence output by tokenize.
Each element returned by the iterable must be a token sequence
with at least two elements, a token number and token value. If
only two tokens are passed, the resulting output is poor.
Round-trip invariant for full input:
Untokenized source will match input source exactly
Round-trip invariant for limited input:
# Output bytes will tokenize back to the input
t1 = [tok[:2] for tok in tokenize(f.readline)]
newcode = untokenize(t1)
readline = BytesIO(newcode).readline
t2 = [tok[:2] for tok in tokenize(readline)]
assert t1 == t2
N)rJ r rO encode)rh utoutr r r r s
c C sH | dd dd}|dks*|dr.dS |dks@|drDd S | S )
z(Imitates get_normal_name in tokenizer.c.N _-utf-8zutf-8-)zlatin-1
iso-8859-1z iso-latin-1)zlatin-1-z iso-8859-1-z iso-latin-1-r )lowerreplace
startswith)orig_encencr r r _get_normal_name s r c s z j jW n tk
r$ dY nX d d}d}fdd} fdd}| }|trpd |d d }d
}|s||g fS ||}|r||gfS t|s||gfS | }|s||gfS ||}|r|||gfS |||gfS ) a
The detect_encoding() function is used to detect the encoding that should
be used to decode a Python source file. It requires one argument, readline,
in the same way as the tokenize() generator.
It will call readline a maximum of twice, and return the encoding used
(as a string) and a list of any lines (left as bytes) it has read in.
It detects the encoding from the presence of a utf-8 bom or an encoding
cookie as specified in pep-0263. If both a bom and a cookie are present,
but disagree, a SyntaxError will be raised. If the encoding cookie is an
invalid charset, raise a SyntaxError. Note that if a utf-8 bom is found,
'utf-8-sig' is returned.
If no encoding is specified, then the default of 'utf-8' will be returned.
NFr c s$ z W S t k
r Y dS X d S )Nr% )
StopIterationr readliner r read_or_stop? s z%detect_encoding..read_or_stopc s z| d}W n4 tk
rB d}d k r6d|}t|Y nX t|}|sVd S t|d}z t|}W n: t k
r d krd| }n d|}t|Y nX r|dkr؈d krd}n
d}t||d 7 }|S )
Nr z'invalid or missing encoding declarationz {} for {!r}rK zunknown encoding: z unknown encoding for {!r}: {}zencoding problem: utf-8z encoding problem for {!r}: utf-8z-sig)
decodeUnicodeDecodeErrorrT SyntaxError cookie_rematchr r$ r LookupError)ro line_stringmsgr rO codec) bom_foundfilenamer r find_cookieE s8
z$detect_encoding..find_cookieT utf-8-sig)__self__nameAttributeErrorr r blank_rer )r rO defaultr r firstsecondr )r r r r r
' s8
&
c C sX t | d}z2t|j\}}|d t||dd}d|_|W S | Y nX dS )zXOpen a file in read only mode using the encoding detected by
detect_encoding().
rbr T)line_bufferingr+ N)
_builtin_openr
r seekr modeclose)r bufferrO linestextr r r r s
r c C s6 t | \}}td}t|t| d|}t|j|S )a
The tokenize() generator requires one argument, readline, which
must be a callable object which provides the same interface as the
readline() method of built-in file objects. Each call to the function
should return one line of input as bytes. Alternatively, readline
can be a callable function terminating with StopIteration:
readline = open(myfile, 'rb').__next__ # Example of alternate readline
The generator produces 5-tuples with these members: the token type; the
token string; a 2-tuple (srow, scol) of ints specifying the row and
column where the token begins in the source; a 2-tuple (erow, ecol) of
ints specifying the row and column where the token ends in the source;
and the line on which the token was found. The line passed is the
physical line.
The first token sequence will always be an ENCODING token
which tells you which encoding was used to decode the bytes stream.
r% )r
r5 repeatrq r^ _tokenize__next__)r rO consumedemptyrl_genr r r r s
c c sl d } }}d}d\}}d }dg} |d k rH|dkr6d}t t|dddV d}
d} z| }
| } W n tk
rv d} Y nX |d k r| |} |d 7 }dt| } }
|rp| std
||| }|r|d } }t t|| d | |||f|| V d\}}d }nf|rZ| d d d krZ| d
d dkrZt t || ||t| f|V d}d }qPn|| }|| }qPn|dkr|s| sqd}| |
k r| | dkr|d 7 }n8| | dkr|t
d t
}n| | dkrd}nq| d 7 } q| |
krq| | dkr| | dkr^| | d d}t t ||| f|| t| f| V | t|7 } t t
| | d || f|t| f| V qP|| d kr| | t t| d | |df|| f| V || d k r.|| krtdd|| | f| d d } t td|| f|| f| V qn| s*td|dfd}| |
k rPtt| | }|r|d \}}||f||f| }}} ||krq.| || | | }}||ks|dkr|dkr|dkrt t|||| V q|dkr|dkrt t
|||| V nt t|||| V q|dkr2t t |||| V q|tkrtt| }|| | }|r|d} | || }t t|||| f| V n||f}| |d }| }qPq|tks|d d tks|d d tkrF|d d kr2||f}tt|pt|d pt|d }| |d d }}| }qPnt t|||| V nf| rdt t |||| V nH|d krtd }n8|d kr|d 7 }n|d kr|d 8 }t t |||| V n*t t | | || f|| d f| V | d 7 } q.qP|
r |
d dkr t td|d t|
f|d t|
d fdV | d d D ] }t td|df|dfdV q,t t d|df|dfdV d S )!Nr
0123456789)r0 r r r )r r r0 r% rK zEOF in multi-line stringrQ z\
rR z#
#z
r] z3unindent does not match any outer indentation levelz
zEOF in multi-line statement.z...r\ r
\z([{z)]}) r ra r r r_ rG r rn rt
ERRORTOKENtabsizerstripCOMMENTrg rU rc IndentationErrorrd rB PseudoTokenspanrs rf
triple_quotedendpats
single_quotedget isidentifierrr r rb )r rO lnumparenlev continuednumcharscontstrneedcontcontlinerj last_linero posmaxstrstartendprogendmatchrn column
comment_token pseudomatchrV sposeposrm initialrp r r r r s<
*
"
. r c C s
t | dS )zTokenize a source reading Python code as unicode strings.
This has the same API as tokenize(), except that it expects the *readline*
callable to return str objects instead of bytes.
N)r r r r r r d s c
s$ dd l } dd d fdd }| jdd}|jdd d
d d |jd
ddddd | }z|jr|j}t|d}tt|j}W 5 Q R X nd}t t
j jd }|D ]>}|j }|j
r|j}d|j|j } td| t| |jf qW n8 tk
r6 }
z0|
jd dd \} } ||
jd || | f W 5 d }
~
X Y n tk
r| }
z(|
jd \} } ||
jd || | f W 5 d }
~
X Y n tk
r }
z||
| W 5 d }
~
X Y nx tk
r }
z ||
W 5 d }
~
X Y nN tk
r td Y n2 tk
r }
z d|
W 5 d }
~
X Y nX d S ) Nr c S s t j| t jd d S )Nr )sysstderrwrite)messager r r perrorp s zmain..perrorc sR |r"|f| | f } d| n"|r8 d|| f n d| t d d S )Nz%s:%d:%d: error: %sz
%s: error: %sz error: %srK )r exit)r r locationargsr r r errort s zmain..errorzpython -m tokenize)progr r( z filename.pyz'the file to tokenize; defaults to stdin)destnargsmetavarhelpz-ez--exactexact
store_truez(display token names using the exact type)r actionr r zz %d,%d-%d,%d:z%-20s%-15s%-15rrK r z interrupted
zunexpected error: %s)NN)argparseArgumentParser add_argument
parse_argsr r listr r r r stdinr
r r rV rn printr r r r rG r OSErrorKeyboardInterrupt Exception)
r r parserr r r- rL rm
token_type token_rangeerrro r r r r mainl sT && r __main__)S__doc__
__author__ __credits__builtinsr r codecsr r collectionsior itertoolsr5 r> r rm r r? ASCIIr r __all__
namedtupler r$ r' r)
WhitespaceCommentIgnoreName Hexnumber Binnumber Octnumber Decnumber IntnumberExponent
PointfloatExpfloat Floatnumber
ImagnumberNumberr= rB StringPrefixSingleDoubleSingle3Double3TripleStringmapescapesortedSpecialFunny
PlainTokenTokenContStr PseudoExtrasr r _prefixsetr r r< r, r8 r r rG rI rJ r r r
r r r r r r r r r s
_ ] 8=