prepare_codes.py

import argparse
from pathlib import Path
from typing import List

template_code = """
#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

template <class T> bool chmax(T &a, const T &b) {
  if (a >= b)
    return false;
  a = b;
  return true;
}

template <class T> bool chmin(T &a, const T &b) {
  if (a <= b)
    return false;
  a = b;
  return true;
}

void solve() {}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  solve();
  return 0;
}
"""


def main(args: argparse.Namespace):
    output_dir: Path = args.dir
    problems: List[str] = args.problems
    output_dir.mkdir(parents=True, exist_ok=True)

    for problem in problems:
        problem_path: Path = output_dir / problem
        if problem_path.with_suffix(".cpp").exists():
            print(f"{problem} already exists")
            continue

        problem_path.with_suffix(".cpp").write_text(template_code.lstrip("\n"), encoding="utf-8")

    list_files = [str(file) for file in output_dir.glob("*.cpp")]
    print(f"Created files: {list_files}")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Prepare code for the contest")
    parser.add_argument(
        "--dir",
        type=Path,
        default=Path().cwd(),
        help="Directory to store the codes",
    )
    parser.add_argument(
        "--problems",
        nargs="+",
        default=["a", "b", "c", "d", "e", "f"],
        help="List of problems to prepare",
    )

    args = parser.parse_args()
    main(args)