airalo-sdk 1.0.1__py3-none-any.whl

airalo/services/topup_service.py

@@ -0,0 +1,127 @@
+"""
+Topup Service Module
+
+This module handles top-up related API operations including creating top-ups.
+"""
+
+import json
+from typing import Any, Dict, Optional
+
+from ..config import Config
+from ..constants.api_constants import ApiConstants
+from ..exceptions.airalo_exception import AiraloException, ValidationError, APIError
+from ..helpers.signature import Signature
+from ..resources.http_resource import HttpResource
+
+
+class TopupService:
+    """
+    Service for managing top-up operations.
+    """
+
+    def __init__(
+        self,
+        config: Config,
+        http_resource: HttpResource,
+        signature: Signature,
+        access_token: str,
+    ):
+        """
+        Initialize top-up service.
+
+        Args:
+            config: SDK configuration
+            http_resource: HTTP client for requests
+            signature: Signature generator for request signing
+            access_token: API access token
+
+        Raises:
+            AiraloException: If access token is invalid
+        """
+        if not access_token:
+            raise AiraloException("Invalid access token, please check your credentials")
+
+        self._config = config
+        self._http = http_resource
+        self._signature = signature
+        self._access_token = access_token
+        self._base_url = config.get_url()
+
+    def create_topup(self, payload: Dict[str, Any]) -> Optional[Dict[str, Any]]:
+        """
+        Create a new top-up.
+
+        Args:
+            payload: Top-up data including:
+                - package_id: Package ID to top-up (required)
+                - iccid: ICCID of the SIM to top-up (required)
+
+        Returns:
+            Top-up response data or None
+
+        Raises:
+            ValidationError: If payload is invalid
+            APIError: If API request fails
+        """
+        self._validate_topup(payload)
+
+        # Set headers with signature
+        headers = self._get_headers(payload)
+        self._http.set_headers(headers)
+
+        # Make request
+        url = self._base_url + ApiConstants.TOPUPS_SLUG
+        response = self._http.post(url, payload)
+
+        if self._http.code != 200:
+            raise APIError(
+                f"Topup creation failed, status code: {self._http.code}, response: {response}"
+            )
+
+        try:
+            return json.loads(response)
+        except json.JSONDecodeError:
+            raise APIError("Failed to parse top-up response")
+
+    def _get_headers(self, payload: Dict[str, Any]) -> Dict[str, str]:
+        """
+        Get headers for top-up request with signature.
+
+        Args:
+            payload: Request payload
+
+        Returns:
+            Headers dictionary
+        """
+        return {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self._access_token}",
+            "airalo-signature": self._signature.get_signature(payload),
+        }
+
+    def _validate_topup(self, payload: Dict[str, Any]) -> None:
+        """
+        Validate top-up payload.
+
+        Args:
+            payload: Top-up data
+
+        Raises:
+            ValidationError: If validation fails
+        """
+        if not payload.get("package_id"):
+            raise ValidationError(
+                f"The package_id is required, payload: {json.dumps(payload)}"
+            )
+
+        iccid = payload.get("iccid")
+
+        if not iccid:
+            raise ValidationError(
+                f"The iccid is required, payload: {json.dumps(payload)}"
+            )
+
+        if len(iccid) < 16 or len(iccid) > 21:
+            raise ValidationError(
+                f"The iccid must be between 16 and 21 characters, received: {iccid}"
+            )

airalo/services/voucher_service.py

