Spaces:
Runtime error
Runtime error
File size: 3,641 Bytes
b115d50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
from __future__ import annotations
import logging
from enum import Enum
from typing import Any, List, Optional, Type
from pydantic import BaseModel, Field
from steamship.base.client import Client
from steamship.base.model import CamelModel
from steamship.base.request import GetRequest, IdentifierRequest
from steamship.base.request import Request
from steamship.base.request import Request as SteamshipRequest
from steamship.base.response import Response
from steamship.base.response import Response as SteamshipResponse
class ListWorkspacesRequest(Request):
pass
class ListWorkspacesResponse(Response):
workspaces: List[Workspace]
class Workspace(CamelModel):
client: Client = Field(None, exclude=True)
id: str = None
handle: str = None
@classmethod
def parse_obj(cls: Type[BaseModel], obj: Any) -> BaseModel:
# TODO (enias): This needs to be solved at the engine side\
obj = obj["workspace"] if "workspace" in obj else obj
return super().parse_obj(obj)
class CreateRequest(SteamshipRequest):
id: Optional[str] = None
handle: Optional[str] = None
fetch_if_exists: Optional[bool] = None
external_id: Optional[str] = None
external_type: Optional[str] = None
metadata: Optional[str] = None
def delete(self) -> Workspace:
return self.client.post("workspace/delete", IdentifierRequest(id=self.id), expect=Workspace)
@staticmethod
def get(
client: Client, id_: str = None, handle: str = None, fetch_if_exists: bool = None
) -> Workspace:
req = GetRequest(id=id_, handle=handle, fetch_if_exists=fetch_if_exists)
return client.post("workspace/get", req, expect=Workspace)
@staticmethod
def create(
client: Client,
handle: Optional[str] = None,
external_id: Optional[str] = None,
external_type: Optional[str] = None,
metadata: Any = None,
fetch_if_exists: bool = True,
) -> Workspace:
req = Workspace.CreateRequest(
handle=handle,
fetch_if_exists=fetch_if_exists,
external_id=external_id,
external_type=external_type,
metadata=metadata,
)
return client.post("workspace/create", req, expect=Workspace)
def create_signed_url(self, request: SignedUrl.Request) -> SignedUrl.Response:
logging.info(f"Requesting signed URL: {request}")
ret = self.client.post(
"workspace/createSignedUrl", payload=request, expect=SignedUrl.Response
)
logging.debug(f"Got signed URL: {ret}")
return ret
@staticmethod
def list(client: Client, t: str = None) -> ListWorkspacesResponse:
return client.post(
"workspace/list",
ListWorkspacesRequest(type=t),
expect=ListWorkspacesResponse,
)
class SignedUrl:
class Bucket(str, Enum):
EXPORTS = "exports"
IMPORTS = "imports"
USER_DATA = "userData"
PLUGIN_DATA = "pluginData"
APP_DATA = "appData"
class Operation(str, Enum):
READ = "Read"
WRITE = "Write"
class Request(SteamshipRequest):
bucket: SignedUrl.Bucket
filepath: str
operation: SignedUrl.Operation
expires_in_minutes: int = None
class Response(SteamshipResponse):
bucket: str = None
filepath: str = None
operation: str = None
expires_in_minutes: int = None
signed_url: str = Field(None, alias="signedUrl")
SignedUrl.Request.update_forward_refs()
ListWorkspacesResponse.update_forward_refs()
|