Merge pull request #34 from elonzh/feat/make-enum-serializable

izanger · web-flow · commit 5924dbe46f5c · 2023-09-25T16:35:09.000-07:00
feat: Inherit enum classes with primiary types
diff --git a/appstoreserverlibrary/api_client.py b/appstoreserverlibrary/api_client.py
@@ -2,7 +2,7 @@
 
 import calendar
 import datetime
-from enum import Enum
+from enum import IntEnum
 from typing import Dict, List, Optional, Type, TypeVar, Union
 from attr import define
 import cattrs
@@ -33,7 +33,7 @@
 
 T = TypeVar('T')
 
-class APIError(Enum):
+class APIError(IntEnum):
     GENERAL_BAD_REQUEST = 4000000
     INVALID_APP_IDENTIFIER = 4000002
     INVALID_REQUEST_REVISION = 4000005
diff --git a/appstoreserverlibrary/models/AccountTenure.py b/appstoreserverlibrary/models/AccountTenure.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class AccountTenure(Enum): 
+class AccountTenure(IntEnum): 
     """
     The age of the customer's account.
     
diff --git a/appstoreserverlibrary/models/AutoRenewStatus.py b/appstoreserverlibrary/models/AutoRenewStatus.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class AutoRenewStatus(Enum): 
+class AutoRenewStatus(IntEnum): 
     """
     The renewal status for an auto-renewable subscription.
     
diff --git a/appstoreserverlibrary/models/ConsumptionStatus.py b/appstoreserverlibrary/models/ConsumptionStatus.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class ConsumptionStatus(Enum): 
+class ConsumptionStatus(IntEnum): 
     """
     A value that indicates the extent to which the customer consumed the in-app purchase.
     
diff --git a/appstoreserverlibrary/models/DeliveryStatus.py b/appstoreserverlibrary/models/DeliveryStatus.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class DeliveryStatus(Enum): 
+class DeliveryStatus(IntEnum): 
     """
     A value that indicates whether the app successfully delivered an in-app purchase that works properly.
     
diff --git a/appstoreserverlibrary/models/Environment.py b/appstoreserverlibrary/models/Environment.py
@@ -2,7 +2,7 @@
 
 from enum import Enum
 
-class Environment(Enum): 
+class Environment(str, Enum): 
     """
     The server environment, either sandbox or production.
     
diff --git a/appstoreserverlibrary/models/ExpirationIntent.py b/appstoreserverlibrary/models/ExpirationIntent.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class ExpirationIntent(Enum): 
+class ExpirationIntent(IntEnum): 
     """
     The reason an auto-renewable subscription expired.
     
diff --git a/appstoreserverlibrary/models/ExtendReasonCode.py b/appstoreserverlibrary/models/ExtendReasonCode.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class ExtendReasonCode(Enum): 
+class ExtendReasonCode(IntEnum): 
     """
     The code that represents the reason for the subscription-renewal-date extension.
     
diff --git a/appstoreserverlibrary/models/FirstSendAttemptResult.py b/appstoreserverlibrary/models/FirstSendAttemptResult.py
@@ -3,7 +3,7 @@
 from enum import Enum, unique
 
 @unique
-class FirstSendAttemptResult(Enum): 
+class FirstSendAttemptResult(str, Enum): 
     """
     An error or result that the App Store server receives when attempting to send an App Store server notification to your server.
     
diff --git a/appstoreserverlibrary/models/InAppOwnershipType.py b/appstoreserverlibrary/models/InAppOwnershipType.py
@@ -2,7 +2,7 @@
 
 from enum import Enum
 
-class InAppOwnershipType(Enum): 
+class InAppOwnershipType(str, Enum): 
     """
     The relationship of the user with the family-shared purchase to which they have access.
     
diff --git a/appstoreserverlibrary/models/LifetimeDollarsPurchased.py b/appstoreserverlibrary/models/LifetimeDollarsPurchased.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class LifetimeDollarsPurchased(Enum): 
+class LifetimeDollarsPurchased(IntEnum): 
     """
     A value that indicates the total amount, in USD, of in-app purchases the customer has made in your app, across all platforms.
     
diff --git a/appstoreserverlibrary/models/LifetimeDollarsRefunded.py b/appstoreserverlibrary/models/LifetimeDollarsRefunded.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class LifetimeDollarsRefunded(Enum): 
+class LifetimeDollarsRefunded(IntEnum): 
     """
     A value that indicates the dollar amount of refunds the customer has received in your app, since purchasing the app, across all platforms.
     
diff --git a/appstoreserverlibrary/models/NotificationTypeV2.py b/appstoreserverlibrary/models/NotificationTypeV2.py
@@ -2,7 +2,7 @@
 
 from enum import Enum
 
-class NotificationTypeV2(Enum): 
+class NotificationTypeV2(str, Enum): 
     """
     A notification type value that App Store Server Notifications V2 uses.
     
