Skip to content

Commit b6f42f2

Browse files
authored
refine(tos): add self-signed url in TOS (#196)
* refine tos_url * fix typo
1 parent 8b259c3 commit b6f42f2

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

veadk/integrations/ve_tos/ve_tos.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,18 @@ def build_tos_url(self, object_key: str, bucket_name: str = "") -> str:
242242
)
243243
return tos_url
244244

245+
def build_tos_signed_url(self, object_key: str, bucket_name: str = "") -> str:
246+
bucket_name = self._check_bucket_name(bucket_name)
247+
248+
out = self._client.pre_signed_url(
249+
self._tos_module.HttpMethodType.Http_Method_Get,
250+
bucket=bucket_name,
251+
key=object_key,
252+
expires=604800,
253+
)
254+
tos_url = out.signed_url
255+
return tos_url
256+
245257
# deprecated
246258
def upload(
247259
self, data: Union[str, bytes], bucket_name: str = "", object_key: str = ""

veadk/runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ async def _upload_image_to_tos(
146146
filename = os.path.basename(part.inline_data.display_name)
147147
object_key = f"{app_name}/{user_id}-{session_id}-{filename}"
148148
ve_tos = VeTOS()
149-
tos_url = ve_tos.build_tos_url(object_key=object_key)
149+
tos_url = ve_tos.build_tos_signed_url(object_key=object_key)
150150
await ve_tos.async_upload_bytes(
151151
object_key=object_key,
152152
data=part.inline_data.data,

0 commit comments

Comments
 (0)