Mercurial > cpython-lru_cache-weakref
annotate Lib/StringIO.py @ 12357:5b59ea41db4e legacy-trunk
Install the docs (with fewer rules).
Add descriptions for the system variables.
| author | Guido van Rossum <guido@python.org> |
|---|---|
| date | Mon, 10 Apr 2000 19:36:27 +0000 |
| parents | d7eca5b0960a |
| children | e2d62d747f7a |
| rev | line source |
|---|---|
|
11691
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
1 """File-like objects that read from or write to a string buffer. |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
2 |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
3 This implements (nearly) all stdio methods. |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
4 |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
5 f = StringIO() # ready for writing |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
6 f = StringIO(buf) # ready for reading |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
7 f.close() # explicitly release resources held |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
8 flag = f.isatty() # always false |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
9 pos = f.tell() # get current position |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
10 f.seek(pos) # set current position |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
11 f.seek(pos, mode) # mode 0: absolute; 1: relative; 2: relative to EOF |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
12 buf = f.read() # read until EOF |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
13 buf = f.read(n) # read up to n bytes |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
14 buf = f.readline() # read until end of line ('\n') or EOF |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
15 list = f.readlines()# list of f.readline() results until EOF |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
16 f.write(buf) # write at current position |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
17 f.writelines(list) # for line in list: f.write(line) |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
18 f.getvalue() # return whole file's contents as a string |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
19 |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
20 Notes: |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
21 - Using a real file is often faster (but less convenient). |
|
11870
d7eca5b0960a
Simple changes by Gerrit Holl - move author acknowledgements out of
Guido van Rossum <guido@python.org>
parents:
11691
diff
changeset
|
22 - There's also a much faster implementation in C, called cStringIO, but |
|
d7eca5b0960a
Simple changes by Gerrit Holl - move author acknowledgements out of
Guido van Rossum <guido@python.org>
parents:
11691
diff
changeset
|
23 it's not subclassable. |
|
11691
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
24 - fileno() is left unimplemented so that code which uses it triggers |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
25 an exception early. |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
26 - Seeking far beyond EOF and then writing will insert real null |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
27 bytes that occupy space in the buffer. |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
28 - There's a simple test set (see end of this file). |
|
e34991022571
Mass patch by Ka-Ping Yee:
Guido van Rossum <guido@python.org>
parents:
9016
diff
changeset
|
29 """ |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
30 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
31 import string |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
32 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
33 class StringIO: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
34 def __init__(self, buf = ''): |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
35 self.buf = buf |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
36 self.len = len(buf) |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
37 self.buflist = [] |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
38 self.pos = 0 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
39 self.closed = 0 |
|
2770
dc2adf79e523
set softspace to 0 in __init__
Guido van Rossum <guido@python.org>
parents:
1680
diff
changeset
|
40 self.softspace = 0 |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
41 def close(self): |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
42 if not self.closed: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
43 self.closed = 1 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
44 del self.buf, self.pos |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
45 def isatty(self): |
|
9016
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
46 if self.closed: |
|
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
47 raise ValueError, "I/O operation on closed file" |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
48 return 0 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
49 def seek(self, pos, mode = 0): |
|
9016
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
50 if self.closed: |
|
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
51 raise ValueError, "I/O operation on closed file" |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
52 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
53 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
54 self.buflist = [] |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
55 if mode == 1: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
56 pos = pos + self.pos |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
57 elif mode == 2: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
58 pos = pos + self.len |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
59 self.pos = max(0, pos) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
60 def tell(self): |
|
9016
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
61 if self.closed: |
|
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
62 raise ValueError, "I/O operation on closed file" |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
63 return self.pos |
|
3352
81f45f69388e
fix default arg for read() -- should be -1
Guido van Rossum <guido@python.org>
parents:
2852
diff
changeset
|
64 def read(self, n = -1): |
|
9016
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
65 if self.closed: |
|
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
66 raise ValueError, "I/O operation on closed file" |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
67 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
68 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
69 self.buflist = [] |
|
3352
81f45f69388e
fix default arg for read() -- should be -1
Guido van Rossum <guido@python.org>
parents:
2852
diff
changeset
|
70 if n < 0: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
71 newpos = self.len |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
72 else: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
73 newpos = min(self.pos+n, self.len) |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
74 r = self.buf[self.pos:newpos] |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
75 self.pos = newpos |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
76 return r |
|
5898
253377a40644
Patch by Lars Wirzenius to allow f.readline(length).
Guido van Rossum <guido@python.org>
parents:
3693
diff
changeset
|
77 def readline(self, length=None): |
|
9016
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
78 if self.closed: |
|
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
79 raise ValueError, "I/O operation on closed file" |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
80 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
81 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
82 self.buflist = [] |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
83 i = string.find(self.buf, '\n', self.pos) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
84 if i < 0: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
85 newpos = self.len |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
86 else: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
87 newpos = i+1 |
|
5898
253377a40644
Patch by Lars Wirzenius to allow f.readline(length).
Guido van Rossum <guido@python.org>
parents:
3693
diff
changeset
|
88 if length is not None: |
|
253377a40644
Patch by Lars Wirzenius to allow f.readline(length).
Guido van Rossum <guido@python.org>
parents:
3693
diff
changeset
|
89 if self.pos + length < newpos: |
|
253377a40644
Patch by Lars Wirzenius to allow f.readline(length).
Guido van Rossum <guido@python.org>
parents:
3693
diff
changeset
|
90 newpos = self.pos + length |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
91 r = self.buf[self.pos:newpos] |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
92 self.pos = newpos |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
93 return r |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
94 def readlines(self): |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
95 lines = [] |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
96 line = self.readline() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
97 while line: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
98 lines.append(line) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
99 line = self.readline() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
100 return lines |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
101 def write(self, s): |
|
9016
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
102 if self.closed: |
|
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
103 raise ValueError, "I/O operation on closed file" |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
104 if not s: return |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
105 if self.pos > self.len: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
106 self.buflist.append('\0'*(self.pos - self.len)) |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
107 self.len = self.pos |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
108 newpos = self.pos + len(s) |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
109 if self.pos < self.len: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
110 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
111 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
112 self.buflist = [] |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
113 self.buflist = [self.buf[:self.pos], s, self.buf[newpos:]] |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
114 self.buf = '' |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
115 else: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
116 self.buflist.append(s) |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
117 self.len = newpos |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
118 self.pos = newpos |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
119 def writelines(self, list): |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
120 self.write(string.joinfields(list, '')) |
| 2852 | 121 def flush(self): |
|
9016
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
122 if self.closed: |
|
b9a070736eac
Raise the right exception (ValueError) for attempted I/O on closed StringIO
Fred Drake <fdrake@acm.org>
parents:
5898
diff
changeset
|
123 raise ValueError, "I/O operation on closed file" |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
124 def getvalue(self): |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
125 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
126 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
127 self.buflist = [] |
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
128 return self.buf |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
129 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
130 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
131 # A little test suite |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
132 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
133 def test(): |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
134 import sys |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
135 if sys.argv[1:]: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
136 file = sys.argv[1] |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
137 else: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
138 file = '/etc/passwd' |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
139 lines = open(file, 'r').readlines() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
140 text = open(file, 'r').read() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
141 f = StringIO() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
142 for line in lines[:-2]: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
143 f.write(line) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
144 f.writelines(lines[-2:]) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
145 if f.getvalue() != text: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
146 raise RuntimeError, 'write failed' |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
147 length = f.tell() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
148 print 'File length =', length |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
149 f.seek(len(lines[0])) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
150 f.write(lines[1]) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
151 f.seek(0) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
152 print 'First line =', `f.readline()` |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
153 here = f.tell() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
154 line = f.readline() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
155 print 'Second line =', `line` |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
156 f.seek(-len(line), 1) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
157 line2 = f.read(len(line)) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
158 if line != line2: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
159 raise RuntimeError, 'bad result after seek back' |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
160 f.seek(len(line2), 1) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
161 list = f.readlines() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
162 line = list[-1] |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
163 f.seek(f.tell() - len(line)) |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
164 line2 = f.read() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
165 if line != line2: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
166 raise RuntimeError, 'bad result after seek back from EOF' |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
167 print 'Read', len(list), 'more lines' |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
168 print 'File length =', f.tell() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
169 if f.tell() != length: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
170 raise RuntimeError, 'bad length' |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
171 f.close() |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
172 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
173 if __name__ == '__main__': |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
174 test() |
