1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 from urllib.parse import urlencode
6 from urllib.request import HTTPError
7 from django.contrib.sites.models import Site
8 from django.db import models
9 from django.urls import reverse
10 from django.utils.timezone import now
11 from django.utils.translation import ugettext_lazy as _
15 class CardToken(models.Model):
16 """ This should be attached to a payment schedule. """
17 pos_id = models.CharField(_('POS id'), max_length=255)
18 disposable_token = models.CharField(_('disposable token'), max_length=255)
19 reusable_token = models.CharField(_('reusable token'), max_length=255, null=True, blank=True)
20 created_at = models.DateTimeField(_('created_at'), auto_now_add=True)
24 verbose_name = _('PayU card token')
25 verbose_name_plural = _('PayU card tokens')
28 class Order(models.Model):
29 pos_id = models.CharField(_('POS id'), max_length=255) # TODO: redundant?
30 customer_ip = models.GenericIPAddressField(_('customer IP'))
31 order_id = models.CharField(_('order ID'), max_length=255, blank=True)
33 status = models.CharField(max_length=128, blank=True, choices=[
34 ('PENDING', _('Pending')),
35 ('WAITING_FOR_CONFIRMATION', _('Waiting for confirmation')),
36 ('COMPLETED', _('Completed')),
37 ('CANCELED', _('Canceled')),
38 ('REJECTED', _('Rejected')),
40 created_at = models.DateTimeField(null=True, blank=True, auto_now_add=True)
41 completed_at = models.DateTimeField(null=True, blank=True)
45 verbose_name = _('PayU order')
46 verbose_name_plural = _('PayU orders')
48 # These need to be provided in a subclass.
51 raise NotImplementedError()
54 raise NotImplementedError()
56 def get_continue_url(self):
57 raise NotImplementedError()
59 def get_description(self):
60 raise NotImplementedError()
62 def get_products(self):
63 """ At least: name, unitPrice, quantity. """
66 'name': self.get_description(),
67 'unitPrice': str(int(self.get_amount() * 100)),
72 def is_recurring(self):
75 def get_notify_url(self):
76 raise NotImplementedError
78 def status_updated(self):
84 return POSS[self.pos_id]
86 def get_representation(self, token=None):
88 "notifyUrl": self.get_notify_url(),
89 "customerIp": self.customer_ip,
90 "merchantPosId": self.pos_id,
91 "currencyCode": self.get_pos().currency_code,
92 "totalAmount": str(int(self.get_amount() * 100)),
93 "extOrderId": "wolne-lektury-%d" % self.pk,
95 "buyer": self.get_buyer() or {},
96 "continueUrl": self.get_continue_url(),
97 "description": self.get_description(),
98 "products": self.get_products(),
101 token = self.get_card_token()
102 rep['recurring'] = 'STANDARD' if token.reusable_token else 'FIRST'
103 rep['payMethods'] = {
105 "type": "CARD_TOKEN",
106 "value": token.reusable_token or token.disposable_token
112 token = self.get_card_token() if self.is_recurring() else None
113 representation = self.get_representation(token)
115 response = self.get_pos().request('POST', 'orders', representation)
116 except HTTPError as e:
117 resp = json.loads(e.read().decode('utf-8'))
118 if resp['status']['statusCode'] == 'ERROR_ORDER_NOT_UNIQUE':
124 reusable_token = response.get('payMethods', {}).get('payMethod', {}).get('value', None)
126 token.reusable_token = reusable_token
130 if 'orderId' not in response:
131 raise ValueError("Expecting dict with `orderId` key, got: %s" % response)
132 self.order_id = response['orderId']
136 return response.get('redirectUri', self.schedule.get_thanks_url())
139 class Notification(models.Model):
140 """ Add `order` FK to real Order model. """
141 body = models.TextField(_('body'))
142 received_at = models.DateTimeField(_('received_at'), auto_now_add=True)
146 verbose_name = _('PayU notification')
147 verbose_name_plural = _('PayU notifications')
149 def get_status(self):
150 return json.loads(self.body)['order']['status']
153 status = self.get_status()
154 if self.order.status not in (status, 'COMPLETED'):
155 self.order.status = status
156 if status == 'COMPLETED':
157 self.order.completed_at = now()
159 self.order.status_updated()