簡體   English   中英

Get data as dataframe from Look through API sdk package in python

[英]Get data as dataframe from Look through API sdk package in python

我有以下代碼可以通過 API 從 Looker 中查看。 我將 api 存儲在 a.ini 中。 所有這些都有效,但現在我想從 python 中的 Dataframe 中獲取數據,以便我可以使用這些數據進行進一步分析。 我怎樣才能做到這一點? 我使用了此代碼,但這僅將其保存為 png。 我找不到從中創建 dataframe 的方法,因為我想要數據本身而不僅僅是結果圖像。

import sys
import textwrap
import time

import looker_sdk
from looker_sdk import models

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

def get_look(title: str) -> models.Look:
    title = title.lower()
    look = next(iter(sdk.search_looks(title=title)), None)
    if not look:
        raise Exception(f"look '{title}' was not found")
    return look
     def download_look(look: models.Look, result_format: str, width: int, height: int):
        """Download specified look as png/jpg"""
        id = int(look.id)
        task = sdk.create_look_render_task(id, result_format, width, height,)
    
        if not (task and task.id):
            raise sdk.RenderTaskError(
                f"Could not create a render task for '{look.title}'"
            )
    
        # poll the render task until it completes
        elapsed = 0.0
        delay = 0.5  # wait .5 seconds
        while True:
            poll = sdk.render_task(task.id)
            if poll.status == "failure":
                print(poll)
                raise Exception(f"Render failed for '{look.title}'")
            elif poll.status == "success":
                break
            time.sleep(delay)
            elapsed += delay
        print(f"Render task completed in {elapsed} seconds")
    
        result = sdk.render_task_results(task.id)
        filename = f"{look.title}.{result_format}"
        with open(filename, "wb") as f:
            f.write(result)
        print(f"Look saved to '{filename}'")
    
    
    look_title = sys.argv[1] if len(sys.argv) > 1 else "Name"
    image_width = int(sys.argv[2]) if len(sys.argv) > 2 else 545
    image_height = int(sys.argv[3]) if len(sys.argv) > 3 else 842
    image_format = sys.argv[4] if len(sys.argv) > 4 else "png"
    
    if not look_title:
        raise Exception(
            textwrap.dedent(
                    """
                  Please provide: <lookTitle> [<img_width>] [<img_height>] [<img_format>]
                    img_width defaults to 545
                    img_height defaults to 842
                    img_format defaults to 'png'"""
                )
            )
    
    
    look = get_look(look_title)
    #Dataframe storage 
    
    download_look(look, image_format, image_width, image_height)

您正在使用的 SDK function ( create_look_render_task ), 此處描述的僅允許您以 pdf 或 jpg 格式下載。

如果您想從查看 dataframe 中獲取數據,那么您可能希望使用run_look function 進行查看,而不是在此處描述。 當您使用run_look時,您可以將result_format更改為 CSV,然后編寫您自己的代碼以轉換為 dataframe。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM