#!/bin/env python
# -*- coding: utf-8 -*-
# need to install requests
import os

import requests
from requests.auth import HTTPDigestAuth


def output(response):
    if response.status_code != 200:
        print('Status:', response.status_code, 'Headers:', response.headers, 'Error Response:', response.text)
        exit()
    print(response.json())

tidb_pub = os.environ.get('TIDB_PUB')
tidb_priv = os.environ.get('TIDB_PRIV')
project_prefix = os.environ.get('TIDB_PROJECT_PREFIX')
dry_run = os.environ.get('DRY_RUN') or False

url = 'https://api.tidbcloud.com/api/v1beta/projects'
headers = {'Content-type': 'application/json'}


# response = requests.get(url, headers=headers, auth=HTTPDigestAuth(tidb_pub,tidb_priv))
for i in range(1,21):
    project_name = f"{project_prefix}{i:02d}"
    json_data = {'name': project_name, 'aws_cmek_enabled': False}
    if dry_run:
        print(json_data)
    else:
        response = requests.post(url, json=json_data, headers=headers, auth=HTTPDigestAuth(tidb_pub,tidb_priv))
        output(response)