main.py

import sys

from src.config import parse_args
from src.attacks import (
    run_individual_suffix_attack,
    run_population_suffix_attack,
    run_individual_replace_attack,
    run_population_replace_attack,
)


def launch():
    args = parse_args()

    if args.level == "individual":
        if args.mode == "suffix":
            success_rate = run_individual_suffix_attack(args)
        else:
            success_rate = run_individual_replace_attack(args)
        print(f"Individual attack success rate = {success_rate:.4f}")

    else:
        if args.mode == "suffix":
            d_overlap = run_population_suffix_attack(args)
        else:
            d_overlap = run_population_replace_attack(args)
        print(f"Population attack overlap change = {d_overlap:.4f}")

    if args.log:
        sys.stdout.close()


if __name__ == "__main__":
    launch()