snap-python 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

snap_python/components/store.py

@@ -1,5 +1,6 @@
 import functools
 import uuid
+from typing import Optional
 
 import retry
 from httpx import AsyncClient, Response
@@ -77,8 +78,19 @@ class StoreEndpoints:
         response.raise_for_status()
         return response.json()
 
+    async def get_snap_name_from_snap_id(self, snap_id: str) -> str | None:
+        response = await self.store_client.get(
+            f"https://api.snapcraft.io/v2/assertions/snap-declaration/16/{snap_id}"
+        )
+        response.raise_for_status()
+        response_json = response.json()
+        return response_json.get("headers", {}).get("snap-name", None)
+
     async def get_snap_info(
-        self, snap_name: str, fields: list[str] | None = None
+        self,
+        snap_name: Optional[str] = None,
+        snap_id: Optional[str] = None,
+        fields: list[str] | None = None,
     ) -> InfoResponse:
         """
         Get information about a snap.
@@ -93,6 +105,15 @@ class StoreEndpoints:
 
         :raises ValueError: If invalid fields are provided.
         """
+        if snap_name is None and snap_id is None:
+            raise ValueError("Either snap_name or snap_id must be provided.")
+        if snap_name is not None and snap_id is not None:
+            raise ValueError("Only one of snap_name or snap_id must be provided.")
+
+        # get snap name from id, then proceed as usual
+        if snap_id is not None:
+            snap_name = await self.get_snap_name_from_snap_id(snap_id)
+
         query = {}
         if fields is not None:
             if not all(field in VALID_SNAP_INFO_FIELDS for field in fields):

snap_python/schemas/common.py

