import logging

import requests
from requests.auth import HTTPDigestAuth as DigestAuth

# auth = DigestAuth("K1273CQ0", "79e90af5-b039-46b0-8c39-36d805ffc00e")
# logging.basicConfig(level=logging.INFO)
# base_url = "https://us-west-2.data.dev.tidbcloud.com/api/v1beta/app/chat2query-oRPXPRMQ/endpoint"


auth = DigestAuth("813JPI50", "e695644c-e41a-4358-a053-3867255da01c")
logging.basicConfig(level=logging.INFO)
base_url = "https://us-west-2.data.tidbcloud.com/api/v1beta/app/chat2query-uWVhrQjt/endpoint"


def list_domain_items():
    url = f"{base_url}/v2/domain_context/domain_items"
    resp = requests.get(url, auth=auth)
    logging.info("list_domain_items: %s", resp.text)


def create_domain_items():
    url = f"{base_url}/v2/domain_context/domain_items"
    payload = {
        "type": "instruction",
        "meta_data": {},
        "raw_data": {
            "instruction": "涉及对店名、类目名称、省份名称、城市名称、所在区名称的筛选时, 必须采用模糊匹配以确保结果的准确性(like '%xx%')"
        }
    }
    resp = requests.post(url, json=payload, auth=auth)
    logging.info("create_domain_items: %s", resp.text)


def update_domain_items(domain_item_id: int):
    url = f"{base_url}/v2/domain_context/domain_items/{domain_item_id}"
    payload = {
        "meta_data": {},
        "raw_data": {
            "instruction": "涉及对店名、类目名称、省份名称、城市名称、所在区名称的筛选时, 必须采用模糊匹配以确保结果的准确性(like '%xx%')"
        }
    }
    resp = requests.put(url, json=payload, auth=auth)
    logging.info("update_domain_items: %s", resp.text)


def delete_domain_items(domain_item_id: int):
    url = f"{base_url}/v2/domain_context/domain_items/{domain_item_id}"
    resp = requests.delete(url, auth=auth)
    logging.info("delete_domain_items: %s", resp.text)


if __name__ == "__main__":
    # list_domain_items()
    create_domain_items()
    # update_domain_items(7)
    # delete_domain_items(7)
    # list_domain_items()