george/borg-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add custom output directory argument

Browse Source
This commit is contained in:
George Lacey 2021-05-04 14:25:16 +01:00
parent 018b96df75
commit 02e4311bc7
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/main.py
View File

@ -8,7 +8,13 @@ from borgmanager.summary import Summary
def main(args, path: Path): def main(args, path: Path):
if args.output is not None:
output_path = Path(args.output)
if not output_path.exists():
output_path.mkdir()
path = output_path
db = BorgDatabase(path / 'borg.sqlite') db = BorgDatabase(path / 'borg.sqlite')
if args.graph is not None: if args.graph is not None:
pass pass
elif args.summary is not None: elif args.summary is not None:
@ -27,6 +33,7 @@ def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("-g", "--graph", help="Produce graphs at specified location", type=str) parser.add_argument("-g", "--graph", help="Produce graphs at specified location", type=str)
parser.add_argument("-s", "--summary", help="Print summary", type=str) parser.add_argument("-s", "--summary", help="Print summary", type=str)
parser.add_argument("-o", "--output", help="Output Directory", type=str)
return parser.parse_args() return parser.parse_args()