mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 22:06:36 +00:00
zxsanny
95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
import io
|
|
import json
|
|
from http import HTTPStatus
|
|
|
|
import requests
|
|
|
|
import constants
|
|
from hardware_service import get_hardware_info
|
|
from security import Security
|
|
|
|
|
|
class ApiCredentials:
|
|
def __init__(self, url, email, password, folder):
|
|
self.url = url
|
|
self.email = email
|
|
self.password = password
|
|
self.folder = folder
|
|
|
|
|
|
class Api:
|
|
def __init__(self, credentials):
|
|
self.token = None
|
|
self.credentials = credentials
|
|
|
|
def login(self):
|
|
response = requests.post(f'{self.credentials.url}/login',
|
|
json={"email": self.credentials.email, "password": self.credentials.password})
|
|
response.raise_for_status()
|
|
token = response.json()["token"]
|
|
self.token = token
|
|
|
|
def upload_file(self, filename: str, file_bytes: bytearray):
|
|
folder = self.credentials.folder
|
|
if self.token is None:
|
|
self.login()
|
|
url = f"{self.credentials.url}/resources/{folder}"
|
|
headers = {"Authorization": f"Bearer {self.token}"}
|
|
files = {'data': (filename, io.BytesIO(file_bytes))}
|
|
try:
|
|
r = requests.post(url, headers=headers, files=files, allow_redirects=True)
|
|
r.raise_for_status()
|
|
print(f"Upload {len(file_bytes)} bytes ({filename}) to {self.credentials.url}. Result: {r.status_code}")
|
|
except Exception as e:
|
|
print(f"Upload fail: {e}")
|
|
|
|
def load_bytes(self, filename, folder = None):
|
|
folder = folder or self.credentials.folder
|
|
|
|
hardware = get_hardware_info()
|
|
|
|
if self.token is None:
|
|
self.login()
|
|
url = f"{self.credentials.url}/resources/get/{folder}"
|
|
headers = {
|
|
"Authorization": f"Bearer {self.token}",
|
|
"Content-Type": "application/json"
|
|
}
|
|
|
|
payload = json.dumps(
|
|
{
|
|
"password": self.credentials.password,
|
|
"hardware": hardware.to_json_object(),
|
|
"fileName": filename
|
|
}, indent=4)
|
|
response = requests.post(url, data=payload, headers=headers, stream=True)
|
|
if response.status_code == HTTPStatus.UNAUTHORIZED or response.status_code == HTTPStatus.FORBIDDEN:
|
|
self.login()
|
|
headers = {
|
|
"Authorization": f"Bearer {self.token}",
|
|
"Content-Type": "application/json"
|
|
}
|
|
response = requests.post(url, data=payload, headers=headers, stream=True)
|
|
|
|
if response.status_code == HTTPStatus.INTERNAL_SERVER_ERROR:
|
|
print('500!')
|
|
|
|
hw_hash = Security.get_hw_hash(hardware)
|
|
key = Security.get_api_encryption_key(self.credentials, hw_hash)
|
|
|
|
resp_bytes = response.raw.read()
|
|
data = Security.decrypt_to(resp_bytes, key)
|
|
print(f'Downloaded file: {filename}, {len(data)} bytes')
|
|
return data
|
|
|
|
def load_resource(self, big_part, small_part):
|
|
with open(big_part, 'rb') as binary_file:
|
|
encrypted_bytes_big = binary_file.read()
|
|
encrypted_bytes_small = self.load_bytes(small_part)
|
|
|
|
encrypted_model_bytes = encrypted_bytes_small + encrypted_bytes_big
|
|
key = Security.get_model_encryption_key()
|
|
|
|
model_bytes = Security.decrypt_to(encrypted_model_bytes, key)
|
|
return model_bytes
|