update.py

import subprocess
import os
import shutil
import requests
import zipfile


def delete(s: str):
    if os.path.isfile(s):
        os.remove(s)
    if os.path.isdir(s):
        shutil.rmtree(s)


def reset_dir(s: str):
    delete(s)
    os.mkdir(s)


def download_atcoder():
    reset_dir("./tmp")
    url = "https://github.com/atcoder/ac-library/archive/refs/heads/master.zip"
    zipfilename = "tmp/master.zip"
    urlData = requests.get(url).content
    with open(zipfilename, mode="wb") as f:  # wb でバイト型を書き込める
        f.write(urlData)
    subprocess.run(["unzip", "-d", "./tmp/", zipfilename])
    delete("./atcoder")
    subprocess.run(["mv", "./tmp/ac-library-master/atcoder", "./atcoder"])


def download_ou_library():
    reset_dir("./tmp")
    url = "https://github.com/rainbou-kpr/library/archive/refs/heads/main.zip"
    zipfilename = "tmp/main.zip"
    urlData = requests.get(url).content
    with open(zipfilename, mode="wb") as f:  # wb でバイト型を書き込める
        f.write(urlData)
    subprocess.run(["unzip", "-d", "./tmp/", zipfilename])
    delete("./ou-library")
    subprocess.run(["mv", "./tmp/library-main/cpp", "./ou-library"])


def download_sh_library():
    reset_dir("./tmp")
    url = "https://github.com/shogo314/sh-library/archive/refs/heads/main.zip"
    zipfilename = "tmp/main.zip"
    urlData = requests.get(url).content
    with open(zipfilename, mode="wb") as f:  # wb でバイト型を書き込める
        f.write(urlData)
    subprocess.run(["unzip", "-d", "./tmp/", zipfilename])
    delete("./sh-library")
    subprocess.run(["mv", "./tmp/sh-library-main/src", "./sh-library"])


def main():
    download_ou_library()
    download_sh_library()
    delete("./tmp")


if __name__ == "__main__":
    main()