Upgrades
[wolnelektury.git] / src / club / payu / models.py
1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
3 #
4 import json
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 gettext_lazy as _
12 from . import POSS
13
14
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)
21
22     class Meta:
23         abstract = True
24         verbose_name = _('PayU card token')
25         verbose_name_plural = _('PayU card tokens')
26
27
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)
32
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')),
39
40         ('ERR-INVALID_TOKEN', _('Invalid token')),
41     ])
42     created_at = models.DateTimeField(null=True, blank=True, auto_now_add=True)
43     completed_at = models.DateTimeField(null=True, blank=True)
44
45     class Meta:
46         abstract = True
47         verbose_name = _('PayU order')
48         verbose_name_plural = _('PayU orders')
49
50     # These need to be provided in a subclass.
51
52     def get_amount(self):
53         raise NotImplementedError()
54
55     def get_buyer(self):
56         raise NotImplementedError()
57
58     def get_continue_url(self):
59         raise NotImplementedError()
60     
61     def get_description(self):
62         raise NotImplementedError()
63
64     def get_products(self):
65         """ At least: name, unitPrice, quantity. """
66         return [
67             {
68                 'name': self.get_description(),
69                 'unitPrice': str(int(self.get_amount() * 100)),
70                 'quantity': '1',
71             },
72         ]
73
74     def is_recurring(self):
75         return False
76
77     def get_notify_url(self):
78         raise NotImplementedError
79
80     def get_thanks_url(self):
81         raise NotImplementedError
82
83     def status_updated(self):
84         pass
85
86     # 
87     
88     def get_pos(self):
89         return POSS[self.pos_id]
90
91     def get_continue_url(self):
92         return "https://{}{}".format(
93             Site.objects.get_current().domain,
94             self.get_thanks_url())
95
96     def get_representation(self, token=None):
97         rep = {
98             "notifyUrl": self.get_notify_url(),
99             "customerIp": self.customer_ip,
100             "merchantPosId": self.pos_id,
101             "currencyCode": self.get_pos().currency_code,
102             "totalAmount": str(int(self.get_amount() * 100)),
103             "extOrderId": "wolne-lektury-%d" % self.pk,
104
105             "buyer": self.get_buyer() or {},
106             "continueUrl": self.get_continue_url(),
107             "description": self.get_description(),
108             "products": self.get_products(),
109         }
110         if token:
111             token = self.get_card_token()
112             rep['recurring'] = 'STANDARD' if token.reusable_token else 'FIRST'
113             rep['payMethods'] = {
114                 "payMethod": {
115                     "type": "CARD_TOKEN",
116                     "value": token.reusable_token or token.disposable_token
117                 }
118             }
119         return rep
120
121     def put(self):
122         token = self.get_card_token() if self.is_recurring() else None
123         representation = self.get_representation(token)
124         try:
125             response = self.get_pos().request('POST', 'orders', representation)
126         except HTTPError as e:
127             resp = json.loads(e.read().decode('utf-8'))
128             if resp['status']['statusCode'] == 'ERROR_ORDER_NOT_UNIQUE':
129                 pass
130             else:
131                 raise
132
133         if token:
134             reusable_token = response.get('payMethods', {}).get('payMethod', {}).get('value', None)
135             if reusable_token:
136                 token.reusable_token = reusable_token
137                 token.save()
138             # else?
139
140         if 'orderId' not in response:
141             code = response.get('status', {}).get('codeLiteral', '')
142             if code:
143                 self.status = 'ERR-' + str(code)
144                 self.save()
145                 self.status_updated()
146             else:
147                 raise ValueError("Expecting dict with `orderId` key, got: %s" % response)
148         else:
149             self.order_id = response['orderId']
150             self.save()
151
152         return response.get('redirectUri', self.get_thanks_url())
153
154
155 class Notification(models.Model):
156     """ Add `order` FK to real Order model. """
157     body = models.TextField(_('body'))
158     received_at = models.DateTimeField(_('received_at'), auto_now_add=True)
159
160     class Meta:
161         abstract = True
162         verbose_name = _('PayU notification')
163         verbose_name_plural = _('PayU notifications')
164
165     def get_status(self):
166         return json.loads(self.body)['order']['status']
167
168     def apply(self):
169         status = self.get_status()
170         if self.order.status not in (status, 'COMPLETED'):
171             self.order.status = status
172             if status == 'COMPLETED':
173                 self.order.completed_at = now()
174             self.order.save()
175             self.order.status_updated()