diff options
Diffstat (limited to 'src/worklog')
| -rwxr-xr-x | src/worklog | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/worklog b/src/worklog index e5903af..c43cca1 100755 --- a/src/worklog +++ b/src/worklog @@ -5,8 +5,6 @@ author: Wacky404 email: wacky404@dev.com """ -import datetime -from datetime import timezone import json import sys import os @@ -83,9 +81,9 @@ else: BACKUPPATH = Path(osp.join(osp.expanduser("~"), "Documents", "worklogger")) -if args.extension: - combine_log(target_job=args.job, specified_ext=args.extension, - delete=args.delete, savepath=SAVEPATH, backuppath=BACKUPPATH) +if args.extension is not None: + combine_log(target_job=args.job, specified_ext=formats[str(args.extension).upper( + )], delete=args.delete, savepath=SAVEPATH, backuppath=BACKUPPATH) sys.exit() |