Complex REST API workflows

from thoughtspot_rest_api_v1 import *

username = 'username'
password = 'password'
server = 'https://{instance}.thoughtspot.cloud'

ts: TSRestApiV2 = TSRestApiV2(server_url=server)

# 1. Find all Liveboards and Answers with a name that includes '(Sample)'

# 2. Add a tag to each item called 'Tutorial Test'

# 1. Find all Liveboards and Answers with a name that includes '(Sample)'

# Get all of the items with names including '(Sample)'
#  Is this a case-sensitive or insensitive operation? Are we finding anywhere in the name or just at start or end?

# 2. Add a tag to each item called 'Tutorial Test'

# Get the ID of the tag called 'Tutorial Test'
#   What if there is no tag called 'Tutorial Test'?

# Assign Tag to each item

# 1. Find all Liveboards and Answers with a name that includes '(Sample)'

# Get all of the items with names including '(Sample)'
#  Is this a case-sensitive or case-insensitive operation? Are we finding anywhere in the name or just at the start or end?

# 2. Add a tag to each item called 'Tutorial Test'

# Get the ID of the tag called 'Tutorial Test'
#   What if there is no tag called 'Tutorial Test'?

# 1. Find all Liveboards and Answers with a name that includes '(Sample)'

# Get all of the items with names including '(Sample)'
#  Is this a case-sensitive or insensitive operation? Are we finding anywhere in the name or just at the start or end?

# Create request to /metadata/search to find the Liveboards and Answers matching the name pattern
# Use the Playground to build your request, then copy the code and paste it in the script
search_request = {
    "metadata": [
    {
      "name_pattern": "(Sample)",
      "type": "ANSWER"
    },
    {
      "name_pattern": "Sample)",
      "type": "LIVEBOARD"
    }
  ],
    'record_offset': 0,
    'record_size': 10000
}

try:
    # Send request to /metadata/search endpoint
    metadata_resp = ts.metadata_search(request=search_request)
except requests.exceptions.HTTPError as e:
    print("Error from the API: ")
    print(e)
    print(e.response.content)
    exit()

# Create List to hold the final set of Answers + Liveboards we want to tag and share
final_list_of_objs =[]

# Iterate through the results from the API response to double-check that the name value matches exactly
for item in metadata_resp:
    m_name = item["metadata_name"]
    m_id = item["metadata_id"]
    # Python string find is Case-Sensitive
    if m_name.find("(Sample)") != -1:
        final_list_of_objs.append(item)  # We'll add the whole object to the new List

# optional print to command line to see what happened
print(json.dumps(final_list_of_objs, indent=2))

# 2. Add a tag to each item called 'Tutorial Test'

# Get the ID of the tag called 'Tutorial Test'
#   What if there is no tag called 'Tutorial Test'?

#
# Find the Tag Identifer so we can assign
# Create new Tag if it doesn't exist
#
try:
    tags = ts.tags_search(tag_identifier="Tutorial Test")
except requests.exceptions.HTTPError as e:
    print("Error from the API: ")
    print(e)
    print(e.response.content)
    exit()

if len(tags) == 0:
    try:
        new_tag = ts.tags_create(name="Tutorial Test")
        tag_id = new_tag['id']
    except requests.exceptions.HTTPError as e:
        print("Error from the API: ")
        print(e)
        print(e.response.content)
        exit()
else:
    tag_id = tags[0]['id']

assign_req = {
  "metadata": [
    {
      "identifier": "identifier4"
    }
  ],
  "tag_identifiers": [
    "tag_identifiers8",
    "tag_identifiers9",
    "tag_identifiers0"
  ]
}

# Assign the tag to the items

try:
   # When we copied from the Playground, the format of the `metadata` section is an array of objects,
   # which needs to be a List of Dicts in Python syntax [ {"identifier": metadata_id}, ...]

   tag_metadata_section = []
   # Iterate through each object and make the Dict in create format
   for obj in final_list_of_objs:
        tag_metadata_section.append({"identifier" : obj['metadata_id']})

   assign_req = {
        "metadata": tag_metadata_section,
        "tag_identifiers": [tag_id]
   }

   assign_resp = ts.tags_assign(requst=assign_req)
except requests.exceptions.HTTPError as e:
    print("Error from the API: ")
    print(e)
    print(e.response.content)
    exit()