Python TodoList.todos方法代码示例

# 需要导入模块: from topydo.lib.TodoList import TodoList [as 别名]
# 或者: from topydo.lib.TodoList.TodoList import todos [as 别名]
class ChangeSet(object):
    """ Class for operations related with backup management. """

    def __init__(self, p_todolist=None, p_archive=None, p_call=[]):
        self.todolist = deepcopy(p_todolist)
        self.archive = deepcopy(p_archive)
        self.timestamp = str(int(time.time()))
        self.call = ' '.join(p_call)

        try:
            self.json_file = open(get_backup_path(), 'r+b')
        except IOError:
            self.json_file = open(get_backup_path(), 'w+b')

        self._read()

    def _read(self):
        """
        Reads backup file from json_file property and sets backup_dict property
        with data decompressed and deserialized from that file. If no usable
        data is found backup_dict is set to the empty dict.
        """
        self.json_file.seek(0)
        try:
            data = zlib.decompress(self.json_file.read())
            self.backup_dict = json.loads(data.decode('utf-8'))
        except (EOFError, zlib.error):
            self.backup_dict = {}

    def _write(self):
        """
        Writes data from backup_dict property in serialized and compressed form
        to backup file pointed in json_file property.
        """
        self.json_file.seek(0)
        self.json_file.truncate()
        dump = json.dumps(self.backup_dict)
        dump_c = zlib.compress(dump.encode('utf-8'))
        self.json_file.write(dump_c)

    def add_archive(self, p_archive):
        """ Sets deep copy of p_archive as archive attribute. """
        self.archive = deepcopy(p_archive)

    def add_todolist(self, p_todolist):
        """ Sets deep copy of p_todolist as todolist attribute. """
        self.todolist = deepcopy(p_todolist)

    def save(self, p_todolist):
        """
        Saves a tuple with archive, todolist and command with its arguments
        into the backup file with unix timestamp as the key. Tuple is then
        indexed in backup file with combination of hash calculated from
        p_todolist and unix timestamp. Backup file is closed afterwards.
        """
        self._trim()

        current_hash = hash_todolist(p_todolist)
        list_todo = (self.todolist.print_todos()+'\n').splitlines(True)
        try:
            list_archive = (self.archive.print_todos()+'\n').splitlines(True)
        except AttributeError:
            list_archive = []

        self.backup_dict[self.timestamp] = (list_todo, list_archive,  self.call)

        index = self._get_index()
        index.insert(0, (self.timestamp, current_hash))
        self._save_index(index)

        self._write()
        self.close()

    def delete(self, p_timestamp=None, p_write=True):
        """ Removes backup from the backup file. """
        timestamp = p_timestamp or self.timestamp
        index = self._get_index()

        try:
            del self.backup_dict[timestamp]
            index.remove(index[[change[0] for change in index].index(timestamp)])
            self._save_index(index)

            if p_write:
                self._write()
        except KeyError:
            pass

    def _get_index(self):
        try:
            index = self.backup_dict['index']
        except KeyError:
            self.backup_dict['index'] = []
            index = self.backup_dict['index']

        return index

    def _save_index(self, p_index):
        """
        Saves index of backups supplied in p_index into the backup_file
#.........这里部分代码省略.........