1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
from __future__ import unicode_literals
from abc import ABCMeta, abstractmethod
from six import with_metaclass
import datetime
import os
__all__ = (
'FileHistory',
'History',
'InMemoryHistory',
)
class History(with_metaclass(ABCMeta, object)):
"""
Base ``History`` interface.
"""
@abstractmethod
def append(self, string):
" Append string to history. "
@abstractmethod
def __getitem__(self, key):
" Return one item of the history. It should be accessible like a `list`. "
@abstractmethod
def __iter__(self):
" Iterate through all the items of the history. Cronologically. "
@abstractmethod
def __len__(self):
" Return the length of the history. "
def __bool__(self):
"""
Never evaluate to False, even when the history is empty.
(Python calls __len__ if __bool__ is not implemented.)
This is mainly to allow lazy evaluation::
x = history or InMemoryHistory()
"""
return True
__nonzero__ = __bool__ # For Python 2.
class InMemoryHistory(History):
"""
:class:`.History` class that keeps a list of all strings in memory.
"""
def __init__(self):
self.strings = []
def append(self, string):
self.strings.append(string)
def __getitem__(self, key):
return self.strings[key]
def __iter__(self):
return iter(self.strings)
def __len__(self):
return len(self.strings)
class FileHistory(History):
"""
:class:`.History` class that stores all strings in a file.
"""
def __init__(self, filename):
self.strings = []
self.filename = filename
self._load()
def _load(self):
lines = []
def add():
if lines:
# Join and drop trailing newline.
string = ''.join(lines)[:-1]
self.strings.append(string)
if os.path.exists(self.filename):
with open(self.filename, 'rb') as f:
for line in f:
line = line.decode('utf-8')
if line.startswith('+'):
lines.append(line[1:])
else:
add()
lines = []
add()
def append(self, string):
self.strings.append(string)
# Save to file.
with open(self.filename, 'ab') as f:
def write(t):
f.write(t.encode('utf-8'))
write('\n# %s\n' % datetime.datetime.now())
for line in string.split('\n'):
write('+%s\n' % line)
def __getitem__(self, key):
return self.strings[key]
def __iter__(self):
return iter(self.strings)
def __len__(self):
return len(self.strings)
|
