import labelbox as lb
import uuid

API_KEY = ""
# To get your API key go to: Workspace settings -> API -> Create API Key
client = lb.Client(api_key=API_KEY)

catalog_slice = client.get_catalog_slice(catalog_slice_id)

# May return a list of slices if multiple slices have the same name
catalog_slice = client.get_catalog_slice(name="slice to search")

catalog_slices = client.get_catalog_slices()

# Get a data row id
slice_data_rows_ids = catalog_slice.get_data_row_ids()

# Get a data row objects
for data_row_id in slice_data_rows_ids:
  print(client.get_data_row(data_row_id))

data_row_identifiers = catalog_slice.get_data_row_identifiers()

drids = [dr for dr in data_row_identifiers]

# get both global keys and data row ids
# and utilize the hash method to combine both global keys and data row ids into a dictionary
for dr in drids:
  print(f"Data row: {dr.id}, Global Key: {dr.global_key}, dr_gk: {dr.to_hash()}")

# Optional: sample Data rows from your Slice
sampled_data_row_ids = random.sample(slice_data_rows_ids, 5)

batch = project.create_batch(
  "test batch", # name of the batch
  sampled_data_row_ids, # list of Data Rows
  1 # priority between 1-5
)

model_run.upsert_data_rows(list(slice_data_rows_ids))