3
� �X~K � @ s d Z ddlmZ ddlZddlZddlmZ ddlmZm Z ddl
m Z m Z m
Z
mZmZmZ ddlmZmZmZmZmZmZ ddlmZ dd lmZm Z dd
l m Z m Z m Z m!Z!m"Z" dd l#m$Z$ dd l%m&Z&m'Z' dd
l(m)Z)m*Z* dZ+dd� Z,dd� Z-dd� Z.dd� Z/dd� Z0ej1fdd�Z2dS )z�
pygments.cmdline
~~~~~~~~~~~~~~~~
Command line interface.
:copyright: Copyright 2006-2017 by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
� )�print_functionN)�dedent)� __version__� highlight)�
ClassNotFound� OptionError�docstring_headline� guess_decode�guess_decode_from_terminal�terminal_encoding)�get_all_lexers�get_lexer_by_name� guess_lexer�load_lexer_from_file�get_lexer_for_filename� find_lexer_class_for_filename)� TextLexer)�LatexEmbeddedLexer�LatexFormatter)�get_all_formatters�get_formatter_by_name�load_formatter_from_file�get_formatter_for_filename�find_formatter_class)�TerminalFormatter)�get_all_filters�find_filter_class)�get_all_styles�get_style_by_namea� Usage: %s [-l <lexer> | -g] [-F <filter>[:<options>]] [-f <formatter>]
[-O <options>] [-P <option=value>] [-s] [-v] [-x] [-o <outfile>] [<infile>]
%s -S <style> -f <formatter> [-a <arg>] [-O <options>] [-P <option=value>]
%s -L [<which> ...]
%s -N <filename>
%s -H <type> <name>
%s -h | -V
Highlight the input file and write the result to <outfile>.
If no input file is given, use stdin, if -o is not given, use stdout.
If -s is passed, lexing will be done in "streaming" mode, reading and
highlighting one line at a time. This will only work properly with
lexers that have no constructs spanning multiple lines!
<lexer> is a lexer name (query all lexer names with -L). If -l is not
given, the lexer is guessed from the extension of the input file name
(this obviously doesn't work if the input is stdin). If -g is passed,
attempt to guess the lexer from the file contents, or pass through as
plain text if this fails (this can work for stdin).
Likewise, <formatter> is a formatter name, and will be guessed from
the extension of the output file name. If no output file is given,
the terminal formatter will be used by default.
The additional option -x allows custom lexers and formatters to be
loaded from a .py file relative to the current working directory. For
example, ``-l ./customlexer.py -x``. By default, this option expects a
file with a class named CustomLexer or CustomFormatter; you can also
specify your own class name with a colon (``-l ./lexer.py:MyLexer``).
Users should be very careful not to use this option with untrusted files,
because it will import and run them.
With the -O option, you can give the lexer and formatter a comma-
separated list of options, e.g. ``-O bg=light,python=cool``.
The -P option adds lexer and formatter options like the -O option, but
you can only give one option per -P. That way, the option value may
contain commas and equals signs, which it can't with -O, e.g.
``-P "heading=Pygments, the Python highlighter".
With the -F option, you can add filters to the token stream, you can
give options in the same way as for -O after a colon (note: there must
not be spaces around the colon).
The -O, -P and -F options can be given multiple times.
With the -S option, print out style definitions for style <style>
for formatter <formatter>. The argument given by -a is formatter
dependent.
The -L option lists lexers, formatters, styles or filters -- set
`which` to the thing you want to list (e.g. "styles"), or omit it to
list everything.
The -N option guesses and prints out a lexer name based solely on
the given filename. It does not take input or highlight anything.
If no specific lexer can be determined "text" is returned.
The -H option prints detailed help for the object <name> of type <type>,
where <type> is one of "lexer", "formatter" or "filter".
The -s option processes lines one at a time until EOF, rather than
waiting to process the entire file. This only works for stdin, and
is intended for streaming input such as you get from 'tail -f'.
Example usage: "tail -f sql.log | pygmentize -s -l sql"
The -v option prints a detailed traceback on unhandled exceptions,
which is useful for debugging and bug reports.
The -h option prints this help.
The -V option prints the package version.
c
C s� i }| s |S x�| D ]x}|j � s q|jd�}x^|D ]V}|j � }y$|jdd�\}}|j � }|j � }W n tk
r| d||< Y q0X |||< q0W qW |S )N�,�=� T)�strip�split�
ValueError)Zo_strs�optsZo_strZo_argsZo_argZo_keyZo_val� r& � /usr/lib/python3.6/cmdline.py�_parse_optionsn s"
r( c C sZ g }| s |S xH| D ]@}d|krD|j dd�\}}|j|t|g�f� q|j|i f� qW |S )N�:r! )r# �appendr( )Zf_strs�filtersZf_str�fname�foptsr& r& r' �_parse_filters� s
r. c C s� y�| dkr0t |�}td|j � tt|j�� nX| dkr^t|�}td|j � tt|j�� n*| dkr�t|�}td| � tt|j�� dS ttfk
r� td| t j
d � d
S X d S ) N�lexerzHelp on the %s lexer:� formatterzHelp on the %s formatter:�filterzHelp on the %s filter:r z
%s not found!)�filer! ) r
�print�namer �__doc__r r �AttributeErrorr$ �sys�stderr)�whatr4 �clsr& r& r' � _print_help� s" r; c C s� | dkr�t � t d� t d� g }xJt� D ]@\}}}}dj|�d ||rZddj|� d p\df}|j|� q*W |j� x|D ]}t d | � q|W �nR| d
k�r0t � t d � t d � g }xPt� D ]F}t|�} dj|j�d | |jr�ddj|j� d p�df}|j|� q�W |j� x�|D ]}t d | � �qW n�| d
k�r�t � t d� t d� x�t � D ].}
t
|
�}t d|
d � t dt|� � �qXW nZ| dk�r�t � t d� t d� x8t � D ].}
t |
�}t d|
d � t dt|� � �q�W d S )Nr/ zLexers:z~~~~~~~z, r) z (filenames �)� z* %s
%s %sr0 z Formatters:z ~~~~~~~~~~~r1 zFilters:z~~~~~~~~z* z %s�stylezStyles:)
r3 r �joinr* �sortr r �aliases� filenamesr r r r ) r9 �info�fullname�namesZexts�_�tup�ir: �docr4 r& r& r' � _print_list� sR
rJ c * { C s� i }g }g }g }xR| D ]J\}}|dkr2|j |� n&|dkrF|j |� n|dkrX|j |� |||< qW |jdd �d k r�t|� dS |jdd �d k r�tdt � dS |jdd �} | d k �r|r�t|tjd � d
S td dg� |s�d d
ddg}x|D ]}t|jd�� q�W dS |jdd �}
|
d k �rr|�s2t |�d
k�rDt|tjd � d
S |\} } | d5k�rht|tjd � d
S t
| | �S t |�}
|jdd � xJ|D ]B}y|j dd�\} }W n t
k
�r� d|
|< Y n
X ||
| < �q�W |jdd � |
jd|
jd��}|
jd|
jd��}|jdd �}|d k �r@t|�}|d k�r.t}t|jd � dS |jdd �}|jdd �}|d k �r|jdd �}|�s�t|tjd � d
S |�s�|�r�t|tjd � d
S y||
d< t|f|
�}W n0 tk
�r� } zt|tjd � dS d }~X nX t|j|�p�d �� dS |d k �r"t|tjd � d
S t|�}|jdd � d }|jd d �d k �rPd}d }|jd d �}|�r.|�r�d |k�r�y:d |k�r�|jd d�\}} t|| f|
�}n t|f|
�}W n2 tk
�r� } ztd!|tjd � dS d }~X nX nHyt|f|
�}W n6 ttfk
�r, } ztd!|tjd � dS d }~X nX d } |�r�t |�dk�rXt|tjd � d
S d"|k�rttd#tjd � d
S |d }y"t|d$��} | j� } W d Q R X W n2 t k
�r� } ztd%|tjd � dS d }~X nX |�s�t | �\} }|�s<yt || f|
�}W n� tk
�rr } zXd&|k�rNyt | f|
�}W n tk
�rJ tf |
�}Y nX ntd!|tjd � dS W Y d d }~X n2 tk
�r� } ztd!|tjd � dS d }~X nX n�d"|k�r$tj d6k�r�tj!j"j� } n
tj!j� } |�s�t#| tj!�\} }|�s<yt | f|
�}W n tk
�r tf |
�}Y nX n|�s<td(tjd � d
S xX|D ]P\} } y|j$| f| � W n2 tk
�r� } ztd!|tjd � dS d }~X nX �qBW |jd)d �} |jdd �}|�r||�r4d |k�r4y:d |k�r�|jd d�\}!}"t%|!|"f|
�}n t%|f|
�}W n2 tk
�r0 } ztd!|tjd � dS d }~X nX nHyt|f|
�}W n6 ttfk
�rz } ztd!|tjd � dS d }~X nX | �r|�s�yt&| f|
�}W n6 ttfk
�r� } ztd!|tjd � dS d }~X nX yt| d*�}#W n2 t k
�r } ztd+|tjd � dS d }~X nX n,|�s$t'f |
�}tj d7k�r:tj(j"}#ntj(}#|�s`| �rT||_)n t*tj(�|_)| �r�tj+d8k�r�|j,d9k�r�tj d:k�r�dd0l-m.}$ |$|#|j)d1�}#d |_)y dd l/}%W n t0k
�r� Y nX |%j1j2|#d d d dd2�}#|
jd3d �}&t |&�d
k� r(t3|t4�� r(|&d }'|&d }(t5|'|(|�}d"|k� rDt6| |||#� dS ypxjtj d;k� rbtj!j"j7� })n
tj!j7� })|)� stP |� s�t#|)tj!�d })t6|)|||#� t8|#d4�� rH|#j9� � qHW dS t:k
� r� dS X d S )<Nz-Oz-Pz-Fz-hr z-Vz3Pygments version %s, (c) 2006-2017 by Georg Brandl.z-L)r2 � r= r/ r0 r1 r> �sz-Hr r! T�
inencoding�encoding� outencodingz-Nz-Sz-az-fFz-xz-lz.pyr) zError:z-sz5Error: -s option not usable when input file specified�rbzError: cannot read infile:z-g� z7Error: when using -s a lexer has to be selected with -lz-o�wbzError: cannot open outfile:�win32�cygwin�Terminal� Terminal256)�UnclosingTextIOWrapper)rN )Zconvertr" Z autoresetZwrap� escapeinside�flush)r/ r0 r1 )rQ )rQ )rS rT )rU rV )rQ )rQ );r* �popr3 r r7 r8 �mainrJ �rstrip�lenr; r( r# r$ �getr r rA r r Zget_style_defsr. �rsplitr r
r �open�read� Exceptionr r r � version_info�stdin�bufferr
Z
add_filterr r r �stdoutrN r �platformr4 �
pygments.utilrW Zcolorama.initialise� ImportErrorZ
initialiseZ wrap_stream�
isinstancer r r �readline�hasattrrY �KeyboardInterrupt)*�popts�args�usager% ZO_optsZP_optsZF_opts�opt�argZL_optZH_optr9 r4 Z parsed_optsZp_opt�valuerM rO Zinfnr/ ZS_optZa_optZf_optZfmter�errZ allow_custom_lexer_formatterZ lexername�filename�codeZinfpr, r- Zoutfnr2 Z fmternameZoutfilerW ZcoloramarX �left�right�liner& r& r' �
main_inner� s�
rz c C s| t | d fd }y tj| dd� d�\}} W n$ tjk
rR t|tjd� dS X y t|| |�S tk
�rv dt|�kr�ttjd� td d
tjd� td tjd� td tjd� td
tjd� td d
tjd� ttjd� � ddl }|j
tj � � }|d j � }t
|�dk�r<|d|d jd�d j � dd� 7 }ttjd� tdtjd� t|tjd� tdtjd� dS X dS )z(
Main command line entry point.
r � r! Nzl:f:F:o:O:P:LS:a:N:vhVHgsx)r2 rK z-v�*�A z3An unhandled exception occurred while highlighting.z9Please report the whole traceback to the issue tracker atz8<https://bitbucket.org/birkenfeld/pygments-main/issues>.rQ z
(f%s)�
z *** Error while highlighting:z>*** If this is a bug you want to report, please rerun with -v.������)�USAGE�getoptZ GetoptErrorr3 r7 r8 rz rb �dict� traceback�format_exception�exc_infor" r] r# )ro rp rn r� rC �msgr& r& r' r[ s@
&
r[ )3r5 Z
__future__r r7 r� �textwrapr Zpygmentsr r rh r r r r r
r Zpygments.lexersr r
r r r r Zpygments.lexers.specialr Zpygments.formatters.latexr r Zpygments.formattersr r r r r Z pygments.formatters.terminalr Zpygments.filtersr r Zpygments.stylesr r r� r( r. r; rJ rz �argvr[ r& r& r& r' �<module>
s, N
3 =