class Reset:
def __init__(self):
self.sourceDir = 'imagesbackup'
self.destDir = 'imagesdir'
self.filesystem = Filesystem()
if not self.filesystem.checkDirExists(self.destDir):
self.filesystem.makeDir(self.destDir)
assert self.filesystem.checkDirExists(self.destDir), \
self.destDir + ' does not exist'
def fill(self):
assert self.filesystem.checkDirExists(self.sourceDir), \
self.sourceDir + ' does not exist'
self.sourceImages = self.filesystem.listJpegs(self.sourceDir)
for image in self.sourceImages:
self.filesystem.copy([self.sourceDir, image],
[self.destDir, image])
def empty(self, path='', removeDir=True):
if path == '':
path = self.destDir
path = self.filesystem.joinPath(path)
if not self.filesystem.checkDirExists(path):
return
self.destImages = self.filesystem.listJpegs(path)
for image in self.destImages:
self.filesystem.removeFile([path, image])
if removeDir:
self.filesystem.removeDir(path)
