GeezerVison is showing github.com — unstract-python-client/tests/client_test.py at main · Zipstack/unstract-python-client

Expand file tree

client_test.py

Copy path

More file actions

Latest commit

History

41 lines (34 loc) · 1.05 KB

client_test.py

Copy path

File metadata and controls

41 lines (34 loc) · 1.05 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

import os

import time

import dotenv

from unstract.api_deployments.client import (

APIDeploymentsClient,

APIDeploymentsClientException,

)

dotenv.load_dotenv()

def main():

try:

adc = APIDeploymentsClient(

api_url=os.getenv("API_URL"),

api_key=os.getenv("UNSTRACT_API_DEPLOYMENT_KEY"),

api_timeout=0,

logging_level="DEBUG",

include_metadata=False,

)

file_paths = os.getenv("TEST_FILES", "").split(",")

response = adc.structure_file(file_paths)

print(response)

if response["pending"]:

while True:

p_response = adc.check_execution_status(

response["status_check_api_endpoint"]

)

print(p_response)

if not p_response["pending"]:

break

print("Sleeping and checking again in 5 seconds..")

time.sleep(5)

except APIDeploymentsClientException as e:

print(e)

if __name__ == "__main__":

main()