Mercurial > cpython-lru_cache-weakref
annotate Lib/StringIO.py @ 9383:2b34461ba4a3 legacy-trunk
PyMapping_SetItemString() returns int, not an object.
| author | Guido van Rossum <guido@python.org> |
|---|---|
| date | Fri, 16 Oct 1998 17:43:53 +0000 |
| parents | b9a070736eac |
| children | e34991022571 |
| rev | line source |
|---|---|
|
1680
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
1 # class StringIO implements file-like objects that read/write a |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
2 # string buffer (a.k.a. "memory files"). |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
3 # |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
4 # This implements (nearly) all stdio methods. |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
5 # |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
6 # f = StringIO() # ready for writing |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
7 # f = StringIO(buf) # ready for reading |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
8 # f.close() # explicitly release resources held |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
9 # flag = f.isatty() # always false |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
10 # pos = f.tell() # get current position |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
11 # f.seek(pos) # set current position |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
12 # f.seek(pos, mode) # mode 0: absolute; 1: relative; 2: relative to EOF |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
13 # buf = f.read() # read until EOF |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
14 # buf = f.read(n) # read up to n bytes |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
15 # buf = f.readline() # read until end of line ('\n') or EOF |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
16 # list = f.readlines()# list of f.readline() results until EOF |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
17 # f.write(buf) # write at current position |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
18 # f.writelines(list) # for line in list: 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
|
19 # f.getvalue() # return whole file's contents as a string |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
20 # |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
21 # Notes: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
22 # - Using a real file is often faster (but less convenient). |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
23 # - fileno() is left unimplemented so that code which uses it triggers |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
24 # an exception early. |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
25 # - Seeking far beyond EOF and then writing will insert real null |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
26 # bytes that occupy space in the buffer. |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
27 # - There's a simple test set (see end of this file). |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
28 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
29 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
|
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 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
|
32 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
|
33 self.buf = buf |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
34 self.len = len(buf) |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
35 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
|
36 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
|
37 self.closed = 0 |
|
2770
dc2adf79e523
set softspace to 0 in __init__
Guido van Rossum <guido@python.org>
parents:
1680
diff
changeset
|
38 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
|
39 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
|
40 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
|
41 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
|
42 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
|
43 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
|
44 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
|
45 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
|
46 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
|
47 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
|
48 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
|
49 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
|
50 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
51 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
52 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
|
53 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
|
54 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
|
55 elif mode == 2: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
56 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
|
57 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
|
58 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
|
59 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
|
60 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
|
61 return self.pos |
|
3352
81f45f69388e
fix default arg for read() -- should be -1
Guido van Rossum <guido@python.org>
parents:
2852
diff
changeset
|
62 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
|
63 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
|
64 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
|
65 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
66 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
67 self.buflist = [] |
|
3352
81f45f69388e
fix default arg for read() -- should be -1
Guido van Rossum <guido@python.org>
parents:
2852
diff
changeset
|
68 if n < 0: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
69 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
|
70 else: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
71 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
|
72 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
|
73 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
|
74 return r |
|
5898
253377a40644
Patch by Lars Wirzenius to allow f.readline(length).
Guido van Rossum <guido@python.org>
parents:
3693
diff
changeset
|
75 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
|
76 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
|
77 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
|
78 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
79 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
80 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
|
81 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
|
82 if i < 0: |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
83 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
|
84 else: |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
85 newpos = i+1 |
|
5898
253377a40644
Patch by Lars Wirzenius to allow f.readline(length).
Guido van Rossum <guido@python.org>
parents:
3693
diff
changeset
|
86 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
|
87 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
|
88 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
|
89 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
|
90 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
|
91 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
|
92 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
|
93 lines = [] |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
94 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
|
95 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
|
96 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
|
97 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
|
98 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
|
99 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
|
100 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
|
101 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
|
102 if not s: return |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
103 if self.pos > self.len: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
104 self.buflist.append('\0'*(self.pos - self.len)) |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
105 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
|
106 newpos = self.pos + len(s) |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
107 if self.pos < self.len: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
108 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
109 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
110 self.buflist = [] |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
111 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
|
112 self.buf = '' |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
113 else: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
114 self.buflist.append(s) |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
115 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
|
116 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
|
117 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
|
118 self.write(string.joinfields(list, '')) |
| 2852 | 119 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
|
120 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
|
121 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
|
122 def getvalue(self): |
|
3693
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
123 if self.buflist: |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
124 self.buf = self.buf + string.joinfields(self.buflist, '') |
|
818ee115054a
Sjoerd's StringIO speed-up
Guido van Rossum <guido@python.org>
parents:
3352
diff
changeset
|
125 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
|
126 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
|
127 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
128 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
129 # 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
|
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 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
|
132 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
|
133 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
|
134 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
|
135 else: |
|
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 = '/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
|
137 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
|
138 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
|
139 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
|
140 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
|
141 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
|
142 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
|
143 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
|
144 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
|
145 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
|
146 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
|
147 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
|
148 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
|
149 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
|
150 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
|
151 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
|
152 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 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
|
154 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
|
155 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
|
156 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
|
157 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
|
158 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
|
159 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
|
160 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
|
161 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
|
162 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
|
163 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
|
164 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
|
165 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
|
166 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
|
167 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
|
168 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
|
169 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
|
170 |
|
0a60460e99f0
Two new generally useful modules: types defines names for all built-in types,
Guido van Rossum <guido@python.org>
parents:
diff
changeset
|
171 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
|
172 test() |
