some small changes
This commit is contained in:
12
prepare.py
12
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):
|
||||
|
||||
Reference in New Issue
Block a user