blob: 2d5bcdc4716e11d4f65c4b719a712052713affd9 [file] [log] [blame]
Fred Drake295da241998-08-10 19:42:37 +00001\section{\module{multifile} ---
Fred Drake812860e1999-04-23 14:46:18 +00002 Support for files containing distinct parts}
Fred Drakeb91e9341998-07-23 17:59:49 +00003
Fred Drake812860e1999-04-23 14:46:18 +00004\declaremodule{standard}{multifile}
Fred Draked795c5c1998-08-07 15:55:14 +00005\modulesynopsis{Support for reading files which contain distinct
Fred Drake812860e1999-04-23 14:46:18 +00006 parts, such as some MIME data.}
7\sectionauthor{Eric S. Raymond}{esr@snark.thyrsus.com}
Fred Drakeb91e9341998-07-23 17:59:49 +00008
Guido van Rossum8668e8e1998-06-28 17:55:53 +00009
Fred Drake1717ba41998-07-02 19:36:50 +000010The \class{MultiFile} object enables you to treat sections of a text
11file as file-like input objects, with \code{''} being returned by
12\method{readline()} when a given delimiter pattern is encountered. The
Guido van Rossum8668e8e1998-06-28 17:55:53 +000013defaults of this class are designed to make it useful for parsing
14MIME multipart messages, but by subclassing it and overriding methods
15it can be easily adapted for more general use.
16
Fred Drake1717ba41998-07-02 19:36:50 +000017\begin{classdesc}{MultiFile}{fp\optional{, seekable}}
Guido van Rossum8668e8e1998-06-28 17:55:53 +000018Create a multi-file. You must instantiate this class with an input
Fred Drake1717ba41998-07-02 19:36:50 +000019object argument for the \class{MultiFile} instance to get lines from,
20such as as a file object returned by \function{open()}.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000021
Fred Drake1717ba41998-07-02 19:36:50 +000022\class{MultiFile} only ever looks at the input object's
23\method{readline()}, \method{seek()} and \method{tell()} methods, and
24the latter two are only needed if you want random access to the
25individual MIME parts. To use \class{MultiFile} on a non-seekable
26stream object, set the optional \var{seekable} argument to false; this
27will prevent using the input object's \method{seek()} and
28\method{tell()} methods.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000029\end{classdesc}
30
Fred Drake1717ba41998-07-02 19:36:50 +000031It will be useful to know that in \class{MultiFile}'s view of the world, text
Guido van Rossum8668e8e1998-06-28 17:55:53 +000032is composed of three kinds of lines: data, section-dividers, and
33end-markers. MultiFile is designed to support parsing of
34messages that may have multiple nested message parts, each with its
35own pattern for section-divider and end-marker lines.
36
Fred Draked795c5c1998-08-07 15:55:14 +000037
38\subsection{MultiFile Objects \label{MultiFile-objects}}
Guido van Rossum8668e8e1998-06-28 17:55:53 +000039
40A \class{MultiFile} instance has the following methods:
41
Guido van Rossum8668e8e1998-06-28 17:55:53 +000042\begin{methoddesc}{readline}{str}
43Read a line. If the line is data (not a section-divider or end-marker
44or real EOF) return it. If the line matches the most-recently-stacked
Guido van Rossum8ec619f1998-06-30 16:35:25 +000045boundary, return \code{''} and set \code{self.last} to 1 or 0 according as
Guido van Rossum8668e8e1998-06-28 17:55:53 +000046the match is or is not an end-marker. If the line matches any other
Fred Drake1717ba41998-07-02 19:36:50 +000047stacked boundary, raise an error. On encountering end-of-file on the
48underlying stream object, the method raises \exception{Error} unless
49all boundaries have been popped.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000050\end{methoddesc}
51
52\begin{methoddesc}{readlines}{str}
Fred Drake1717ba41998-07-02 19:36:50 +000053Return all lines remaining in this part as a list of strings.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000054\end{methoddesc}
55
Fred Drake1717ba41998-07-02 19:36:50 +000056\begin{methoddesc}{read}{}
Guido van Rossum8668e8e1998-06-28 17:55:53 +000057Read all lines, up to the next section. Return them as a single
58(multiline) string. Note that this doesn't take a size argument!
59\end{methoddesc}
60
Fred Drake1717ba41998-07-02 19:36:50 +000061\begin{methoddesc}{seek}{pos\optional{, whence}}
Guido van Rossum8668e8e1998-06-28 17:55:53 +000062Seek. Seek indices are relative to the start of the current section.
Fred Drake1717ba41998-07-02 19:36:50 +000063The \var{pos} and \var{whence} arguments are interpreted as for a file
64seek.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000065\end{methoddesc}
66
Fred Drake1717ba41998-07-02 19:36:50 +000067\begin{methoddesc}{tell}{}
68Return the file position relative to the start of the current section.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000069\end{methoddesc}
70
Fred Drakef0ebbe02001-03-08 22:46:41 +000071\begin{methoddesc}{next}{}
72Skip lines to the next section (that is, read lines until a
73section-divider or end-marker has been consumed). Return true if
74there is such a section, false if an end-marker is seen. Re-enable
75the most-recently-pushed boundary.
76\end{methoddesc}
77
Guido van Rossum8668e8e1998-06-28 17:55:53 +000078\begin{methoddesc}{is_data}{str}
Fred Drake1717ba41998-07-02 19:36:50 +000079Return true if \var{str} is data and false if it might be a section
Fred Drake812860e1999-04-23 14:46:18 +000080boundary. As written, it tests for a prefix other than \code{'-}\code{-'} at
Fred Drake1717ba41998-07-02 19:36:50 +000081start of line (which all MIME boundaries have) but it is declared so
82it can be overridden in derived classes.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000083
84Note that this test is used intended as a fast guard for the real
Fred Drake1717ba41998-07-02 19:36:50 +000085boundary tests; if it always returns false it will merely slow
86processing, not cause it to fail.
Guido van Rossum8668e8e1998-06-28 17:55:53 +000087\end{methoddesc}
88
Fred Drakef0ebbe02001-03-08 22:46:41 +000089\begin{methoddesc}{push}{str}
90Push a boundary string. When an appropriately decorated version of
91this boundary is found as an input line, it will be interpreted as a
92section-divider or end-marker. All subsequent
93reads will return the empty string to indicate end-of-file, until a
94call to \method{pop()} removes the boundary a or \method{next()} call
95reenables it.
96
97It is possible to push more than one boundary. Encountering the
98most-recently-pushed boundary will return EOF; encountering any other
99boundary will raise an error.
100\end{methoddesc}
101
102\begin{methoddesc}{pop}{}
103Pop a section boundary. This boundary will no longer be interpreted
104as EOF.
105\end{methoddesc}
106
Guido van Rossum8668e8e1998-06-28 17:55:53 +0000107\begin{methoddesc}{section_divider}{str}
108Turn a boundary into a section-divider line. By default, this
Fred Drake812860e1999-04-23 14:46:18 +0000109method prepends \code{'-}\code{-'} (which MIME section boundaries have) but
Fred Drake1717ba41998-07-02 19:36:50 +0000110it is declared so it can be overridden in derived classes. This
111method need not append LF or CR-LF, as comparison with the result
112ignores trailing whitespace.
Guido van Rossum8668e8e1998-06-28 17:55:53 +0000113\end{methoddesc}
114
115\begin{methoddesc}{end_marker}{str}
116Turn a boundary string into an end-marker line. By default, this
Fred Drake812860e1999-04-23 14:46:18 +0000117method prepends \code{'-}\code{-'} and appends \code{'-}\code{-'} (like a
Fred Drake1717ba41998-07-02 19:36:50 +0000118MIME-multipart end-of-message marker) but it is declared so it can be
119be overridden in derived classes. This method need not append LF or
120CR-LF, as comparison with the result ignores trailing whitespace.
Guido van Rossum8668e8e1998-06-28 17:55:53 +0000121\end{methoddesc}
122
123Finally, \class{MultiFile} instances have two public instance variables:
124
125\begin{memberdesc}{level}
Fred Drake1717ba41998-07-02 19:36:50 +0000126Nesting depth of the current part.
Guido van Rossum8668e8e1998-06-28 17:55:53 +0000127\end{memberdesc}
128
129\begin{memberdesc}{last}
Fred Drake1717ba41998-07-02 19:36:50 +0000130True if the last end-of-file was for an end-of-message marker.
Guido van Rossum8668e8e1998-06-28 17:55:53 +0000131\end{memberdesc}
132
Fred Drake1717ba41998-07-02 19:36:50 +0000133
Fred Draked795c5c1998-08-07 15:55:14 +0000134\subsection{\class{MultiFile} Example \label{multifile-example}}
Fred Drake9164f882000-04-08 04:53:29 +0000135\sectionauthor{Skip Montanaro}{skip@mojam.com}
Guido van Rossum8668e8e1998-06-28 17:55:53 +0000136
137\begin{verbatim}
Fred Drakec2c46c32000-04-07 16:09:59 +0000138import mimetools
Martin v. Löwisd15a9422000-09-30 17:04:40 +0000139import multifile
Fred Drakec2c46c32000-04-07 16:09:59 +0000140import StringIO
141
142def extract_mime_part_matching(stream, mimetype):
143 """Return the first element in a multipart MIME message on stream
144 matching mimetype."""
145
146 msg = mimetools.Message(stream)
147 msgtype = msg.gettype()
148 params = msg.getplist()
149
150 data = StringIO.StringIO()
151 if msgtype[:10] == "multipart/":
152
153 file = multifile.MultiFile(stream)
154 file.push(msg.getparam("boundary"))
155 while file.next():
156 submsg = mimetools.Message(file)
157 try:
158 data = StringIO.StringIO()
159 mimetools.decode(file, data, submsg.getencoding())
160 except ValueError:
161 continue
162 if submsg.gettype() == mimetype:
163 break
164 file.pop()
165 return data.getvalue()
Guido van Rossum8668e8e1998-06-28 17:55:53 +0000166\end{verbatim}