import requests

def get_categories(
  url:str, headers, next_page_cursor: str = ""
) -> object:
    """Get all categories."""
    querystring = {"cursor": next_page_cursor}
    response = requests.request(
        "GET",
        url,
        data="",
        headers=headers,
        params=querystring,
    )
    print(response.text)
    return response

token = ""            # Insert bearer access token here
next_page_cursor = "" # Insert nextPageCursor value from previous response here (OPTIONAL: Leave empty string)

host = "https://ws-custom-categories.api.forcepoint.io"
api_version = "1.0.0"
headers = {"Authorization": f"Bearer {token}"}
url = f"{host}/v{api_version}"

get_categories(url, headers, next_page_cursor)

import requests

def get_limits(url:str, headers:object) -> object:
    """Get number of sites, number of custom categories, site limits and custom category limits."""
    response = requests.request(
        "GET",
        url,
        data="",
        headers=headers,
    )
    print(response.text)
    return response

token = ""            # Insert bearer access token here

host = "https://ws-custom-categories.api.forcepoint.io"
api_version = "1.0.0"
headers = {"Authorization": f"Bearer {token}"}
url = f"{host}/v{api_version}/limits"

get_limits(url, headers)

import requests

def get_category_by_id(
  url:str, headers, next_page_cursor: str = ""
) -> object:
    """Get category by id."""
    querystring = {"cursor": next_page_cursor}
    response = requests.request(
        "GET",
        url,
        data="",
        headers=headers,
        params=querystring,
    )
    print(response.text)
    return response

token = ""            # Insert bearer access token here
category_id = 0       # Insert category id here
next_page_cursor = "" # Insert nextPageCursor value from previous response here (OPTIONAL: Leave empty string)

host = "https://ws-custom-categories.api.forcepoint.io"
api_version = "1.0.0"
headers = {"Authorization": f"Bearer {token}"}
url = f"{host}/v{api_version}/{category_id}"

get_category_by_id(url, headers, next_page_cursor)

import requests

def delete_category(url:str, headers: object) -> object:
  """Method to delete a Custom Category."""
  response = requests.request("DELETE", url, headers=headers)
  print(response.text)
  return response

token = ""            # Insert bearer access token here
category_id = 0       # Insert category id here

host = "https://ws-custom-categories.api.forcepoint.io"
api_version = "1.0.0"
headers = {"Authorization": f"Bearer {token}"}
url = f"{host}/v{api_version}/{category_id}"

delete_category(url, headers)

import requests

def get_transaction_status(url: str, headers) -> object:
  """Get transaction status."""
  response = requests.request(
      "GET", url, data="", headers=headers,
  )
  print(response.text)
  return response

token = ""          # Insert bearer access token here
transaction_id = "" # Insert transactionId here

host = "https://ws-custom-categories.api.forcepoint.io"
api_version = "1.0.0"
url = f"{host}/v{api_version}/transaction/{transaction_id}"
headers = {"Authorization": f"Bearer {token}"}

get_transaction_status(url, headers)