@@ -0,0 +1,138 @@
+import json
+
+from airalo.config import Config
+from airalo.constants.api_constants import ApiConstants
+from airalo.constants.sdk_constants import SdkConstants
+from airalo.exceptions.airalo_exception import AiraloException
+from airalo.helpers.signature import Signature
+from airalo.resources.http_resource import HttpResource
+
+
+class VoucherService:
+    def __init__(
+        self,
+        config: Config,
+        curl: HttpResource,
+        signature: Signature,
+        access_token: str,
+    ):
+        if not access_token:
+            raise AiraloException("Invalid access token please check your credentials")
+
+        self.config = config
+        self.curl = curl
+        self.signature = signature
+        self.access_token = access_token
+
+    def create_voucher(self, payload: dict) -> dict | None:
+        self._validate_voucher(payload)
+
+        headers = self._get_headers(payload)
+        response = self.curl.set_headers(headers).post(
+            self.config.get_url() + ApiConstants.VOUCHERS_SLUG, payload
+        )
+
+        if self.curl.code != 200:
+            raise AiraloException(
+                f"Voucher creation failed, status code: {self.curl.code}, response: {response}"
+            )
+
+        return json.loads(response)
+
+    def create_esim_voucher(self, payload: dict) -> dict | None:
+        self._validate_esim_voucher(payload)
+
+        headers = self._get_headers(payload)
+        response = self.curl.set_headers(headers).post(
+            self.config.get_url() + ApiConstants.VOUCHERS_ESIM_SLUG, payload
+        )
+
+        if self.curl.code != 200:
+            raise AiraloException(
+                f"Voucher creation failed, status code: {self.curl.code}, response: {response}"
+            )
+
+        return json.loads(response)
+
+    def _get_headers(self, payload: dict) -> list[str]:
+        return [
+            "Content-Type: application/json",
+            f"Authorization: Bearer {self.access_token}",
+            f"airalo-signature: {self.signature.get_signature(payload)}",
+        ]
+
+    def _validate_voucher(self, payload: dict) -> None:
+        if "amount" not in payload or payload["amount"] == "" or payload["amount"] < 1:
+            raise AiraloException(
+                f"The amount is required, payload: {json.dumps(payload)}"
+            )
+
+        if payload["amount"] > SdkConstants.VOUCHER_MAX_NUM:
+            raise AiraloException(
+                f"The amount may not be greater than {SdkConstants.VOUCHER_MAX_NUM}"
+            )
+
+        if (
+            "voucher_code" in payload
+            and isinstance(payload["voucher_code"], str)
+            and len(payload["voucher_code"]) > 255
+        ):
+            raise AiraloException("The voucher code may not exceed 255 characters.")
+
+        if (
+            "voucher_code" in payload
+            and "quantity" in payload
+            and payload["quantity"] > 1
+        ):
+            raise AiraloException(
+                "The selected voucher code allows a maximum quantity of 1"
+            )
+
+        if "usage_limit" in payload and (
+            payload["usage_limit"] < 1
+            or payload["usage_limit"] > SdkConstants.VOUCHER_MAX_NUM
+        ):
+            raise AiraloException(
+                f"The usage_limit may not be greater than {SdkConstants.VOUCHER_MAX_NUM}"
+            )
+
+        if (
+            "quantity" not in payload
+            or payload["quantity"] == ""
+            or payload["quantity"] < 1
+        ):
+            raise AiraloException(
+                f"The quantity is required, payload: {json.dumps(payload)}"
+            )
+
+        if payload["quantity"] > SdkConstants.VOUCHER_MAX_QUANTITY:
+            raise AiraloException(
+                f"The quantity may not be greater than {SdkConstants.VOUCHER_MAX_QUANTITY}"
+            )
+
+    def _validate_esim_voucher(self, payload: dict) -> None:
+        if not payload.get("vouchers"):
+            raise AiraloException(
+                f"vouchers field is required, payload: {json.dumps(payload)}"
+            )
+
+        if not isinstance(payload["vouchers"], list):
+            raise AiraloException(
+                f"vouchers field should be an array, payload: {json.dumps(payload)}"
+            )
+
+        for voucher in payload["vouchers"]:
+            if not voucher.get("package_id"):
+                raise AiraloException(
+                    f"The vouchers.package_id is required, payload: {json.dumps(payload)}"
+                )
+
+            if not voucher.get("quantity"):
+                raise AiraloException(
+                    f"The vouchers.quantity is required and should be greater than 0, payload: {json.dumps(payload)}"
+                )
+
+            if payload.get("quantity", 0) > SdkConstants.VOUCHER_MAX_QUANTITY:
+                raise AiraloException(
+                    f"The vouchers.quantity may not be greater than {SdkConstants.VOUCHER_MAX_QUANTITY}"
+                )