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)
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