PTHMINT-45 Fix ruff error code ANN201

Author: zulquer

pyproject.toml

@@ -98,7 +98,6 @@ extend-safe-fixes = [
     "D415", # docstrings should end with a period, question mark, or exclamation point
 ]
 ignore = [
-    "ANN201",
     "ANN204",
     "ANN401",
     "ARG002",

src/multisafepay/api/base/listings/listing.py

@@ -101,7 +101,7 @@ def get_data(self: "Listing") -> List[T]:
         """
         return self.data
 
-    def append(self: "Listing", item: T):
+    def append(self: "Listing", item: T) -> None:
         """
         Append an item to the listing.
 

src/multisafepay/api/base/response/custom_api_response.py

@@ -41,7 +41,7 @@ def __init__(
         for key, value in kwargs.items():
             setattr(self, key, value)
 
-    def get_data(self: "CustomApiResponse"):
+    def get_data(self: "CustomApiResponse") -> Optional[Any]:
         """
         Get the data contained in the response.
 

src/multisafepay/api/paths/capture/request/capture_request.py

@@ -25,7 +25,7 @@ class CaptureRequest(RequestModel):
     status: Optional[str]
     reason: Optional[str]
 
-    def add_status(self: "CaptureRequest", status: str):
+    def add_status(self: "CaptureRequest", status: str) -> "CaptureRequest":
         """
         Add a status to the capture request.
 
@@ -41,7 +41,7 @@ def add_status(self: "CaptureRequest", status: str):
         self.status = status
         return self
 
-    def add_reason(self: "CaptureRequest", reason: str):
+    def add_reason(self: "CaptureRequest", reason: str) -> "CaptureRequest":
         """
         Add a reason to the capture request.
 

src/multisafepay/api/paths/orders/order_id/refund/request/components/checkout_data.py

@@ -26,7 +26,10 @@ class CheckoutData(RequestModel):
 
     items: Optional[List[CartItem]]
 
-    def add_items(self: "CheckoutData", items: List[CartItem] = ()):
+    def add_items(
+        self: "CheckoutData",
+        items: List[CartItem] = (),
+    ) -> "CheckoutData":
         """
         Adds multiple items to the checkout data.
 
@@ -45,7 +48,7 @@ def add_items(self: "CheckoutData", items: List[CartItem] = ()):
             self.add_item(item)
         return self
 
-    def add_item(self: "CheckoutData", item: CartItem):
+    def add_item(self: "CheckoutData", item: CartItem) -> "CheckoutData":
         """
         Adds a single item to the checkout data.
 
@@ -63,7 +66,7 @@ def add_item(self: "CheckoutData", item: CartItem):
         self.items.append(item)
         return self
 
-    def get_items(self: "CheckoutData"):
+    def get_items(self: "CheckoutData") -> List[CartItem]:
         """
         Retrieves all items from the checkout data.
 
@@ -74,7 +77,7 @@ def get_items(self: "CheckoutData"):
         """
         return self.items
 
-    def get_item(self: "CheckoutData", index: int):
+    def get_item(self: "CheckoutData", index: int) -> CartItem:
         """
         Retrieves an item by its index from the checkout data.
 
@@ -93,7 +96,7 @@ def generate_from_shopping_cart(
         self: "CheckoutData",
         shopping_cart: ShoppingCart,
         tax_table_selector: str = "",
-    ):
+    ) -> None:
         """
         Generates checkout data from a shopping cart.
 
@@ -114,7 +117,7 @@ def refund_by_merchant_item_id(
         self: "CheckoutData",
         merchant_item_id: str,
         quantity: int = 0,
-    ):
+    ) -> None:
         """
         Processes a refund by merchant item ID.
 
@@ -146,7 +149,7 @@ def refund_by_merchant_item_id(
     def get_item_by_merchant_item_id(
         self: "CheckoutData",
         merchant_item_id: str,
-    ):
+    ) -> CartItem:
         """
         Retrieves an item by its merchant item ID.
 

src/multisafepay/api/shared/cart/shopping_cart.py

@@ -34,7 +34,10 @@ def get_items(self: "ShoppingCart") -> List[CartItem]:
         """
         return self.items
 
-    def add_items(self: "ShoppingCart", items: List[CartItem]):
+    def add_items(
+        self: "ShoppingCart",
+        items: List[CartItem],
+    ) -> "ShoppingCart":
         """
         Add multiple items to the shopping cart.
 
@@ -50,7 +53,7 @@ def add_items(self: "ShoppingCart", items: List[CartItem]):
         self.items = items
         return self
 
-    def add_item(self: "ShoppingCart", item: CartItem):
+    def add_item(self: "ShoppingCart", item: CartItem) -> "ShoppingCart":
         """
         Add a single item to the shopping cart.
 

src/multisafepay/client/api_key.py

@@ -25,7 +25,7 @@ class ApiKey(BaseModel):
     api_key: str
 
     @validator("api_key")
-    def validate_api_key(cls: "ApiKey", api_key: str):
+    def validate_api_key(cls: "ApiKey", api_key: str) -> str:
         """
         Validate the API key.
 

src/multisafepay/exception/api.py

@@ -7,6 +7,7 @@
 
 
 import json
+from typing import Any
 
 
 class ApiException(Exception):
@@ -110,7 +111,7 @@ def get_context_as_array(self: "ApiException") -> list:
             lines.append(f"{context_name}: {debug_value}")
         return lines
 
-    def get_context_value(self: "ApiException", name: str):
+    def get_context_value(self: "ApiException", name: str) -> Any:
         """
         Get a specific context value by name.
 

src/multisafepay/util/dict_utils.py

@@ -8,7 +8,7 @@
 from typing import Any, Optional
 
 
-def merge_recursive(dict1: dict, dict2: dict):
+def merge_recursive(dict1: dict, dict2: dict) -> dict:
     """
     Recursively merge two dictionaries.
 

src/multisafepay/util/version.py

@@ -34,7 +34,7 @@ def get_plugin_version(self: "Version") -> str:
         """
         return self.plugin_version
 
-    def set_plugin_version(self: "Version", version: Optional[str]):
+    def set_plugin_version(self: "Version", version: Optional[str]) -> None:
         """
         Set the plugin version.