a
    4XhP                     @   sR   d Z ddlZddlZddlmZ ddlZddlmZ dd Zdd Z	dd	d
Z
dS )ae  Module that contains the command line app.

Why does this file exist, and why not put this in __main__?
  You might be tempted to import things from __main__ later, but that will
  cause problems: the code will get executed twice:
  - When you run `python -m sqlparse` python will execute
    ``__main__.py`` as a script. That means there won't be any
    ``sqlparse.__main__`` in ``sys.modules``.
  - When you import __main__ it will get executed again (as a module) because
    there's no ``sqlparse.__main__`` in ``sys.modules``.
  Also see (1) from http://click.pocoo.org/5/setuptools/#setuptools-integration
    N)TextIOWrapper)SQLParseErrorc                  C   s  g d} t jdddd}|d |jddd	d
dd |jddtjd |d}|jdddd| dddd | D d |jdddd| dddd | D d |jdd d!d"d#d$gd%d |jd&d'd(d)d*d+ |jd,d-d.d(d)d/d+ |jd0d1d2td3d4 |jd5d6d(d)d7d+ |jd8d9d(d)d:d+ |jd;d<d(d)d=d> |jd?d@d(d)dAd> |jdBdCdDtdEd4 |jdFdGd)t	dHd4 |jdIdJdKdLdM |S )NN)upperlower
capitalizeZ	sqlformatzEFormat FILE according to OPTIONS. Use "-" as FILE to read from stdin.z%(prog)s  [OPTIONS] FILE, ...)progdescriptionusagefilenamez-oz	--outfileoutfileFILEz)write output to FILE (defaults to stdout))destmetavarhelpz	--versionversion)actionr   zFormatting Optionsz-kz
--keywordsZCHOICEZkeyword_casez,change case of keywords, CHOICE is one of {}z, c                 s   s   | ]}d  |V  qdS z"{}"Nformat.0x r   Q/var/www/viveiro_mudafortebrasil/venv/lib/python3.9/site-packages/sqlparse/cli.py	<genexpr>?       z create_parser.<locals>.<genexpr>)r   r   choicesr   z-iz--identifiersZidentifier_casez/change case of identifiers, CHOICE is one of {}c                 s   s   | ]}d  |V  qdS r   r   r   r   r   r   r   G   r   z-lz
--languageLANGZoutput_formatpythonZphpzJoutput a snippet in programming language LANG, choices are "python", "php"z--strip-commentsstrip_comments
store_trueFzremove comments)r   r   defaultr   z-rz
--reindentZreindentzreindent statementsz--indent_widthZindent_width   z(indentation width (defaults to 2 spaces))r   r!   typer   z--indent_after_firstZindent_after_firstz2indent after first line of statement (e.g. SELECT)z--indent_columnsZindent_columnsz<indent all columns by indent_width instead of keyword lengthz-az--reindent_alignedz%reindent statements to aligned format)r   r!   r   z-sz--use_space_around_operatorsz*place spaces around mathematical operatorsz--wrap_afterZ
wrap_afterr   z*Column after which lists should be wrappedz--comma_firstZcomma_firstz-Insert linebreak before comma (default False)z
--encodingencodingzutf-8z*Specify the input encoding (default utf-8))r   r!   r   )
argparseArgumentParseradd_argumentsqlparse__version__add_argument_groupr   joinintbool)Z_CASE_CHOICESparsergroupr   r   r   create_parser    s    

r0   c                 C   s   t jd|  dS )z5Print msg and optionally exit with return code exit_.z[ERROR] {}
   )sysstderrwriter   )msgr   r   r   _error   s    r6   c           
   
   C   s  t  }|| } | jdkrNttjj| jd}z| }W |	  q|	  0 nzzBt
| j| jd}d| }W d    n1 s0    Y  W n6 ty } ztd| j|W  Y d }~S d }~0 0 d}| jr*zt
| jd| jd}d}W n8 ty& } ztd| j|W  Y d }~S d }~0 0 ntj}t| }ztj|}W n4 ty| } ztd	|W  Y d }~S d }~0 0 tj|fi |}	||	 |  |r|  d
S )N-)r$    zFailed to read {}: {}FwTzFailed to open {}: {}zInvalid options: {}r   )r0   
parse_argsr
   r   r2   stdinbufferr$   readdetachopenr+   	readlinesOSErrorr6   r   r   stdoutvarsr(   	formatterZvalidate_optionsr   r4   flushclose)
argsr.   wrapperdatafeZclose_streamstreamZformatter_optssr   r   r   main   sB    


0*$
rN   )N)__doc__r%   r2   ior   r(   Zsqlparse.exceptionsr   r0   r6   rN   r   r   r   r   <module>	   s   w