-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathclient_test.py
More file actions
41 lines (34 loc) · 1.05 KB
/
client_test.py
File metadata and controls
41 lines (34 loc) · 1.05 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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()