Use constants for statuses

1 files changed, 33 insertions, 0 deletions

diff --git a/payments/__init__.py b/payments/__init__.py
index e5ce8b7..cc73e6d 100644
--- a/payments/__init__.py
+++ b/payments/__init__.py
@@ -6,11 +6,44 @@ except ImportError:
     get_model = apps.get_model
 from django.conf import settings
 from django.core.exceptions import ImproperlyConfigured
+from django.utils.translation import pgettext_lazy
 
 PurchasedItem = namedtuple('PurchasedItem',
                            'name, quantity, price, currency, sku')
 
 
+class PaymentStatus:
+    WAITING = 'waiting'
+    PREAUTH = 'preauth'
+    CONFIRMED = 'confirmed'
+    REJECTED = 'rejected'
+    REFUNDED = 'refunded'
+    ERROR = 'error'
+    INPUT = 'input'
+
+    CHOICES = [
+        (WAITING, pgettext_lazy('payment status', 'Waiting for confirmation')),
+        (PREAUTH, pgettext_lazy('payment status', 'Pre-authorized')),
+        (CONFIRMED, pgettext_lazy('payment status', 'Confirmed')),
+        (REJECTED, pgettext_lazy('payment status', 'Rejected')),
+        (REFUNDED, pgettext_lazy('payment status', 'Refunded')),
+        (ERROR, pgettext_lazy('payment status', 'Error')),
+        (INPUT, pgettext_lazy('payment status', 'Input'))]
+
+
+class FraudStatus:
+    UNKNOWN = 'unknown'
+    ACCEPT = 'accept'
+    REJECT = 'reject'
+    REVIEW = 'review'
+
+    CHOICES = [
+        (UNKNOWN, pgettext_lazy('fraud status', 'Unknown')),
+        (ACCEPT, pgettext_lazy('fraud status', 'Passed')),
+        (REJECT, pgettext_lazy('fraud status', 'Rejected')),
+        (REVIEW, pgettext_lazy('fraud status', 'Review'))]
+
+
 class RedirectNeeded(Exception):
     pass