insert_char.py

"""Insert new characters into the roles.json."""

import json

import requests

NIGHTSHEET_URL = "https://script.bloodontheclocktower.com/data/nightsheet.json"
JINXES_URL = "https://script.bloodontheclocktower.com/data/jinx.json"
JSON_FILEPATH = "./public/roles.json"
JSON_STRUCT = {
    "id": "",
    "name": "",
    "team": "",
    "image": "",
    "firstNight": 0,
    "firstNightReminder": "",
    "otherNight": 0,
    "otherNightReminder": "",
    "reminders": [],
    "setup": False,
    "ability": "",
    "jinxes": [],
}


def main():
    """Main function."""
    name = input("Enter new character name: ").strip()
    char_id = name.replace(" ", "").replace("-", "").lower()
    JSON_STRUCT["name"] = name
    JSON_STRUCT["id"] = char_id
    JSON_STRUCT["image"] = (
        f"https://raw.githubusercontent.com/tomozbot/botc-icons/refs/heads/main/WEBP/{char_id}.webp"
    )
    resp = requests.get(NIGHTSHEET_URL, timeout=10)
    resp_json = resp.json()
    first_order, other_order = resp_json["firstNight"], resp_json["otherNight"]
    resp = requests.get(JINXES_URL, timeout=10)
    resp_json = resp.json()
    for entry in resp_json:
        if entry["id"] == char_id:
            JSON_STRUCT["jinxes"] = entry["jinx"]
    with open(JSON_FILEPATH, "r", encoding="utf-8") as json_file:
        role_json = json.load(json_file)
    role_json.append(JSON_STRUCT)

    for role in role_json:
        if role["id"] in first_order:
            role["firstNight"] = first_order.index(role["id"]) + 1
        else:
            role["firstNight"] = 0
        if role["id"] in other_order:
            role["otherNight"] = other_order.index(role["id"]) + 1
        else:
            role["otherNight"] = 0

    with open(JSON_FILEPATH, "w", encoding="utf-8") as json_file:
        json.dump(role_json, json_file)

    print(f"First Night Dusk: {first_order.index('DUSK') + 1}")
    print(f"Other Night Dusk: {other_order.index('DUSK') + 1}")
    print(f"First Night Dawn: {first_order.index('DAWN') + 1}")
    print(f"Other Night Dawn: {other_order.index('DAWN') + 1}")
    print(f"Minion Info: {first_order.index('MINION') + 1}")
    print(f"Demon Info: {first_order.index('DEMON') + 1}")


if __name__ == "__main__":
    main()