3
T'ícA ã @ s. d Z ddlmZ dZddlZddlZddlZddlZddlm Z ddl
m Z m Z m
Z
ejd"krbeZG dd d eZG d
d d eZd d
ZG dd de ZG dd de ZG dd deZG dd deZG dd deZG dd deZG dd deZG d d d eZ G d d d eZ G d d! d!eZ dS )#z
I/O classes provide a uniform API for low-level input and output. Subclasses
exist for a variety of input/output mechanisms.
é )Úprint_functionZreStructuredTextN)Ú
TransformSpec)Úlocale_encodingÚ ErrorStringÚ ErrorOutputé c @ s e Zd ZdS )Ú
InputErrorN)Ú__name__Ú
__module__Ú __qualname__© r r ú//tmp/pip-build-gk9425m9/docutils/docutils/io.pyr s r c @ s e Zd ZdS )Ú OutputErrorN)r r
r r r r r
r s r c
C s6 yt j| jt j|kS tttfk
r0 dS X dS )aN Test, whether the encoding of `stream` matches `encoding`.
Returns
:None: if `encoding` or `stream.encoding` are not a valid encoding
argument (e.g. ``None``) or `stream.encoding is missing.
:True: if the encoding argument resolves to the same value as `encoding`,
:False: if the encodings differ.
N)ÚcodecsÚlookupÚencodingÚ LookupErrorÚAttributeErrorÚ TypeError)Ústreamr r r r
Úcheck_encoding s
r c @ sh e Zd ZdZdZdZdddZdd Zd d
Zd d Z e
j d
Z e
jdfe
jdfe
jdffZdd ZdS )ÚInputz1
Abstract base class for input wrappers.
ÚinputNÚstrictc C s. || _ || _|| _|| _|s$| j| _d | _d S )N)r Ú
error_handlerÚsourceÚ source_pathÚdefault_source_pathÚsuccessful_encoding)Úselfr r r r r r r
Ú__init__5 s zInput.__init__c C s d| j | j| jf S )Nz %s: source=%r, source_path=%r)Ú __class__r r )r r r r
Ú__repr__I s
zInput.__repr__c C s t d S )N)ÚNotImplementedError)r r r r
ÚreadM s z
Input.readc C sô | j r&| j j dkr&t|ts&tdt|tr4|S | j rD| j g}n.| j|}|rZ|g}nddg}trr|jdt xX|D ]P}y t||| j}|| _ |j
ddS t t fk
rÆ } z
|}W Y dd}~X qxX qxW t d d
j
d d |D t|f dS )
a
Decode a string, `data`, heuristically.
Raise UnicodeError if unsuccessful.
The client application should call ``locale.setlocale`` at the
beginning of processing::
locale.setlocale(locale.LC_ALL, '')
Úunicodez=input encoding is "unicode" but input is not a unicode objectzutf-8zlatin-1é u ï»¿Ú NzEUnable to decode input data. Tried the following encodings: %s.
(%s)z, c S s g | ] }t |qS r )Úrepr)Ú.0Úencr r r
ú
<listcomp>} s z Input.decode.<locals>.<listcomp>)r ÚlowerÚ
isinstancer% ÚAssertionErrorÚ determine_encoding_from_datar Úinsertr r ÚreplaceÚ UnicodeErrorr Újoinr )r ÚdataÚ encodingsÚ
data_encodingr* ÚdecodedÚerrÚerrorr r r
ÚdecodeP s0
z Input.decodes coding[:=]\s*([-\w.]+)zutf-8z utf-16-bez utf-16-lec C s` x | j D ]\}}|j|r|S qW x8|j dd
D ]$}| jj|}|r4|jdjdS q4W dS )z
Try to determine the encoding of `data` by looking *in* `data`.
Check for a byte order mark (BOM) or an encoding declaration.
Né r& Úascii)Úbyte_order_marksÚ
startswithÚ
splitlinesÚ coding_slugÚsearchÚgroupr: )r r4 Z start_bytesr ÚlineÚmatchr r r
r/ s
z"Input.determine_encoding_from_data)NNNr )r r
r Ú__doc__Úcomponent_typer r r" r$ r: ÚreÚcompiler@ r ÚBOM_UTF8Ú BOM_UTF16_BEÚ BOM_UTF16_LEr= r/ r r r r
r + s
0
r c @ s: e Zd ZdZdZdZd
ddZdd Zd d
Zd d Z dS )ÚOutputz2
Abstract base class for output wrappers.
ÚoutputNr c C s, || _ |p d| _|| _|| _|s(| j| _d S )Nr )r r Ú destinationÚdestination_pathÚdefault_destination_path)r rN rO r r r r r
r ¥ s
zOutput.__init__c C s d| j | j| jf S )Nz'%s: destination=%r, destination_path=%r)r! rN rO )r r r r
r" ¶ s zOutput.__repr__c C s t dS )z;`data` is a Unicode string, to be encoded by `self.encode`.N)r# )r r4 r r r
Úwriteº s z Output.writec C sL | j r*| j j dkr*t|ts&td|S t|ts8|S |j| j | jS d S )Nr% zFthe encoding given is "unicode" but the output is not a Unicode string)r r, r- r% r. Úencoder )r r4 r r r
rR ¾ s
z
Output.encode)NNNr )
r r
r rE rF rP r r" rQ rR r r r r
rL s
rL c @ sL e Zd ZdZdddddejdkr$dndfd d
Zd d Zd
d Zdd Z dS )Ú FileInputz5
Input for single, simple file-like objects.
Nr Tr r ÚrZrUc C s t j| |||| || _t | _|dkr|rtjdkrF| j| jd}ni }yt ||f|| _
W q t k
r } zt |j
|j|W Y dd}~X qX qÐtj| _
n2tjdkrÐt| j
| jdkrÐtd| j| j
jf | süy| j
j| _W n tk
rú Y nX dS ) aÅ
:Parameters:
- `source`: either a file-like object (which is read directly), or
`None` (which implies `sys.stdin` if no `source_path` given).
- `source_path`: a path to a file, which is opened and then read.
- `encoding`: the expected text encoding of the input file.
- `error_handler`: the encoding error handler to use.
- `autoclose`: close automatically after read (except when
`sys.stdin` is the source).
- `mode`: how the file is to be opened (see standard function
`open`). The default 'rU' provides universal newline support
for text files with Python 2.x.
Nr r )r ÚerrorsFzOEncoding clash: encoding given is "%s" but source is opened with encoding "%s".)r r )r r )r r Ú autocloser Ú_stderrÚsysÚ version_infor r Úopenr ÚIOErrorr ÚerrnoÚstrerrorÚstdinr r2 Únamer r ) r r r r r rV ÚmodeÚkwargsr9 r r r
r Ð s.
$
zFileInput.__init__c C sÌ z°yD| j tjkr:tjdkr:| j jj }dj|j d }n
| j j }W nf tt fk
r¬ } zF| j
r| j rt | j d}|j }|j
dj|j d }n W Y dd}~X nX W d| jrÀ| j
X | j|S )zU
Read and decode a single file and return the data (Unicode string).
r r ó
ÚrbN)r r )r rX r^ rY Úbufferr$ r3 r? r2 r r r rZ ÚcloserV r: )r r4 r8 Zb_sourcer r r
r$ ÿ s
zFileInput.readc C s | j jdS )zK
Return lines of a single file as list of Unicode strings.
T)r$ r? )r r r r
Ú readlines s zFileInput.readlinesc C s | j tjk r| j j d S )N)r rX r^ re )r r r r
re s zFileInput.close)r r )
r r
r rE rX rY r r$ rf re r r r r
rS Ë s ,rS c @ s6 e Zd ZdZdZdddZdd Zd
d Zd d
ZdS )Ú
FileOutputz6
Output for single, simple file-like objects.
ÚwNr Tc C s´ t j| |||| d| _|| _|dk r,|| _t | _|dkrR|rHd| _qtj| _ n6|rt
| j dr|| j jkrt d| j j|f | jd |s°y| j j | _
W n tk
r® Y nX dS )aA
:Parameters:
- `destination`: either a file-like object (which is written
directly) or `None` (which implies `sys.stdout` if no
`destination_path` given).
- `destination_path`: a path to a file, which is opened and then
written.
- `encoding`: the text encoding of the output file.
- `error_handler`: the encoding error handler to use.
- `autoclose`: close automatically after write (except when
`sys.stdout` or `sys.stderr` is the destination).
- `handle_io_errors`: ignored, deprecated, will be removed.
- `mode`: how the file is to be opened (see standard function
`open`). The default is 'w', providing universal newline
support for text files.
TNFr` z?Warning: Destination mode "%s" differs from specified mode "%s")Úfile)rL r ÚopenedrV r` r rW rX ÚstdoutrN ÚhasattrÚprintr_ rO r )r rN rO r r rV Zhandle_io_errorsr` r r r
r 1 s*
zFileOutput.__init__c C s t jdkr$d| jkr$| j| jd}ni }yt| j| jf|| _W n4 tk
rt } zt |j
|j | jW Y d d }~X nX d| _ d S )Nr r Úb)r rU T)r r )
rX rY r` r r rZ rO rN r[ r r\ r] rj )r ra r9 r r r
rZ \ s
zFileOutput.openc C sl | j s| j d| jkr"tjd k s4t| j| jdkrf| j|}tjd
krft j
dkrf|j dt t j
d}zîy| jj
| W nØ tk
r } z~tjdkot|t ry| jjj
| W nN tk
r t| j| jdk rútd| jpæd | jj| jf n|Y nX W Y d
d
}~X n> ttfk
rP } z td | jt|f W Y d
d
}~X nX W d
| jrf| j X |S )zÐEncode `data`, write it to a single file, and return it.
With Python 3 or binary output mode, `data` is returned unchanged,
except when specified encoding and output encoding differ.
rn r r FÚ
rb r< z;Encoding of %s (%s) differs
from specified encoding (%s)rN Nz:Unable to encode output data. output-encoding is: %s.
(%s))r r )r r )r r )rj rZ r` rX rY r rN r rR ÚosÚlinesepr1 ÚbytesrQ r r- rd r Ú
ValueErrorrO r2 r r rV re )r r4 r8 r r r
rQ j s6
(
zFileOutput.writec C s& | j tjtjfkr"| j j d| _d S )NF)rN rX rk Ústderrre rj )r r r r
re s
zFileOutput.close)NNNr TNN) r r
r rE r` r rZ rQ re r r r r
rg % s
)'rg c @ s e Zd ZdZdZdS )ÚBinaryFileOutputzL
A version of docutils.io.FileOutput which writes to a binary file.
ÚwbN)r r
r rE r` r r r r
ru s ru c @ s e Zd ZdZdZdd ZdS )Ú StringInputz
Direct string input.
z<string>c C s | j | jS )z$Decode and return the source string.)r: r )r r r r
r$ ¨ s zStringInput.readN)r r
r rE r r$ r r r r
rw s rw c @ s e Zd ZdZdZdd ZdS )Ú StringOutputz
Direct string output.
z<string>c C s | j || _| jS )z=Encode `data`, store it in `self.destination`, and return it.)rR rN )r r4 r r r
rQ µ s zStringOutput.writeN)r r
r rE rP rQ r r r r
rx s rx c @ s e Zd ZdZdZdd ZdS )Ú NullInputz)
Degenerate input: read nothing.
z
null inputc C s dS )zReturn a null string.r' r )r r r r
r$ Ã s zNullInput.readN)r r
r rE r r$ r r r r
ry » s ry c @ s e Zd ZdZdZdd ZdS )Ú
NullOutputz+
Degenerate output: write nothing.
z null outputc C s dS )z6Do nothing ([don't even] send data to the bit bucket).Nr )r r4 r r r
rQ Ð s zNullOutput.writeN)r r
r rE rP rQ r r r r
rz È s rz c @ s e Zd ZdZdZdd ZdS )Ú DocTreeInputzm
Adapter for document tree input.
The document tree must be passed in the ``source`` parameter.
z
doctree inputc C s | j S )zReturn the document tree.)r )r r r r
r$ ß s zDocTreeInput.readN)r r
r rE r r$ r r r r
r{ Õ s r{ )r r ) rE Ú
__future__r Z
__docformat__rX rp rG r Zdocutilsr Z docutils.utils.error_reportingr r r rY Ústrr% r[ r r r r rL rS rg ru rw rx ry rz r{ r r r r
Ú<module> s.
p0Zr