@@ -52,7 +52,7 @@ class SnapApp(BaseModel):
 
 
 class Media(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     height: Optional[float] = None
     type: str
@@ -65,13 +65,13 @@ class AsyncResponse(BaseResponse):
 
 
 class Download(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     size: float
 
 
 class Revision(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     base: Optional[SnapBaseVersion] = None
     channel: Optional[str] = None
@@ -86,7 +86,7 @@ class Revision(BaseModel):
 
 
 class Publisher(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     display_name: str = Field(
         ...,

snap_python/schemas/snaps.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from pydantic import AliasChoices, AwareDatetime, BaseModel, ConfigDict, Field
+from pydantic import AliasChoices, AwareDatetime, BaseModel, Field
 
 from snap_python.schemas.common import (
     BaseErrorResult,
@@ -66,8 +66,6 @@ class StoreSnapFields(BaseModel):
 
 
 class Snap(BaseModel):
-    model_config = ConfigDict(extra="forbid")
-
     contact: Optional[str] = None
     description: Optional[str] = None
     license: str = "unset"

snap_python/schemas/store/categories.py

@@ -10,7 +10,7 @@ VALID_CATEGORY_FIELDS = ["name", "type", "title", "summary", "description", "med
 
 
 class Media(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     height: Optional[float] = None
     type: str
@@ -19,7 +19,7 @@ class Media(BaseModel):
 
 
 class Category(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     description: Optional[str] = None
     media: Optional[Media] = None
@@ -31,12 +31,12 @@ class Category(BaseModel):
 
 
 class CategoryResponse(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     categories: Optional[List[Category]] = None
 
 
 class SingleCategoryResponse(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     category: Category

snap_python/schemas/store/info.py

@@ -45,7 +45,7 @@ VALID_SNAP_INFO_FIELDS = [
 
 
 class Channel(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     architecture: str
     name: str
@@ -59,7 +59,7 @@ class Channel(BaseModel):
 
 
 class Delta(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     format: str
     sha3_384: Optional[str] = Field(
@@ -72,7 +72,7 @@ class Delta(BaseModel):
 
 
 class Download(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     deltas: List[Delta]
     sha3_384: Optional[str] = Field(
@@ -83,7 +83,7 @@ class Download(BaseModel):
 
 
 class Epoch(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     read: List[float]
     write: List[float]
@@ -98,7 +98,7 @@ class Download1(BaseModel):
 
 
 class Resource(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     architectures: Optional[List[str]] = None
     created_at: Optional[str] = Field(
@@ -115,7 +115,7 @@ class Resource(BaseModel):
 
 
 class ChannelMapItem(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     architectures: Optional[List[str]] = None
     base: Optional[str] = None
@@ -145,14 +145,14 @@ class ChannelMapItem(BaseModel):
 
 
 class ErrorListItem(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     code: str
     message: str
 
 
 class InfoResponse(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     channel_map: List[ChannelMapItem] = Field(
         ...,

snap_python/schemas/store/refresh.py

@@ -79,7 +79,7 @@ class RefreshResultError(BaseModel):
 
 
 class RefreshResultData(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     default_track: Optional[str] = Field(
         None,

snap_python/schemas/store/search.py

@@ -38,7 +38,7 @@ VALID_SEARCH_CATEGORY_FIELDS = [
 
 
 class ErrorListItem(BaseModel):
-    model_config = ConfigDict(extra="forbid", exclude_unset=True)
+    model_config = ConfigDict(exclude_unset=True)
 
     code: str
     message: str
@@ -90,8 +90,6 @@ class SnapDetails(BaseModel):
 
 
 class SearchResult(BaseModel):
-    model_config = ConfigDict(extra="forbid")
-
     name: str
     revision: Optional[Revision] = None
     snap: StoreSnap
@@ -124,15 +122,11 @@ class SearchResult(BaseModel):
 
 
 class SearchResponse(BaseModel):
-    model_config = ConfigDict(extra="forbid")
-
     error_list: Optional[List[ErrorListItem]] = Field(None, alias="error-list")
     results: List[SearchResult]
 
 
 class ArchSearchItem(BaseModel):
-    model_config = ConfigDict(extra="ignore")
-
     aliases: Optional[list[dict[str, str]]]
     apps: list[str]
     package_name: str

snap_python/schemas/store/track.py

@@ -13,6 +13,13 @@ logger = logging.getLogger("snap_python.schemas.store.track")
 
 
 class TrackRevisionDetails(BaseModel):
+    """Contains details for a specific track/risk/architecture
+
+
+    Raises:
+        ValueError: If the architecture is invalid.
+    """
+
     arch: str
     base: str
     channel: Channel
@@ -44,6 +51,8 @@ class TrackRevisionDetails(BaseModel):
 
 
 class TrackRiskMap(BaseModel):
+    """Map of architectures to their revision details for a specific risk level."""
+
     amd64: Optional[TrackRevisionDetails] = None
     arm64: Optional[TrackRevisionDetails] = None
     armhf: Optional[TrackRevisionDetails] = None
@@ -63,10 +72,7 @@ class TrackRiskMap(BaseModel):
 def channel_map_to_current_track_map(
     channel_map: list[ChannelMapItem],
 ) -> dict[str, dict[str, TrackRiskMap]]:
-    """Convert a list of ChannelMapItem to a map of TrackRiskMap.
-
-    track -> risk -> arch
-    """
+    """Convert a list of ChannelMapItem to a map of track -> risk -> TrackRiskMap."""
     # track -> risk -> arch
     current_track_map: dict[str, dict[str, TrackRiskMap]] = {}
 

{snap_python-0.1.6.dist-info → snap_python-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: snap-python
-Version: 0.1.6
+Version: 0.1.7
 Summary: Interact with snapd and snapstore(s) from Python
 License: MIT
 Author: Alexander Lukens
@@ -12,8 +12,8 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: httpx (>=0.27.2,<0.28.0)
-Requires-Dist: pydantic (>=2.9.2,<3.0.0)
+Requires-Dist: httpx (>0.27,<1)
+Requires-Dist: pydantic (>2.9,<3)
 Requires-Dist: retry (>=0.9.2,<0.10.0)
 Project-URL: Documentation, https://alexdlukens.github.io/snap-python/
 Project-URL: Repository, https://github.com/alexdlukens/snap-python

{snap_python-0.1.6.dist-info → snap_python-0.1.7.dist-info}/RECORD

@@ -3,24 +3,24 @@ snap_python/client.py,sha256=HrgLu0ebhPci8cN3GmfOWc9-ccri80Pk4Oys2RVdoQY,6539
 snap_python/components/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 snap_python/components/config.py,sha256=IO_gCrumphEZioINFmXxb8I5OVsKC9cgSa28sQbZ700,1775
 snap_python/components/snaps.py,sha256=NfZAgoNvsvf-aJa-ZZJoJxS-KrwrG6gBe1ScN2WM5To,18003
-snap_python/components/store.py,sha256=gryTStqw4Wu7S7book4arqzfyzFcITRSCbW5BK89ce4,17839
+snap_python/components/store.py,sha256=tN7LNsSpEjRLZBE1JTrUdQdcuWYZ6h7srZY7xQyZBf8,18735
 snap_python/schemas/changes.py,sha256=7SYUFXAQq9arl7hAA3YQ0YubMurPJtknTOVxP8ryqew,1919
-snap_python/schemas/common.py,sha256=GLtC9aBo3O8vLKX9PNvzE-_DKN02OGpwsS7Ik2WK9-8,2608
+snap_python/schemas/common.py,sha256=-_a9o45piK0jzwwrW96swAuaXvh_jh9WwlxKU_waCEA,2544
 snap_python/schemas/config.py,sha256=xXTGhk6TpwI8RlR6_qnfWlBHrBnTwr3hRCO5qCnN-6U,107
-snap_python/schemas/snaps.py,sha256=GTDi6_n5P7o9h7-cDllkkFYn8wS9EQ6VZ32rwYJD07A,3657
+snap_python/schemas/snaps.py,sha256=nh2LaqSVFJl6Enpo_SiPZllndRcSK-9bk4E4Q_CxI6w,3598
 snap_python/schemas/store/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-snap_python/schemas/store/categories.py,sha256=PewotRqoPs7cMKtlZKsHVNfnJoItmNKMekS1Be-46l0,1082
-snap_python/schemas/store/info.py,sha256=C5T88QMgMJZzG5BwRXGeSDVMlElefjv30i-Gqm8qRyA,4318
+snap_python/schemas/store/categories.py,sha256=6NZyfi23zWpL2jEgoIKLrTzX9AIjq2APdiQEuCaUzEQ,1018
+snap_python/schemas/store/info.py,sha256=6TqxAqFAXf_z1bkz9MjiNKYpbd87eXL20w4rkgJZQnM,4190
 snap_python/schemas/store/json/category_by_name_schema.json,sha256=3-95m6V2WXaPCu_DZw-STa9YHPEbr7CI8fbhjaq881s,1674
 snap_python/schemas/store/json/category_schema.json,sha256=c4h_eyyBbChOiNby3hYfjAh9i_1MDbPJnHHU4rGOGss,1898
 snap_python/schemas/store/json/snaps_find_schema.json,sha256=TfKpnievs5QOc7ah97ATMCQTOEfKspONTZnwSwq_0fI,8866
 snap_python/schemas/store/json/snaps_info_schema.json,sha256=udoOovb6Ea7EnE_R9RuZ1CpY1wOJ3-K8xP4-8zKHgrQ,15888
-snap_python/schemas/store/refresh.py,sha256=HAlt-rKfy7vmZSCujMXtW9HHXlVYbHGFwy_BKfD1uCU,3190
-snap_python/schemas/store/search.py,sha256=4ghZb8xY6IESxZuAvzSjM9RaEwZAH3zjiC06zLc0zvU,4561
-snap_python/schemas/store/track.py,sha256=peOjBVccukjud7_Wn_M2hX6KvoJhyhX74FX5HeeHpm8,4099
+snap_python/schemas/store/refresh.py,sha256=fKYffBNki2-rLUk2oVnjAdQTO6P6kJTfcMtAZ1gouTo,3174
+snap_python/schemas/store/search.py,sha256=pFdWXwscM36M9pRfHcpqpbQyNazAW0IOxy5iv4c_tWI,4404
+snap_python/schemas/store/track.py,sha256=5x8UIXhLxDVCtG8H-JJaaaUe_5nFLOth9TkpZ51w474,4307
 snap_python/scrape.py,sha256=ZDJAROp8jwossaAA78gBFTKQzrfnO11RgyyOZqevNd0,4576
 snap_python/utils.py,sha256=cmUooPg5DzNL197Xxvu7LGD4Gwk3xxsT68wbVytRtV0,1240
-snap_python-0.1.6.dist-info/LICENSE,sha256=lYXNZszxC_l2rlBvrHYs-esDegxWh8MKtjrdxA24uOk,1072
-snap_python-0.1.6.dist-info/METADATA,sha256=cfN_x0M8MSEtVq0Du6RlDQTrt6xzbd144tH6jyr_w34,1883
-snap_python-0.1.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-snap_python-0.1.6.dist-info/RECORD,,
+snap_python-0.1.7.dist-info/LICENSE,sha256=lYXNZszxC_l2rlBvrHYs-esDegxWh8MKtjrdxA24uOk,1072
+snap_python-0.1.7.dist-info/METADATA,sha256=aCfIUhlj5WgeIOQgwKMisa60GUSEPO6OM01JqrBySZ8,1868
+snap_python-0.1.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+snap_python-0.1.7.dist-info/RECORD,,