def test_backup_config_zip(self): """ Test backup_config_zip """ here = os.path.dirname(__file__) files = [f for f in os.listdir(here) if f[-3:] in [".db", ".py"]] zip_path = os.path.join(here, '_backup_test.zip') self.assertTrue(helpers.backup_config_zip(files, zip_path, here)) self.assertFalse(helpers.backup_config_zip(files, '/:/_backup_test.zip')) if os.path.isfile(zip_path): os.remove(zip_path)
def backup(backupDir=None): finalResult = '' if backupDir: source = [ ek(os.path.join, sickbeard.DATA_DIR, 'sickbeard.db'), sickbeard.CONFIG_FILE, ek(os.path.join, sickbeard.DATA_DIR, 'failed.db'), ek(os.path.join, sickbeard.DATA_DIR, 'cache.db') ] target = ek(os.path.join, backupDir, 'sickchill-' + time.strftime('%Y%m%d%H%M%S') + '.zip') for (path, dirs, files) in ek(os.walk, sickbeard.CACHE_DIR, topdown=True): for dirname in dirs: if path == sickbeard.CACHE_DIR and dirname not in [ 'images' ]: dirs.remove(dirname) for filename in files: source.append(ek(os.path.join, path, filename)) if helpers.backup_config_zip(source, target, sickbeard.DATA_DIR): finalResult += "Successful backup to " + target else: finalResult += "Backup FAILED" else: finalResult += "You need to choose a folder to save your backup to!" finalResult += "<br>\n" return finalResult
def test_restore_config_zip(self): """ Test restore_config_zip """ here = os.path.dirname(__file__) files = [f for f in os.listdir(here) if f[-3:] in [".db", ".py"]] zip_path = os.path.join(here, '_restore_test.zip') helpers.backup_config_zip(files, zip_path, here) restore_container = os.path.join(here, '_restore_tests') os.mkdir(restore_container) restore_path = os.path.join(restore_container, 'test') self.assertFalse(helpers.restore_config_zip(os.path.abspath(files[1]), restore_path)) # test invalid zip self.assertTrue(helpers.restore_config_zip(zip_path, restore_path)) self.assertTrue(helpers.restore_config_zip(zip_path, restore_path)) # test extractDir exists if os.path.isfile(zip_path): os.remove(zip_path) if os.path.isdir(restore_container): rmtree(restore_container)
def _backup(backupDir=None): if not backupDir: return False source = [ ek(os.path.join, sickbeard.DATA_DIR, 'sickbeard.db'), sickbeard.CONFIG_FILE, ek(os.path.join, sickbeard.DATA_DIR, 'failed.db'), ek(os.path.join, sickbeard.DATA_DIR, 'cache.db') ] target = ek(os.path.join, backupDir, 'sickrage-' + time.strftime('%Y%m%d%H%M%S') + '.zip') for (path, dirs, files) in ek(os.walk, sickbeard.CACHE_DIR, topdown=True): for dirname in dirs: if path == sickbeard.CACHE_DIR and dirname not in ['images']: dirs.remove(dirname) for filename in files: source.append(ek(os.path.join, path, filename)) return helpers.backup_config_zip(source, target, sickbeard.DATA_DIR)
def _backup(backupDir=None): if not backupDir: return False source = [ ek(os.path.join, sickbeard.DATA_DIR, 'sickbeard.db'), sickbeard.CONFIG_FILE, ek(os.path.join, sickbeard.DATA_DIR, 'failed.db'), ek(os.path.join, sickbeard.DATA_DIR, 'cache.db') ] target = ek(os.path.join, backupDir, 'sickchill-' + time.strftime('%Y%m%d%H%M%S') + '.zip') for (path, dirs, files) in ek(os.walk, sickbeard.CACHE_DIR, topdown=True): for dirname in dirs: if path == sickbeard.CACHE_DIR and dirname not in ['images']: dirs.remove(dirname) for filename in files: source.append(ek(os.path.join, path, filename)) return helpers.backup_config_zip(source, target, sickbeard.DATA_DIR)