kscale 0.0.9__py3-none-any.whl → 0.0.11__py3-none-any.whl

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/store/api.py

@@ -1,6 +1,7 @@
 """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
@@ -17,8 +18,47 @@ class StoreAPI(APIBase):
 
         self.api_key = api_key
 
-    async def urdf(self, artifact_id: str) -> Path:
+    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/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/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.dist-info → kscale-0.0.11.dist-info}/METADATA

@@ -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.dist-info → kscale-0.0.11.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kscale/__init__.py,sha256=dcfdPhyC9nhbFk5nRj-R299UnVGP9ZLrZLgIGWgE5P0,177
+kscale/__init__.py,sha256=4m3jAZmGYlh2S1MiGDtND17NHYndlo25f3NsHy6M3RI,178
 kscale/api.py,sha256=xBtKj8rgZ400r1Xx9LRY0AzSgIIttoXdejmhHhdVGS0,333
 kscale/conf.py,sha256=9fShFaYTbnrm_eiGjmy8ZtC4Q4m6PQkWPyoF3eNyov8,1424
 kscale/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -8,19 +8,19 @@ kscale/artifacts/__init__.py,sha256=RK8wdybtCJPgdLLJ8R8-YMi1Ph5ojqAKVJZowHONtgo,
 kscale/artifacts/plane.obj,sha256=x59-IIrWpLjhotChiqT2Ul6U8s0RcHkaEeUZb4KXL1c,348
 kscale/artifacts/plane.urdf,sha256=LCiTk14AyTHjkZ1jvsb0hNaEaJUxDb8Z1JjsgpXu3YM,819
 kscale/store/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kscale/store/api.py,sha256=rydanYGg8zlc_cNuWo9YBG_WVtPKMxFgp6fpec-B8M0,671
+kscale/store/api.py,sha256=JBp4h6yz_ESSdB-2FpkD_1-hdI8_iEKh9svzkyy3jhs,2386
 kscale/store/cli.py,sha256=8ygg_1tZzOOHJotEIgSN9pfumcriPmA31sI_FCFQiTo,859
 kscale/store/client.py,sha256=R1IDnf2J4ojAcP8nmUUHfXhcHUt4zP0-mxtVI7MIC5U,2664
 kscale/store/pybullet.py,sha256=zoeATQStuRWgmPhku65xjfgvE3Y8ReheUIAkZnDr2C0,7614
 kscale/store/urdf.py,sha256=5x8tK2BYv901S_yYWYPWEnHv-3T0ALBQMdDwb70EZFw,6395
-kscale/store/utils.py,sha256=rFXGkem2oAttAf3bhWmFEhxrqYnaVvlVJsC268IMw6Y,906
+kscale/store/utils.py,sha256=euePSkUllLgbJa4RXrCAhB65-BFJOhEImH0ipDs265w,904
 kscale/store/gen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kscale/store/gen/api.py,sha256=82D41J6pg9KWdgD0lx7NggLcNS32SpnN8DqE3Md6ON0,9559
+kscale/store/gen/api.py,sha256=GnNTEfFPXVkr0GuZDCCTx3HZ6VD2Yg1ImWpYV-QfmlM,13323
 kscale/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kscale/utils/api_base.py,sha256=Kk_WtRDdJHmOg6NtHmVxVrcfARSUkhfr29ypLch_pO0,112
-kscale-0.0.9.dist-info/LICENSE,sha256=HCN2bImAzUOXldAZZI7JZ9PYq6OwMlDAP_PpX1HnuN0,1071
-kscale-0.0.9.dist-info/METADATA,sha256=3-UEHjXp3IQo3Cdn7drrOvciylD5MnoOl2gm1V4P4Iw,2504
-kscale-0.0.9.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-kscale-0.0.9.dist-info/entry_points.txt,sha256=PaVs1ivqB0BBdGUsiFkxGUYjGLz05VqagxwRVwi4yV4,54
-kscale-0.0.9.dist-info/top_level.txt,sha256=C2ynjYwopg6YjgttnI2dJjasyq3EKNmYp-IfQg9Xms4,7
-kscale-0.0.9.dist-info/RECORD,,
+kscale-0.0.11.dist-info/LICENSE,sha256=HCN2bImAzUOXldAZZI7JZ9PYq6OwMlDAP_PpX1HnuN0,1071
+kscale-0.0.11.dist-info/METADATA,sha256=-BE86Y4RVRIVbvfakjSFbU6QcELc3cFQYofVB01_NkE,2505
+kscale-0.0.11.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+kscale-0.0.11.dist-info/entry_points.txt,sha256=PaVs1ivqB0BBdGUsiFkxGUYjGLz05VqagxwRVwi4yV4,54
+kscale-0.0.11.dist-info/top_level.txt,sha256=C2ynjYwopg6YjgttnI2dJjasyq3EKNmYp-IfQg9Xms4,7
+kscale-0.0.11.dist-info/RECORD,,

{kscale-0.0.9.dist-info → kscale-0.0.11.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any