some small changes

This commit is contained in:
2022-02-17 16:01:34 +03:00
parent f7ce92c38f
commit e2d6f1057a

View File

@@ -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):