mirror of
https://github.com/azaion/loader.git
synced 2026-04-22 13:16:33 +00:00
Quality cleanup refactoring
Made-with: Cursor
This commit is contained in:
+5
-39
@@ -83,35 +83,15 @@ cdef class ApiClient:
|
||||
role = RoleEnum.NONE
|
||||
self.user = User(id, email, role)
|
||||
|
||||
cdef get_user(self):
|
||||
if self.user is None:
|
||||
self.login()
|
||||
return self.user
|
||||
|
||||
cdef upload_file(self, str filename, bytes resource, str folder):
|
||||
if self.token is None:
|
||||
self.login()
|
||||
url = f"{self.api_url}/resources/{folder}"
|
||||
headers = { "Authorization": f"Bearer {self.token}" }
|
||||
files = {'data': (filename, resource)}
|
||||
try:
|
||||
r = requests.post(url, headers=headers, files=files, allow_redirects=True)
|
||||
r.raise_for_status()
|
||||
constants.log(f"Uploaded {filename} to {self.api_url}/{folder} successfully: {r.status_code}.")
|
||||
except Exception as e:
|
||||
constants.logerror(f"Upload fail: {e}")
|
||||
|
||||
cdef list_files(self, str folder, str search_file):
|
||||
response = self.request('get', f'{self.api_url}/resources/list/{folder}', {
|
||||
"search": search_file
|
||||
}, is_stream=False)
|
||||
constants.log(<str> f'Get files list by {folder}')
|
||||
return response.json()
|
||||
|
||||
cdef check_resource(self):
|
||||
cdef str hardware = HardwareService.get_hardware_info()
|
||||
payload = json.dumps({ "hardware": hardware }, indent=4)
|
||||
response = self.request('post', f'{self.api_url}/resources/check', payload, is_stream=False)
|
||||
r = requests.post(url, headers=headers, files=files, allow_redirects=True)
|
||||
r.raise_for_status()
|
||||
constants.log(f"Uploaded {filename} to {self.api_url}/{folder} successfully: {r.status_code}.")
|
||||
|
||||
cdef load_bytes(self, str filename, str folder):
|
||||
if self.credentials is None:
|
||||
@@ -200,22 +180,8 @@ cdef class ApiClient:
|
||||
|
||||
part_big = resource_encrypted[part_small_size:]
|
||||
|
||||
self.cdn_manager.upload(folder, <str>big_part_name, part_big)
|
||||
if not self.cdn_manager.upload(folder, <str>big_part_name, part_big):
|
||||
raise Exception(f'Failed to upload {big_part_name} to CDN bucket {folder}')
|
||||
with open(path.join(<str>folder, <str>big_part_name), 'wb') as f:
|
||||
f.write(part_big)
|
||||
self.upload_file(small_part_name, part_small, folder)
|
||||
|
||||
cpdef upload_to_cdn(self, str bucket, str filename, bytes file_bytes):
|
||||
if self.cdn_manager is None:
|
||||
raise Exception("CDN manager not initialized. Call set_credentials first.")
|
||||
if not self.cdn_manager.upload(bucket, filename, file_bytes):
|
||||
raise Exception(f"Failed to upload {filename} to CDN bucket {bucket}")
|
||||
|
||||
cpdef download_from_cdn(self, str bucket, str filename):
|
||||
if self.cdn_manager is None:
|
||||
raise Exception("CDN manager not initialized. Call set_credentials first.")
|
||||
if not self.cdn_manager.download(bucket, filename):
|
||||
raise Exception(f"Failed to download {filename} from CDN bucket {bucket}")
|
||||
local_path = path.join(bucket, filename)
|
||||
with open(local_path, 'rb') as f:
|
||||
return f.read()
|
||||
|
||||
Reference in New Issue
Block a user