o bc!@sdZdZddgZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z GdddeZd d Z  dddZddZddZddZedkrXeee dSdS)zZParser driver. This provides a high-level interface to parse a file into a syntax tree. z#Guido van Rossum Driver load_grammarN)grammarparsetokentokenizepgenc@sHeZdZdddZdddZdddZdd d Zdd d Zdd dZdS)rNcCs&||_|dur t}||_||_dS)N)rlogging getLoggerloggerconvert)selfrr r r+/usr/lib/python3.10/lib2to3/pgen2/driver.py__init__s  zDriver.__init__FcCst|j|j}|d}d}d}}}} } d} |D]} | \}}}} } |||fkr]||f|ks;J||f|f|\} }|| krO| d| |7} | }d}||kr]| | ||7} |}|tjtjfvry| |7} | \}}|drx|d7}d}q|t j krtj |}|r|j dt j||| |||| |fr|r|j d|jSd} | \}}|dr|d7}d}qtd||| |f) z4Parse a series of tokens and return the syntax tree.rrN z%s %r (prefix=%r)zStop.zincomplete input)rParserrr setuprCOMMENTNLendswithrOPopmapr debugtok_nameaddtoken ParseErrorrootnode)rtokensrplinenocolumntypevaluestartend line_textprefix quintuples_linenos_columnrrr parse_tokens&sZ         zDriver.parse_tokenscCst|j}|||Sz*Parse a stream and return the syntax tree.)rgenerate_tokensreadliner-)rstreamrr rrrparse_stream_rawVs  zDriver.parse_stream_rawcCs |||Sr.)r2)rr1rrrr parse_stream[s zDriver.parse_streamcCs@tj|d|d}|||WdS1swYdS)z(Parse a file and return the syntax tree.r)encodingN)ioopenr3)rfilenamer5rr1rrr parse_file_s $zDriver.parse_filecCstt|j}|||S)z*Parse a string and return the syntax tree.)rr/r6StringIOr0r-)rtextrr rrr parse_stringds zDriver.parse_string)NN)F)NF) __name__ __module__ __qualname__rr-r2r3r9r<rrrrrs   0  cCs:tj|\}}|dkrd}||dtttjdS)Nz.txtr.z.pickle)ospathsplitextjoinmapstrsys version_info)gtheadtailrrr_generate_pickle_namejsrL Grammar.txtTFcCsd|durt}|durt|n|}|st||s'|d|t|} |St }| ||S)z'Load the grammar (maybe from a pickle).Nz!Generating grammar tables from %s) r r rL_newerinfor generate_grammardumpOSErrorrGrammarload)rIgpsaveforcer gerrrrqs    cCs8tj|sdStj|sdStj|tj|kS)z0Inquire whether file a was written since file b.FT)rArBexistsgetmtime)abrrrrNs  rNcCsFtj|r t|Sttj|}t||}t }| ||S)aNormally, loads a pickled grammar by doing pkgutil.get_data(package, pickled_grammar) where *pickled_grammar* is computed from *grammar_source* by adding the Python version and using a ``.pickle`` extension. However, if *grammar_source* is an extant file, load_grammar(grammar_source) is called instead. This facilitates using a packaged grammar file when needed but preserves load_grammar's automatic regeneration behavior when possible. ) rArBisfilerrLbasenamepkgutilget_datarrSloads)packagegrammar_source pickled_namedatarXrrrload_packaged_grammars   rgcGsB|s tjdd}tjtjtjdd|D] }t|dddqdS)zMain program, when run as a script: produce grammar pickle files. Calls load_grammar for each argument, a path to a grammar text file. rNz %(message)s)levelr1formatT)rVrW)rGargvr basicConfigINFOstdoutr)argsrIrrrmains ro__main__)rMNTFN)__doc__ __author____all__r6rAr r`rGrrrrrr objectrrLrrNrgror=exitintrrrrs(M