accept plain dicts in create/update methods

Author: bfontaine

CHANGELOG.md

@@ -1,5 +1,12 @@
 # PyMagento Changelog
 
+## 2.11.1 (2025/12/05)
+
+* `get_custom_attribute` and similar functions now accept the new typed dicts `Category` and `Customer` introduced in
+  the previous release
+* `update_category` and `create_category` now accept both a `Category` or a plain dict
+* `save_product_media` now accepts both a `MediaGalleryEntry` or a plain dict
+
 ## 2.11.0 (2025/12/05)
 
 ### Breaking changes

magento/attributes.py

@@ -7,7 +7,7 @@
 
 T = TypeVar('T')
 
-Item = TypeVar('Item', bound=Union[Product, MagentoEntity])
+Item = TypeVar('Item', bound=Union[Category, Customer, Product, MagentoEntity])
 
 # From Magento
 CATEGORY_ENTITY_TYPE_ID = 3

magento/client.py

@@ -299,7 +299,8 @@ def get_category_by_name(self, name: str, *, assert_one: bool = False, **kwargs:
 
         return None
 
-    def update_category(self, category_id: PathId, category_data: Category, **kwargs: Any) -> Category:
+    def update_category(self, category_id: PathId, category_data: Union[Category, Dict[str, Any]],
+                        **kwargs: Any) -> Category:
         """Update a category.
 
         :param category_id:
@@ -311,7 +312,7 @@ def update_category(self, category_id: PathId, category_data: Category, **kwargs
                                                **kwargs)
         return category
 
-    def create_category(self, category: Category, **kwargs: Any) -> Category:
+    def create_category(self, category: Union[Category, Dict[str, Any]], **kwargs: Any) -> Category:
         """Create a new category and return it."""
         category = self.post_json_api("/V1/categories", json={"category": category}, **kwargs)
         return category
@@ -939,7 +940,7 @@ def get_product_media(self, sku: Sku, media_id: PathId, **kwargs: Any) -> MediaG
         entry: MediaGalleryEntry = self.get_json_api(f"/V1/products/{escape_path(sku)}/media/{media_id}", **kwargs)
         return entry
 
-    def save_product_media(self, sku: Sku, media_entry: MediaGalleryEntry, **kwargs: Any) -> Any:
+    def save_product_media(self, sku: Sku, media_entry: Union[MediaGalleryEntry, Dict[str, Any]], **kwargs: Any) -> Any:
         """Save a product media."""
         return self.post_json_api(f"/V1/products/{escape_path(sku)}/media", json={"entry": media_entry}, **kwargs)