template-create.py

import csv
import requests
import json
import sys


def get_base_url(env):
    if env == "local":
        return "http://127.0.0.1:8000"
    elif env == "prod":
        return "https://web.attentive.ai"
    else:
        return f"https://{env}.web.attentive.ai"


def get_branches(env, company_id, headers):
    if env == "local":
        base_url = get_base_url("dev1")
    else:
        base_url = get_base_url(env)
    branch_url = f"{base_url}/core/api/v1/branch/?company={company_id}"
    print(f"branch_url: {branch_url}")
    branch_response = requests.get(branch_url, headers=headers)
    print(f"\nStatus Code: {branch_response.status_code}\n")
    return branch_response.json()


def create_template(env, branch_id, division_id, html_text, headers, name):
    base_url = get_base_url(env)
    url = f"{base_url}/payments/api/v1/templates/"
    data = {
        "branch_id": branch_id,
        "division_id": division_id,
        "type": 3,
        "name": f"Group Invoice Template - {name}",
        "html_text": html_text,
    }
    response = requests.post(url, headers=headers, data=json.dumps(data))
    return response.status_code, response.json()


def update_template(env, template_id, placeholders, headers):
    base_url = get_base_url(env)
    url = f"{base_url}/payments/api/v1/templates/{template_id}/"
    print(f"url: {url}")
    response = requests.patch(url, headers=headers, data=json.dumps(placeholders))
    return response.status_code, response.json()


def main():
    env = sys.argv[1]
    csv_file_path = "companies.csv"

    with open("html_text.txt", "r") as file:
        html_text = file.read()

    with open("placeholders.json", "r") as file:
        placeholders = json.load(file)

    with open(csv_file_path, newline="") as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            company_id = row["id"]
            name = row["name"]
            token = row["token"]
            headers = {
                "Authorization": f"Bearer {token}",
                "Content-Type": "application/json",
            }

            print(f"Processing Company: {name} with ID: {company_id}")

            branches = get_branches(env, company_id, headers)

            for branch in branches:
                for division in branch["divisions"]:
                    status_code, response_json = create_template(
                        env, branch["id"], division["id"], html_text, headers, name
                    )
                    print(
                        f"-------------------\nTemplate Create Status Code: {status_code}\n-------------------"
                    )
                    if status_code == 201:
                        template_id = response_json["id"]
                        print(f"Template ID: {template_id}")
                        status_code, response_json = update_template(
                            env, template_id, placeholders, headers
                        )
                        print(
                            f"-------------------\nTemplate Update Status Code: {status_code}\n-------------------"
                        )


if __name__ == "__main__":
    main()