refine tos_url

Author: ZQlQZ

veadk/integrations/ve_tos/ve_tos.py

@@ -242,6 +242,18 @@ def build_tos_url(self, object_key: str, bucket_name: str = "") -> str:
         )
         return tos_url
 
+    def build_tos_signed_url(self, object_key: str, bucket_name: str = "") -> str:
+        bucket_name = self._check_bucket_name(bucket_name)
+
+        out = self._client.pre_signed_url(
+            self._tos_module.HttpMethodType.Http_Method_Get,
+            bucket=bucket_name,
+            key=object_key,
+            expires=604800,
+        )
+        tos_url = out.signed_url
+        return tos_url
+
     # deprecated
     def upload(
         self, data: Union[str, bytes], bucket_name: str = "", object_key: str = ""
@@ -347,7 +359,7 @@ def upload_bytes(
         bucket_name = self._check_bucket_name(bucket_name)
         if not object_key:
             object_key = self._build_object_key_for_bytes()
-        # Use common function to check client and bucket
+        # Use common function to check lient and bucket
         if not self._ensure_client_and_bucket(bucket_name):
             return
         try:

veadk/runner.py

@@ -146,7 +146,7 @@ async def _upload_image_to_tos(
             filename = os.path.basename(part.inline_data.display_name)
             object_key = f"{app_name}/{user_id}-{session_id}-{filename}"
             ve_tos = VeTOS()
-            tos_url = ve_tos.build_tos_url(object_key=object_key)
+            tos_url = ve_tos.build_tos_signed_url(object_key=object_key)
             await ve_tos.async_upload_bytes(
                 object_key=object_key,
                 data=part.inline_data.data,