def get_task_status(job_id,
token_file,
server_address="https://api.neuralet.io"):
"""
Get the status of initiated job
:param job_id: Job id of initiated task
:param token_file: The path of token file
:param server_address: Server Address
:return:
"""
url = server_address + "/api/v1/model/status/"
token = token_reader(token_file)
headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": f"Bearer {token}",
}
data = {"job_id": str(job_id)}
data = json.dumps(data)
print(f"Waiting for {url} ...")
r = requests.post(url, headers=headers, data=data)
if r.status_code == 200:
return r.json() # TODO: parse and return
else:
print(f"ERROR! ({r.status_code})")
return None
def get_model(job_id,
token_file,
server_address="https://api.neuralet.io"):
"""
Download the frozen graph of trained model
:param job_id: Job id of initiated task
:param token_file: Path of token file
:param server_address: Server Address
:return:
"""
url = server_address + "/api/v1/file/download/"
token = token_reader(token_file)
headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": f"Bearer {token}",
}
data = {"job_id": str(job_id)}
data = json.dumps(data)
print(f"Waiting for {url} ...")
r = requests.post(url, headers=headers, data=data)
if r.status_code == 200:
if "download_link" in r.json():
download_link = r.json()["download_link"]
print("Downloading model/output file...")
command = 'curl "' + download_link + '" --output output.zip'
os.system(command)
print("The model is saved at output.zip file.")
else:
print(
"The download link is not prepared. Please check your task status and try it after finishing the training process!"
)
else:
print(f"ERROR! ({r.status_code})")
def user_uploads(token_file,
server_address="https://api.neuralet.io",
page_number=1):
"""
Get user uploads
:param token_file: The path of token file
:param server_address: Server Address
:return:
"""
url = server_address + "/api/v1/users/me/uploads/"
token = token_reader(token_file)
headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": f"Bearer {token}",
}
params = {"page": page_number}
print(f"Waiting for {url} ...")
r = requests.get(url, headers=headers, params=params)
if r.status_code == 200:
response = r.json()
pprint(pop_classess_from_response(response))
return response
else:
print(f"ERROR! ({r.status_code})")
return None
def send_config(config_path,
token_file,
server_address="https://api.neuralet.io"):
"""
Send a config to server for starting a new job
:param config_path: Path of .json config file
:param server_address: Server address
:param token_file: Path of token file
:return:
"""
config_json = json_reader(config_path)
url = server_address + "/api/v1/model/train/"
token = token_reader(token_file)
headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": f"Bearer {token}",
}
print(f"Waiting for {url} ...")
r = requests.post(url, headers=headers, data=config_json)
if r.status_code == 200:
job_id = r.json()["job_id"]
print(f"The job is successfully initiated, job_id= {job_id}")
return job_id
else:
print(f"ERROR! ({r.status_code})")
return None
def send_file(file_path,
token_file,
server_address="https://api.neuralet.io",
params=dict):
"""
Upload a file to the server and provide a uuid for further using
:param file_path: .zip file path
:param server_address: Server Address
:return: uuid
"""
url = server_address + "/api/v1/file/upload/"
token = token_reader(token_file)
headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": f"Bearer {token}",
}
print(f"Waiting for {url} ...")
print("Uploading file, please wait it may take a while ...!")
r = requests.get(url=url, headers=headers, params=params)
if r.status_code == 200:
upload_object = r.json()["upload_object"]
post_data = ""
for k, v in upload_object.get("fields").items():
post_data += ' -F "{}"="{}" '.format(k, v)
command = ("curl " + str(upload_object.get("url")) + post_data +
' -F "file=@{}"'.format(file_path))
os.system(command)
uploade_file_dir = r.json()["name"]
return uploade_file_dir
else:
print(f"ERROR! ({r.status_code})")
return None
def user_detail(token_file, server_address="https://api.neuralet.io"):
"""
Get user detail
:param token_file: The path of token file
:param server_address: Server Address
:return:
"""
url = server_address + "/api/v1/users/me/detail/"
token = token_reader(token_file)
headers = {
"Content-Type": "application/json",
"Accept": "application/json",
"Authorization": f"Bearer {token}",
}
print(f"Waiting for {url} ...")
r = requests.get(url, headers=headers)
if r.status_code == 200:
response = r.json()
pprint(response)
return response
else:
print(f"ERROR! ({r.status_code})")
return None