Veronica K. B. Olsen
100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""novelWriter Project Backup
|
||
|
||
novelWriter – Project Backup
|
||
==============================
|
||
Class handling project backups
|
||
|
||
File History:
|
||
Created: 2019-06-16 [0.1.5]
|
||
|
||
This file is a part of novelWriter
|
||
Copyright 2020, Veronica Berglyd Olsen
|
||
|
||
This program is free software: you can redistribute it and/or modify
|
||
it under the terms of the GNU General Public License as published by
|
||
the Free Software Foundation, either version 3 of the License, or
|
||
(at your option) any later version.
|
||
|
||
This program is distributed in the hope that it will be useful, but
|
||
WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
General Public License for more details.
|
||
|
||
You should have received a copy of the GNU General Public License
|
||
along with this program. If not, see https://www.gnu.org/licenses/.
|
||
"""
|
||
|
||
import logging
|
||
import nw
|
||
|
||
from os import path
|
||
from shutil import make_archive
|
||
from datetime import datetime
|
||
|
||
from nw.constants import nwAlert
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class NWBackup():
|
||
|
||
def __init__(self, theParent, theProject):
|
||
self.mainConf = nw.CONFIG
|
||
self.theParent = theParent
|
||
self.theProject = theProject
|
||
return
|
||
|
||
def zipIt(self):
|
||
|
||
if self.mainConf.backupPath is None or self.mainConf.backupPath == "":
|
||
self.theParent.makeAlert((
|
||
"Cannot backup project because no backup path is set. "
|
||
"Please set a valid backup location in Tools > Preferences."
|
||
), nwAlert.WARN)
|
||
return False
|
||
|
||
if self.theProject.projName is None or self.theProject.projName == "":
|
||
self.theParent.makeAlert((
|
||
"Cannot backup project because no project name is set. "
|
||
"Please set a Working Title in Project > Project Settings."
|
||
), nwAlert.WARN)
|
||
return False
|
||
|
||
if not path.isdir(self.mainConf.backupPath):
|
||
self.theParent.makeAlert((
|
||
"Cannot backup project because the backup path does not exist. "
|
||
"Please set a valid backup location in Tools > Preferences."
|
||
), nwAlert.WARN)
|
||
return False
|
||
|
||
logger.info("Backing up project")
|
||
self.theParent.statusBar.setStatus("Backing up project ...")
|
||
|
||
archName = ""
|
||
for c in self.theProject.projName:
|
||
if c.isalnum():
|
||
archName += c
|
||
else:
|
||
archName += "_"
|
||
|
||
archName = archName+"_"+datetime.now().strftime("%Y%m%d-%H%M%S")
|
||
baseName = path.join(self.mainConf.backupPath, archName)
|
||
|
||
try:
|
||
self.theProject._clearLockFile()
|
||
make_archive(baseName, "zip", self.theProject.projPath, ".")
|
||
self.theProject._writeLockFile()
|
||
logger.info("Backup written to: %s" % archName)
|
||
except Exception as e:
|
||
self.theParent.makeAlert(
|
||
["Could not write backup archive.",str(e)],
|
||
nwAlert.ERROR
|
||
)
|
||
return False
|
||
|
||
self.theParent.statusBar.setStatus("Project backup complete")
|
||
|
||
return True
|
||
|
||
# END Class NWBackup
|