refactor: query auth use CredentialInput instead of str

executor/python_executor/credential.py

@@ -1,9 +1,6 @@
 from typing import Any, Dict
-from oocana import InputHandleDef
-class CredentialInput:
-    def __init__(self, type: str, id: str):
-        self.type = type
-        self.id = id
+from oocana import InputHandleDef, CredentialInput
+
 
 def generate_credential_input(credential_path: str) -> CredentialInput | None:
     """Generate a CredentialInput from a credential path string.

oocana/oocana/__init__.py

@@ -6,4 +6,5 @@
 from .extra import *  # noqa: F403
 from .schema import *  # noqa: F403
 from .mainframe import Mainframe as Mainframe  # noqa: F403
-from .serialization import setup_dataframe_serialization, CompressionOptions  # noqa: F403
+from .serialization import setup_dataframe_serialization, CompressionOptions  # noqa: F403
+from .credential import CredentialInput

oocana/oocana/context.py

@@ -11,6 +11,7 @@
 from .preview import PreviewPayload, DataFrame, PreviewPayloadInternal, ShapeDataFrame
 from .data import EXECUTOR_NAME
 from .internal import random_string, InternalAPI
+from .credential import CredentialInput
 import os.path
 import logging
 import hashlib
@@ -680,7 +681,7 @@ def response_callback(payload: Dict[str, Any]):
 
         return await f
 
-    async def query_auth(self, id: str) -> Dict[str, Any]:
+    async def query_auth(self, credential: CredentialInput) -> Dict[str, Any]:
         request_id = random_string(16)
         loop = asyncio.get_running_loop()
         f: asyncio.Future[Dict[str, Any]] = loop.create_future()
@@ -699,7 +700,7 @@ def response_callback(payload: Dict[str, Any]):
         self.__mainframe.send(self.job_info, {
             "type": "BlockRequest",
             "action": "QueryAuth",
-            "id": id,
+            "id": credential.id,
             "session_id": self.session_id,
             "job_id": self.job_id,
             "request_id": request_id,

oocana/oocana/credential.py

@@ -0,0 +1,4 @@
+class CredentialInput:
+    def __init__(self, type: str, id: str):
+        self.type = type
+        self.id = id