def _setup(self):
disk_dir = self.conf['cut']['disk']['path'][self.disk_no]
ipc_name = os.path.basename(self.ddir)
scenario_name = os.path.basename(os.path.dirname(self.ddir))
scenario_dir = os.path.join(disk_dir, scenario_name)
target_dir = os.path.join(scenario_dir, ipc_name)
if os.path.exists(target_dir):
shutil.rmtree(target_dir)
utils.makedirs(target_dir)
self.ddir = target_dir
self.yitu_dir = os.path.join(self.ddir, 'yitu_dir')
self.log_path = os.path.join(self.ddir, 'result.log')
output_num = self.conf['output']['total']
output_dir = self.conf['output']['path'][int(scenario_name) %
output_num]
scenario_dir = os.path.join(output_dir, scenario_name)
if not os.path.exists(scenario_dir):
utils.makedirs(scenario_dir)
self.output_dir = os.path.join(scenario_dir, 'cluster', 'id_data_cut')
if not os.path.exists(self.video_dir):
raise RuntimeError('{} is not exists'.format(self.video_dir))
if not os.path.isdir(self.video_dir):
raise RuntimeError('{} is not dir'.format(self.video_dir))
def _move_mp4(self, path):
file_name = os.path.basename(path)
ddir = os.path.dirname(path)
ipc = file_name.split('_')[0]
try:
new_name = self._get_new_name(ipc, file_name)
except AttributeError:
return
ipc_dir = os.path.join(ddir, ipc)
if not os.path.exists(ipc_dir):
utils.makedirs(ipc_dir)
new_path = os.path.join(ipc_dir, new_name)
utils.move_file(path, new_path)
def _between_job(self, ddir):
utils.makedirs(os.path.join(ddir, 'cluster', 'output'))
import logging
from face.common.utils import makedirs
from face.common import constants as consts
LOG_FORMATTER = logging.Formatter(
'%(asctime)s [%(levelname)s] %(name)s %(filename)s:%(lineno)d %(message)s')
LOG_STREAM_HANDLER = logging.StreamHandler()
LOG_STREAM_HANDLER.setFormatter(LOG_FORMATTER)
LOG_STREAM_HANDLER.setLevel(logging.INFO)
makedirs(consts.LOG_DIR)
LOG_FILE_HANDLER = logging.FileHandler(consts.LOG_FILE)
LOG_FILE_HANDLER.setFormatter(LOG_FORMATTER)
LOG_FILE_HANDLER.setLevel(logging.DEBUG)
del logging.root.handlers[:]
logging.root.addHandler(LOG_STREAM_HANDLER)
logging.root.addHandler(LOG_FILE_HANDLER)
def _create_video_dir(self, ddir):
video_dir = os.path.join(ddir, 'video')
if not os.path.exists(video_dir):
utils.makedirs(video_dir)