Function cmd_write

tools/save_editor.py:648–740

(args: argparse.Namespace)

def cmd_write(args: argparse.Namespace) -> int:
    path = Path(args.file) if args.file else default_save_path()

    if args.from_existing and path.exists():
        data = read_save(path) or fresh_default()
    else:
        data = fresh_default()

    if args.reset:
        data = fresh_default()

    if args.all:
        args.unlock_chars = True
        args.unlock_stages = True
        args.item_switch = True
        args.sound_test = True
        args.complete_1p = True
        args.perfect_btt = True
        args.perfect_btp = True
        args.vs_fill = True
        # ground_mask records unique VS stages played; max it so the save
        # also reflects "every stage has been played at least once."
        if args.ground_mask is None:
            data.ground_mask = GROUND_MASK_ALL

    if args.unlock_chars:
        apply_unlock_chars(data)
    if args.unlock_stages:
        apply_unlock_inishie(data)
    if args.sound_test:
        apply_unlock_soundtest(data)
    if args.item_switch:
        apply_unlock_itemswitch(data)
    if args.complete_1p:
        apply_complete_1p(data, difficulty=args.completion_difficulty,
                          hiscore=args.completion_hiscore)
    if args.perfect_btt:
        apply_perfect_btt(data, time_ticks=args.perfect_time_ticks)
    if args.perfect_btp:
        apply_perfect_btp(data, time_ticks=args.perfect_time_ticks)
    if args.vs_fill:
        apply_vs_fill(
            data,
            ko_each=args.vs_ko_each,
            time_seconds=args.vs_time_seconds,
            damage_given=args.vs_damage_given,
            damage_taken=args.vs_damage_taken,
            selfdestructs=args.vs_selfdestructs,
            games_played=args.vs_games_played,
            avg_player_count=args.vs_avg_players,
        )

    if args.unlock_mask is not None:
        data.unlock_mask = parse_int(args.unlock_mask) & 0xFF
    if args.fighter_mask is not None:
        data.fighter_mask = parse_int(args.fighter_mask) & 0xFFFF
    if args.ground_mask is not None:
        data.ground_mask = parse_int(args.ground_mask) & 0xFFFF