import sys from pathlib import Path def resolve_user_data_dir(): """ Resolve the user_data directory path. Order of precedence: 1. --user-data-dir CLI flag (pre-parsed from sys.argv before argparse) 2. In --portable mode, prefer ../user_data if it exists 3. Default: 'user_data' """ script_dir = Path(__file__).resolve().parent.parent # Check sys.argv for --user-data-dir before argparse runs for i, arg in enumerate(sys.argv): if arg == '--user-data-dir' and i + 1 < len(sys.argv): return Path(sys.argv[i + 1]) elif arg.startswith('--user-data-dir='): return Path(arg.split('=', 1)[1]) # In portable mode, prefer ../user_data if it exists is_portable = '--portable' in sys.argv if is_portable: parent_path = script_dir.parent / 'user_data' if parent_path.exists(): return parent_path return Path('user_data')