diff --git a/run_historical_conf.py b/run_historical_conf.py index 769283b5..64d2e4cf 100644 --- a/run_historical_conf.py +++ b/run_historical_conf.py @@ -1,4 +1,5 @@ -from configparser import ConfigParser, RawConfigParser +from configparser import RawConfigParser +import argparse import io import os @@ -24,33 +25,26 @@ def new_camera(camera, camera_index, config_file): with open(config_file, 'a') as add_configfile: new_config.write(add_configfile) - add_configfile.close() -def edit_camera(camera, camera_index, config_file): - config.set(f'Source_{camera_index}', 'VideoPath', f'/repo/data/historical_data/videos/{camera}') - config.set(f'Source_{camera_index}', 'Tags', f'tags_{camera}') - config.set(f'Source_{camera_index}', 'Name', f'name_{camera}') +def delete_camera(config_file): + config.read(config_file) + for section in config.sections(): + if 'Source_' in section: + config.remove_section(section) - with open(config_file, 'w') as configfile: - config.write(configfile) - configfile.close() + with open(config_file, 'w') as del_configfile: + config.write(del_configfile) if __name__ == '__main__': + parser = argparse.ArgumentParser(description="Processor's automatic setting.") + parser.add_argument('--ini_file', type=str, default=os.environ['DEFAULT_INI_CONFIG']) + parser.add_argument('--data_hist', type=str, default=os.environ['DEFAULT_HIST_VIDEOS_FOLDER']) + args = parser.parse_args() config = RawConfigParser(allow_no_value=True) - ini_file = 'config-x86.ini' + ini_file = args.ini_file config.read(ini_file) - list_cameras = os.listdir('data/historical_data/') - if len(list_cameras) > 1: - for i, camera in enumerate(list_cameras): - if i == 0: - edit_camera(camera, i, ini_file) - elif i > 0: - new_camera(camera, i, ini_file) - elif len(list_cameras) == 1: - if 'Source_0' in config.sections(): - edit_camera(list_cameras[0], 0, ini_file) - else: - new_camera(list_cameras[0], 0, ini_file) - - \ No newline at end of file + list_cameras = os.listdir(args.data_hist) + delete_camera(ini_file) + for i, camera in enumerate(list_cameras): + new_camera(camera, i, ini_file)