extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-08-09 00:04:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

lint and re-arm the flake8 test

Browse Source
This commit is contained in:
Kenneth Bingham
2025-03-04 19:53:04 -05:00
parent 0851e3e560
commit ef10caa21f
6 changed files with 21 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sdk/python/src/zrok/environment/root.py
View File

@ -48,16 +48,16 @@ class Root:
cfg = Configuration()
cfg.host = apiEndpoint[0] + "/api/v1"
# Update: Configure authentication token
# The token needs to be set with 'key' instead of 'x-token'
# This matches the securityDefinitions in the OpenAPI spec
cfg.api_key["key"] = self.env.Token
# Create the API client with the configured authentication
auth_client = zrok.ApiClient(configuration=cfg)
self.client_version_check(auth_client)
return auth_client
def ApiEndpoint(self) -> ApiEndpoint:

16
sdk/python/src/zrok/share.py
View File

@ -1,4 +1,3 @@
from zrok import model
from zrok_api.api import ShareApi
from zrok.environment.root import Root
from zrok_api.models.auth_user import AuthUser
@ -58,7 +57,7 @@ def CreateShare(root: Root, request: model.ShareRequest) -> model.Share:
zrok = root.Client()
except Exception as e:
raise Exception("error getting zrok client", e)
try:
# Use share_with_http_info to get access to the HTTP info and handle custom response format
share_api = ShareApi(zrok)
@ -66,18 +65,18 @@ def CreateShare(root: Root, request: model.ShareRequest) -> model.Share:
custom_headers = {
'Accept': 'application/json, application/zrok.v1+json'
}
response_data = share_api.share_with_http_info(
body=out,
_headers=custom_headers
)
# Parse response
if hasattr(response_data, 'data') and response_data.data is not None:
res = response_data.data
else:
raise Exception("invalid response from server")
except ApiException as e:
# If it's a content type error, try to parse the raw JSON
if "Unsupported content type: application/zrok.v1+json" in str(e) and hasattr(e, 'body'):
@ -85,11 +84,12 @@ def CreateShare(root: Root, request: model.ShareRequest) -> model.Share:
# Parse the response body directly
res_dict = json.loads(e.body)
# Create a response object with the expected fields
class ShareResponse:
def __init__(self, share_token, frontend_proxy_endpoints):
self.share_token = share_token
self.frontend_proxy_endpoints = frontend_proxy_endpoints
res = ShareResponse(
share_token=res_dict.get('shareToken', ''),
frontend_proxy_endpoints=res_dict.get('frontendProxyEndpoints', [])
@ -149,7 +149,7 @@ def DeleteShare(root: Root, shr: model.Share):
custom_headers = {
'Accept': 'application/json, application/zrok.v1+json'
}
# Use unshare_with_http_info to get access to the HTTP info
share_api.unshare_with_http_info(
body=req,
@ -182,7 +182,7 @@ def ReleaseReservedShare(root: Root, shr: model.Share):
custom_headers = {
'Accept': 'application/json, application/zrok.v1+json'
}
# Use unshare_with_http_info to get access to the HTTP info
share_api.unshare_with_http_info(
body=req,