kscale 0.0.9__tar.gz → 0.0.11__tar.gz

{kscale-0.0.9/kscale.egg-info → kscale-0.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kscale
-Version: 0.0.9
+Version: 0.0.11
 Summary: The kscale project
 Home-page: https://github.com/kscalelabs/kscale
 Author: Benjamin Bolte

{kscale-0.0.9 → kscale-0.0.11}/kscale/__init__.py

@@ -1,6 +1,6 @@
 """Defines the common interface for the K-Scale Python API."""
 
-__version__ = "0.0.9"
+__version__ = "0.0.11"
 
 from pathlib import Path
 

kscale-0.0.11/kscale/store/api.py

@@ -0,0 +1,64 @@
+"""Defines a common interface for the K-Scale Store API."""
+
+from pathlib import Path
+from typing import overload
+
+from kscale.store.gen.api import UploadArtifactResponse
+from kscale.store.urdf import download_urdf, upload_urdf
+from kscale.utils.api_base import APIBase
+
+
+class StoreAPI(APIBase):
+    def __init__(
+        self,
+        *,
+        api_key: str | None = None,
+    ) -> None:
+        super().__init__()
+
+        self.api_key = api_key
+
+    async def artifact_root(self, artifact_id: str) -> Path:
+        return await download_urdf(artifact_id)
+
+    @overload
+    async def urdf_path(self, artifact_id: str) -> Path: ...
+
+    @overload
+    async def urdf_path(self, artifact_id: str, *, throw_if_missing: bool = True) -> Path | None: ...
+
+    async def urdf_path(self, artifact_id: str, *, throw_if_missing: bool = True) -> Path | None:
+        root_dir = await self.artifact_root(artifact_id)
+        urdf_path = next(root_dir.glob("*.urdf"), None)
+        if urdf_path is None and throw_if_missing:
+            raise FileNotFoundError(f"No URDF found for artifact {artifact_id}")
+        return urdf_path
+
+    @overload
+    async def mjcf_path(self, artifact_id: str) -> Path: ...
+
+    @overload
+    async def mjcf_path(self, artifact_id: str, *, throw_if_missing: bool = True) -> Path | None: ...
+
+    async def mjcf_path(self, artifact_id: str, *, throw_if_missing: bool = True) -> Path | None:
+        root_dir = await self.artifact_root(artifact_id)
+        mjcf_path = next(root_dir.glob("*.mjcf"), None)
+        if mjcf_path is None and throw_if_missing:
+            raise FileNotFoundError(f"No MJCF found for artifact {artifact_id}")
+        return mjcf_path
+
+    @overload
+    async def xml_path(self, artifact_id: str) -> Path: ...
+
+    @overload
+    async def xml_path(self, artifact_id: str, *, throw_if_missing: bool = True) -> Path | None: ...
+
+    async def xml_path(self, artifact_id: str, *, throw_if_missing: bool = True) -> Path | None:
+        root_dir = await self.artifact_root(artifact_id)
+        xml_path = next(root_dir.glob("*.xml"), None)
+        if xml_path is None and throw_if_missing:
+            raise FileNotFoundError(f"No XML found for artifact {artifact_id}")
+        return xml_path
+
+    async def upload_urdf(self, listing_id: str, root_dir: Path) -> UploadArtifactResponse:
+        return await upload_urdf(listing_id, root_dir)

{kscale-0.0.9 → kscale-0.0.11}/kscale/store/gen/api.py

@@ -2,7 +2,7 @@
 
 # generated by datamodel-codegen:
 #   filename:  openapi.json
-#   timestamp: 2024-09-04T04:33:58+00:00
+#   timestamp: 2024-10-23T08:09:02+00:00
 
 from __future__ import annotations
 
@@ -34,6 +34,15 @@ class ClientIdResponse(BaseModel):
     client_id: str = Field(..., title="Client Id")
 
 
+class CreateCheckoutSessionRequest(BaseModel):
+    product_id: str = Field(..., title="Product Id")
+    cancel_url: str = Field(..., title="Cancel Url")
+
+
+class CreateCheckoutSessionResponse(BaseModel):
+    session_id: str = Field(..., title="Session Id")
+
+
 class DeleteTokenResponse(BaseModel):
     message: str = Field(..., title="Message")
 
@@ -59,6 +68,7 @@ class GetListingResponse(BaseModel):
     score: int = Field(..., title="Score")
     user_vote: Optional[bool] = Field(..., title="User Vote")
     creator_id: str = Field(..., title="Creator Id")
+    creator_name: Optional[str] = Field(..., title="Creator Name")
 
 
 class GetTokenResponse(BaseModel):
@@ -78,6 +88,26 @@ class GoogleLogin(BaseModel):
     token: str = Field(..., title="Token")
 
 
+class KernelImageResponse(BaseModel):
+    id: str = Field(..., title="Id")
+    user_id: str = Field(..., title="User Id")
+    name: str = Field(..., title="Name")
+    description: Optional[str] = Field(..., title="Description")
+    size: int = Field(..., title="Size")
+    timestamp: int = Field(..., title="Timestamp")
+    is_public: bool = Field(..., title="Is Public")
+    is_official: bool = Field(..., title="Is Official")
+    downloads: int = Field(..., title="Downloads")
+
+
+class KernelImageUploadRequest(BaseModel):
+    name: str = Field(..., title="Name")
+    file: str = Field(..., title="File")
+    is_public: bool = Field(..., title="Is Public")
+    is_official: bool = Field(..., title="Is Official")
+    description: Optional[str] = Field(None, title="Description")
+
+
 class Permission(Enum):
     read = "read"
     write = "write"
@@ -138,6 +168,7 @@ class LoginResponse(BaseModel):
 
 class Permission1(Enum):
     is_admin = "is_admin"
+    is_mod = "is_mod"
 
 
 class MyUserInfoResponse(BaseModel):
@@ -171,6 +202,47 @@ class NewListingResponse(BaseModel):
     listing_id: str = Field(..., title="Listing Id")
 
 
+class Status(Enum):
+    processing = "processing"
+    in_development = "in_development"
+    being_assembled = "being_assembled"
+    shipped = "shipped"
+    delivered = "delivered"
+    cancelled = "cancelled"
+    refunded = "refunded"
+    failed = "failed"
+
+
+class Order(BaseModel):
+    id: str = Field(..., title="Id")
+    user_id: str = Field(..., title="User Id")
+    user_email: str = Field(..., title="User Email")
+    stripe_checkout_session_id: str = Field(..., title="Stripe Checkout Session Id")
+    stripe_payment_intent_id: str = Field(..., title="Stripe Payment Intent Id")
+    created_at: int = Field(..., title="Created At")
+    updated_at: int = Field(..., title="Updated At")
+    status: Status = Field(..., title="Status")
+    amount: int = Field(..., title="Amount")
+    currency: str = Field(..., title="Currency")
+    quantity: int = Field(..., title="Quantity")
+    product_id: Optional[str] = Field(None, title="Product Id")
+    shipping_name: Optional[str] = Field(None, title="Shipping Name")
+    shipping_address_line1: Optional[str] = Field(None, title="Shipping Address Line1")
+    shipping_address_line2: Optional[str] = Field(None, title="Shipping Address Line2")
+    shipping_city: Optional[str] = Field(None, title="Shipping City")
+    shipping_state: Optional[str] = Field(None, title="Shipping State")
+    shipping_postal_code: Optional[str] = Field(None, title="Shipping Postal Code")
+    shipping_country: Optional[str] = Field(None, title="Shipping Country")
+
+
+class ProductInfo(BaseModel):
+    id: str = Field(..., title="Id")
+    name: str = Field(..., title="Name")
+    description: Optional[str] = Field(..., title="Description")
+    images: List[str] = Field(..., title="Images")
+    metadata: Dict[str, str] = Field(..., title="Metadata")
+
+
 class PublicUserInfoResponseItem(BaseModel):
     id: str = Field(..., title="Id")
     email: str = Field(..., title="Email")
@@ -187,6 +259,11 @@ class PublicUsersInfoResponse(BaseModel):
     users: List[PublicUserInfoResponseItem] = Field(..., title="Users")
 
 
+class SetModeratorRequest(BaseModel):
+    user_id: str = Field(..., title="User Id")
+    is_mod: bool = Field(..., title="Is Mod")
+
+
 class SetRequest(BaseModel):
     onshape_url: Optional[str] = Field(..., title="Onshape Url")
 
@@ -220,7 +297,6 @@ class SingleArtifactResponse(BaseModel):
     description: Optional[str] = Field(..., title="Description")
     timestamp: int = Field(..., title="Timestamp")
     urls: ArtifactUrls
-    is_new: Optional[bool] = Field(None, title="Is New")
 
 
 class SortOption(Enum):
@@ -241,6 +317,16 @@ class UpdateListingRequest(BaseModel):
     tags: Optional[List[str]] = Field(None, title="Tags")
 
 
+class UpdateOrderAddressRequest(BaseModel):
+    shipping_name: str = Field(..., title="Shipping Name")
+    shipping_address_line1: str = Field(..., title="Shipping Address Line1")
+    shipping_address_line2: Optional[str] = Field(..., title="Shipping Address Line2")
+    shipping_city: str = Field(..., title="Shipping City")
+    shipping_state: str = Field(..., title="Shipping State")
+    shipping_postal_code: str = Field(..., title="Shipping Postal Code")
+    shipping_country: str = Field(..., title="Shipping Country")
+
+
 class UpdateUserRequest(BaseModel):
     email: Optional[str] = Field(None, title="Email")
     password: Optional[str] = Field(None, title="Password")
@@ -256,6 +342,11 @@ class UploadArtifactResponse(BaseModel):
     artifacts: List[SingleArtifactResponse] = Field(..., title="Artifacts")
 
 
+class UpvotedListingsResponse(BaseModel):
+    upvoted_listing_ids: List[str] = Field(..., title="Upvoted Listing Ids")
+    has_more: bool = Field(..., title="Has More")
+
+
 class UserInfoResponseItem(BaseModel):
     id: str = Field(..., title="Id")
     email: str = Field(..., title="Email")
@@ -299,3 +390,8 @@ class HTTPValidationError(BaseModel):
 
 class ListArtifactsResponse(BaseModel):
     artifacts: List[SingleArtifactResponse] = Field(..., title="Artifacts")
+
+
+class OrderWithProduct(BaseModel):
+    order: Order
+    product: ProductInfo

{kscale-0.0.9 → kscale-0.0.11}/kscale/store/utils.py

@@ -13,7 +13,7 @@ def get_api_root() -> str:
     Returns:
         The base URL for the K-Scale Store API.
     """
-    return os.getenv("KSCALE_API_ROOT", "https://api.kscale.store")
+    return os.getenv("KSCALE_API_ROOT", "https://api.kscale.dev")
 
 
 def get_api_key() -> str:

{kscale-0.0.9 → kscale-0.0.11/kscale.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kscale
-Version: 0.0.9
+Version: 0.0.11
 Summary: The kscale project
 Home-page: https://github.com/kscalelabs/kscale
 Author: Benjamin Bolte

kscale-0.0.9/kscale/store/api.py

@@ -1,24 +0,0 @@
-"""Defines a common interface for the K-Scale Store API."""
-
-from pathlib import Path
-
-from kscale.store.gen.api import UploadArtifactResponse
-from kscale.store.urdf import download_urdf, upload_urdf
-from kscale.utils.api_base import APIBase
-
-
-class StoreAPI(APIBase):
-    def __init__(
-        self,
-        *,
-        api_key: str | None = None,
-    ) -> None:
-        super().__init__()
-
-        self.api_key = api_key
-
-    async def urdf(self, artifact_id: str) -> Path:
-        return await download_urdf(artifact_id)
-
-    async def upload_urdf(self, listing_id: str, root_dir: Path) -> UploadArtifactResponse:
-        return await upload_urdf(listing_id, root_dir)