diff --git a/appstoreserverlibrary/models/OfferType.py b/appstoreserverlibrary/models/OfferType.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class OfferType(Enum): 
+class OfferType(IntEnum): 
     """
     The type of subscription offer.
     
diff --git a/appstoreserverlibrary/models/OrderLookupStatus.py b/appstoreserverlibrary/models/OrderLookupStatus.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class OrderLookupStatus(Enum): 
+class OrderLookupStatus(IntEnum): 
     """
     A value that indicates whether the order ID in the request is valid for your app.
     
diff --git a/appstoreserverlibrary/models/Platform.py b/appstoreserverlibrary/models/Platform.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class Platform(Enum): 
+class Platform(IntEnum): 
     """
     The platform on which the customer consumed the in-app purchase.
     
diff --git a/appstoreserverlibrary/models/PlayTime.py b/appstoreserverlibrary/models/PlayTime.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class PlayTime(Enum): 
+class PlayTime(IntEnum): 
     """
     A value that indicates the amount of time that the customer used the app.
     
diff --git a/appstoreserverlibrary/models/PriceIncreaseStatus.py b/appstoreserverlibrary/models/PriceIncreaseStatus.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class PriceIncreaseStatus(Enum): 
+class PriceIncreaseStatus(IntEnum): 
     """
     The status that indicates whether an auto-renewable subscription is subject to a price increase.
     
diff --git a/appstoreserverlibrary/models/RevocationReason.py b/appstoreserverlibrary/models/RevocationReason.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class RevocationReason(Enum): 
+class RevocationReason(IntEnum): 
     """
     The reason for a refunded transaction.
     
diff --git a/appstoreserverlibrary/models/SendAttemptResult.py b/appstoreserverlibrary/models/SendAttemptResult.py
@@ -2,7 +2,7 @@
 
 from enum import Enum
 
-class SendAttemptResult(Enum): 
+class SendAttemptResult(str, Enum): 
     """
     The success or error information the App Store server records when it attempts to send an App Store server notification to your server.
     
diff --git a/appstoreserverlibrary/models/Status.py b/appstoreserverlibrary/models/Status.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class Status(Enum): 
+class Status(IntEnum): 
     """
     The status of an auto-renewable subscription.
     
diff --git a/appstoreserverlibrary/models/Subtype.py b/appstoreserverlibrary/models/Subtype.py
@@ -2,7 +2,7 @@
 
 from enum import Enum
 
-class Subtype(Enum): 
+class Subtype(str, Enum): 
     """
     A notification subtype value that App Store Server Notifications 2 uses.
     
diff --git a/appstoreserverlibrary/models/TransactionHistoryRequest.py b/appstoreserverlibrary/models/TransactionHistoryRequest.py
@@ -7,13 +7,13 @@
 
 from .InAppOwnershipType import InAppOwnershipType
 
-class ProductType(Enum):
+class ProductType(str, Enum):
     AUTO_RENEWABLE = "AUTO_RENEWABLE"
     NON_RENEWABLE = "NON_RENEWABLE"
     CONSUMABLE = "CONSUMABLE"
     NON_CONSUMABLE = "NON_CONSUMABLE"
 
-class Order(Enum):
+class Order(str, Enum):
     ASCENDING = "ASCENDING"
     DESCENDING = "DESCENDING"
 
diff --git a/appstoreserverlibrary/models/TransactionReason.py b/appstoreserverlibrary/models/TransactionReason.py
@@ -2,7 +2,7 @@
 
 from enum import Enum
 
-class TransactionReason(Enum): 
+class TransactionReason(str, Enum): 
     """
     The cause of a purchase transaction, which indicates whether it’s a customer’s purchase or a renewal for an auto-renewable subscription that the system initiates.
     
diff --git a/appstoreserverlibrary/models/Type.py b/appstoreserverlibrary/models/Type.py
@@ -2,7 +2,7 @@
 
 from enum import Enum
 
-class Type(Enum): 
+class Type(str, Enum): 
     """
     The type of in-app purchase products you can offer in your app.
     
diff --git a/appstoreserverlibrary/models/UserStatus.py b/appstoreserverlibrary/models/UserStatus.py
@@ -1,8 +1,8 @@
 # Copyright (c) 2023 Apple Inc. Licensed under MIT License.
 
-from enum import Enum
+from enum import IntEnum
 
-class UserStatus(Enum): 
+class UserStatus(IntEnum): 
     """
     The status of a customer's account within your app.
     
diff --git a/appstoreserverlibrary/signed_data_verifier.py b/appstoreserverlibrary/signed_data_verifier.py
@@ -2,7 +2,7 @@
 
 from typing import List
 from base64 import b64decode
-from enum import Enum
+from enum import IntEnum
 import time
 import datetime
 
@@ -274,7 +274,7 @@ def check_ocsp_status(self, cert: crypto.X509, issuer: crypto.X509, root: crypto
         raise VerificationException(VerificationStatus.VERIFICATION_FAILURE)
 
 
-class VerificationStatus(Enum):
+class VerificationStatus(IntEnum):
     OK = 0
     VERIFICATION_FAILURE = 1
     INVALID_APP_IDENTIFIER = 2
