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)