def _prepare_chrome_user_data_dir(parsed_args): global _USER_DATA_DIR if parsed_args.use_temporary_data_dirs: _USER_DATA_DIR = tempfile.mkdtemp( prefix=build_common.CHROME_USER_DATA_DIR_PREFIX + '-') atexit.register(lambda: file_util.rmtree_with_retries(_USER_DATA_DIR)) elif parsed_args.user_data_dir: _USER_DATA_DIR = parsed_args.user_data_dir else: _USER_DATA_DIR = build_common.get_chrome_default_user_data_dir()
def _get_temp_dir(): global _TEMP_DIR if not _TEMP_DIR: _TEMP_DIR = tempfile.mkdtemp(prefix='arc_') atexit.register(lambda: file_util.rmtree_with_retries(_TEMP_DIR)) return _TEMP_DIR
def remove_arc_data_dir(): if os.path.exists(parsed_args.arc_data_dir): file_util.rmtree_with_retries(parsed_args.arc_data_dir)