1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
|
from __future__ import unicode_literals
import json
from uuid import uuid4
from django.conf import settings
from django.urls import reverse
from django.db import models
from django.utils.translation import ugettext_lazy as _
from .core import provider_factory
from . import FraudStatus, PaymentStatus
class PaymentAttributeProxy(object):
def __init__(self, payment):
self._payment = payment
super(PaymentAttributeProxy, self).__init__()
def __getattr__(self, item):
data = json.loads(self._payment.extra_data or '{}')
return data[item]
def __setattr__(self, key, value):
if key == '_payment':
return super(PaymentAttributeProxy, self).__setattr__(key, value)
try:
data = json.loads(self._payment.extra_data)
except ValueError:
data = {}
data[key] = value
self._payment.extra_data = json.dumps(data)
class BasePayment(models.Model):
'''
Represents a single transaction. Each instance has one or more PaymentItem.
'''
variant = models.CharField(max_length=255)
#: Transaction status
status = models.CharField(
max_length=10, choices=PaymentStatus.CHOICES,
default=PaymentStatus.WAITING)
fraud_status = models.CharField(
_('fraud check'), max_length=10, choices=FraudStatus.CHOICES,
default=FraudStatus.UNKNOWN)
fraud_message = models.TextField(blank=True, default='')
#: Creation date and time
created = models.DateTimeField(auto_now_add=True)
#: Date and time of last modification
modified = models.DateTimeField(auto_now=True)
#: Transaction ID (if applicable)
transaction_id = models.CharField(max_length=255, blank=True)
#: Currency code (may be provider-specific)
currency = models.CharField(max_length=10)
#: Total amount (gross)
total = models.DecimalField(max_digits=9, decimal_places=2, default='0.0')
delivery = models.DecimalField(
max_digits=9, decimal_places=2, default='0.0')
tax = models.DecimalField(max_digits=9, decimal_places=2, default='0.0')
description = models.TextField(blank=True, default='')
billing_first_name = models.CharField(max_length=256, blank=True)
billing_last_name = models.CharField(max_length=256, blank=True)
billing_address_1 = models.CharField(max_length=256, blank=True)
billing_address_2 = models.CharField(max_length=256, blank=True)
billing_city = models.CharField(max_length=256, blank=True)
billing_postcode = models.CharField(max_length=256, blank=True)
billing_country_code = models.CharField(max_length=2, blank=True)
billing_country_area = models.CharField(max_length=256, blank=True)
billing_email = models.EmailField(blank=True)
customer_ip_address = models.GenericIPAddressField(blank=True, null=True)
extra_data = models.TextField(blank=True, default='')
message = models.TextField(blank=True, default='')
token = models.CharField(max_length=36, blank=True, default='')
captured_amount = models.DecimalField(
max_digits=9, decimal_places=2, default='0.0')
class Meta:
abstract = True
def change_status(self, status, message=''):
'''
Updates the Payment status and sends the status_changed signal.
'''
from .signals import status_changed
self.status = status
self.message = message
self.save()
status_changed.send(sender=type(self), instance=self)
def change_fraud_status(self, status, message='', commit=True):
available_statuses = [choice[0] for choice in FraudStatus.CHOICES]
if status not in available_statuses:
raise ValueError(
'Wrong status "%s", it should be one of: %s' % (
status, ', '.join(available_statuses)))
self.fraud_status = status
self.fraud_message = message
if commit:
self.save()
def save(self, **kwargs):
if not self.token:
tries = {} # Stores a set of tried values
while True:
token = str(uuid4())
if token in tries and len(tries) >= 100: # After 100 tries we are impliying an infinite loop
raise SystemExit('A possible infinite loop was detected')
else:
if not self.__class__._default_manager.filter(token=token).exists():
self.token = token
break
tries.add(token)
return super(BasePayment, self).save(**kwargs)
def __unicode__(self):
return self.variant
def get_form(self, data=None):
provider = provider_factory(self.variant)
return provider.get_form(self, data=data)
def get_purchased_items(self):
return []
def get_failure_url(self):
raise NotImplementedError()
def get_success_url(self):
raise NotImplementedError()
def get_process_url(self):
return reverse('process_payment', kwargs={'token': self.token})
def capture(self, amount=None):
if self.status != PaymentStatus.PREAUTH:
raise ValueError(
'Only pre-authorized payments can be captured.')
provider = provider_factory(self.variant)
amount = provider.capture(self, amount)
if amount:
self.captured_amount = amount
self.change_status(PaymentStatus.CONFIRMED)
def release(self):
if self.status != PaymentStatus.PREAUTH:
raise ValueError(
'Only pre-authorized payments can be released.')
provider = provider_factory(self.variant)
provider.release(self)
self.change_status(PaymentStatus.REFUNDED)
def refund(self, amount=None):
if self.status != PaymentStatus.CONFIRMED:
raise ValueError(
'Only charged payments can be refunded.')
if amount:
if amount > self.captured_amount:
raise ValueError(
'Refund amount can not be greater then captured amount')
provider = provider_factory(self.variant)
amount = provider.refund(self, amount)
self.captured_amount -= amount
if self.captured_amount == 0 and self.status != PaymentStatus.REFUNDED:
self.change_status(PaymentStatus.REFUNDED)
self.save()
@property
def attrs(self):
return PaymentAttributeProxy(self)
|
