some small changes
This commit is contained in:
10
prepare.py
10
prepare.py
@@ -16,12 +16,7 @@ threading.current_thread().name = "main"
|
|||||||
|
|
||||||
|
|
||||||
if not os.path.isdir('log'):
|
if not os.path.isdir('log'):
|
||||||
try:
|
|
||||||
os.makedirs('log')
|
os.makedirs('log')
|
||||||
except:
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
|
|
||||||
a = 'log/' + str(time.strftime("%Y-%m-%d %H:%M:%S")) + '.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',
|
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,
|
parser.add_argument('-n', action='store', dest="count", default=10,
|
||||||
help="number of smcreenshots, default 10", type=int)
|
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('-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()
|
args = parser.parse_args()
|
||||||
|
|
||||||
logging.info(' '.join(f'{k}={v}' for k, v in vars(args).items()))
|
logging.info(' '.join(f'{k}={v}' for k, v in vars(args).items()))
|
||||||
|
|
||||||
|
|
||||||
@@ -82,7 +78,7 @@ class Head:
|
|||||||
self.bname: str
|
self.bname: str
|
||||||
|
|
||||||
self.bname = os.path.basename(Args.args.input)
|
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)
|
logging.info("Filename is %s" % self.bname)
|
||||||
|
|
||||||
if os.path.isdir(self.fold):
|
if os.path.isdir(self.fold):
|
||||||
|
|||||||
Reference in New Issue
Block a user