U
    ڲg                     @   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  dddg} 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/d0d*d+d1d- |jd2d3d4td5d6 |jd7d8d*d+d9d- |jd:d;d*d+d<d- |jd=d>d*d+d?d@ |jdAdBd*d+dCd@ |jdDdEdFtdGd6 |jdHdId+t	dJd6 |jdKdLd+t	dMd6 |jdNdOdPdQdR |S )SN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   0/tmp/pip-unpacked-wheel-4xfyt83m/sqlparse/cli.py	<genexpr>?   s     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   s     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	--compactcompactz2Try to produce more compact output (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rFttjj| jd}z|	 }W 5 |  X nhz.t
| j| jd}d| }W 5 Q R X W n8 tk
r } ztd| j| W Y S d }~X Y nX d}| jrzt
| jd| jd}d}W n: tk
r } ztd| j| W Y S d }~X Y nX ntj}t| }ztj|}W n6 tk
rf } ztd	| W Y S d }~X Y nX tj|f|}	||	 |  |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$   detachread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    

*$
rN   )N)__doc__r%   r2   ior   r(   Zsqlparse.exceptionsr   r0   r6   rN   r   r   r   r   <module>	   s   ~