backupdef.py

import pickle
import os

from entries import FolderEntry


class BackupDef:
    def __init__(self, folder: FolderEntry):
        self.folder = folder

    def __str__(self) -> str:
        return str(self.folder)

    @property
    def fileSize(self) -> int:
        return len(pickle.dumps(self, protocol=pickle.HIGHEST_PROTOCOL))

    def saveToFile(self, path: str):
        # Remove old backupdef file if it exists
        try:
            os.remove(path)
        except OSError:
            pass

        # Save backupdef to file
        with open(path, 'wb') as pkl_file:
            pickle.dump(self, pkl_file, protocol=pickle.HIGHEST_PROTOCOL)

    def loadFromFile(path: str = None) -> 'BackupDef':
        with open(path, 'rb') as pkl_file:
            return pickle.load(pkl_file)

    def delta(self, older_def: 'BackupDef'):
        """
        Computes the changes between this and an older BackupDef.
        Returns a BackupDef containing only the changes.
        """
        return BackupDef(self.folder.delta(older_def.folder))

    def processDelta(self, older_def: 'BackupDef', source: str, destination: str, bar=None):
        self.folder.processDelta(older_def.folder, source, destination, bar)