Statistics
| Branch: | Tag: | Revision:

root / kamaki / cli / history.py @ 98093aac

History | View | Annotate | Download (2.9 kB)

1
#!/usr/bin/env python
2

    
3
# Copyright 2012-2013 GRNET S.A. All rights reserved.
4
#
5
# Redistribution and use in source and binary forms, with or
6
# without modification, are permitted provided that the following
7
# conditions are met:
8
#
9
#   1. Redistributions of source code must retain the above
10
#      copyright notice, this list of conditions and the following
11
#      disclaimer.
12
#
13
#   2. Redistributions in binary form must reproduce the above
14
#      copyright notice, this list of conditions and the following
15
#      disclaimer in the documentation and/or other materials
16
#      provided with the distribution.
17
#
18
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
19
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
22
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
25
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
26
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
28
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29
# POSSIBILITY OF SUCH DAMAGE.
30
#
31
# The views and conclusions contained in the software and
32
# documentation are those of the authors and should not be
33
# interpreted as representing official policies, either expressed
34
# or implied, of GRNET S.A.
35

    
36

    
37
class History(object):
38
    def __init__(self, filepath):
39
        self.filepath = filepath
40

    
41
    @classmethod
42
    def _match(self, line, match_terms):
43
        if match_terms is None:
44
            return True
45
        for term in match_terms.split():
46
            if term not in line:
47
                return False
48
        return True
49

    
50
    def get(self, match_terms=None, limit=0):
51
        limit = int(limit) or 0
52
        with open(self.filepath, 'r') as f:
53
            result = ['%s.  \t%s' % (
54
                i + 1, line) for i, line in enumerate(f.readlines())
55
                if self._match(line, match_terms)]
56
            return result[- limit:]
57

    
58
    def add(self, line):
59
        with open(self.filepath, 'a+') as f:
60
            f.write(line + '\n')
61

    
62
    def clean(self):
63
        with open(self.filepath, 'w'):
64
            pass
65

    
66
    def retrieve(self, cmd_id):
67
        """
68
        :param cmd_id: (int) the id of the command to retrieve can be positive
69
            or negative, zero values are ignored
70

71
        :returns: (str) the stored command record without the id
72
        """
73
        cmd_id = int(cmd_id)
74
        if not cmd_id:
75
            return None
76
        with open(self.filepath) as f:
77
            try:
78
                cmd_list = f.readlines()[:-1]  # exclude current command
79
                return cmd_list[cmd_id - (1 if cmd_id > 0 else 0)]
80
            except IndexError:
81
                return None