download_tile.py

importsys
importtime

importlooker_sdk
fromlooker_sdkimportmodels40asmodels
fromlooker_sdk.rtlimporttransport

classMyTransportOptions(transport.PTransportSettings): timeout=300

sdk=looker_sdk.init40("../../looker.ini")


defget_dashboard(title: str):
"""Get a dashboard by title"""
dashboards=sdk.search_dashboards(title=title)
ifnotdashboards:
print(f"dashboard {title} was not found")
returnNone
returndashboards[0]


defget_tile(dash: models.Dashboard, title: str):
"""Find a dashboard tile by title"""
title=title.lower()
assertdash.dashboard_elements
found=None
fortileindash.dashboard_elements:
asserttile.title
iftile.title.lower() ==title:
found=tile
break
else:
print(f"tile {title} of dashboard {dash.title} was not found")
returnfound


defdownload_tile(tile: models.DashboardElement, format: str="png"):
"""Download the dashboard tile in the specified format"""
ifnottile.query_id:
print(f"tile {tile.title} has no query_id")
returnNone
task=sdk.create_query_render_task(
query_id=tile.query_id, result_format=format, width=500, height=500
 )

ifnottaskornottask.id:
print(f"Could not create a render task for {tile.title}")
returnNone

# poll the render task until it completes
elapsed=0.0
delay=0.5# wait .5 seconds
whileTrue:
poll=sdk.render_task(task.id)
ifpoll.status=="failure":
print(poll)
print(f"Render failed for {tile.title}")
returnNone
elifpoll.status=="success":
break

time.sleep(delay)
elapsed+=delay
print(f"{elapsed} seconds elapsed")

result=sdk.render_task_results(
task.id,
# wait up to 300 seconds
transport_options=MyTransportOptions,
 )
fileName=f"{tile.title}.{format}"
withopen(fileName, "wb") asf:
f.write(result)
returnfileName


defmain():
dashboard_title=sys.argv[1] iflen(sys.argv) >1else""
tile_title=sys.argv[2] iflen(sys.argv) >2else""
render_format=sys.argv[3] iflen(sys.argv) >3else"png"
ifnotdashboard_titleornottile_title:
print("Please provide: <dashboardTitle> <titleTitle> [<renderFormat>]")
print(' renderFormat defaults to "png"')
return

dashboard=get_dashboard(dashboard_title)
ifdashboard:
tile=get_tile(dashboard, tile_title)
iftile:
download_tile(tile, render_format)


main()