From e2d6f1057a599ecf4a9657173082feae37036834 Mon Sep 17 00:00:00 2001 From: beacon Date: Thu, 17 Feb 2022 16:01:34 +0300 Subject: [PATCH] some small changes --- prepare.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/prepare.py b/prepare.py index 7f542b1..ffaea83 100755 --- a/prepare.py +++ b/prepare.py @@ -16,12 +16,7 @@ threading.current_thread().name = "main" if not os.path.isdir('log'): - try: - os.makedirs('log') - except: - pass -else: - pass + os.makedirs('log') a = 'log/' + str(time.strftime("%Y-%m-%d %H:%M:%S")) + '.log' logging.basicConfig(filename=a, filemode='w', format='%(asctime)s - %(threadName)s - %(levelname)s - %(message)s', @@ -38,8 +33,9 @@ class Args: parser.add_argument('-n', action='store', dest="count", default=10, help="number of smcreenshots, default 10", type=int) parser.add_argument('-c', action='store', dest='config', help='config file', default='config/config.ini', type=str) - parser.add_argument('-i', action='store', dest='input', help="input file", required=True, type=str) + parser.add_argument('-i', action='store', dest='input', help="input file", type=str, required=True) args = parser.parse_args() + logging.info(' '.join(f'{k}={v}' for k, v in vars(args).items())) @@ -82,7 +78,7 @@ class Head: self.bname: str self.bname = os.path.basename(Args.args.input) - self.fold = save_dir + os.path.splitext(self.bname)[0] + self.fold = save_dir + '/' + os.path.splitext(self.bname)[0] logging.info("Filename is %s" % self.bname) if os.path.isdir(self.fold):