1cd689ddefcee33a26da2369164b53014dd93cf7
[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 . import POSS
11
12
13 class CardToken(models.Model):
14     """ This should be attached to a payment schedule. """
15     pos_id = models.CharField(max_length=255)
16     disposable_token = models.CharField(max_length=255)
17     reusable_token = models.CharField(max_length=255, null=True, blank=True)
18     created_at = models.DateTimeField(auto_now_add=True)
19
20     class Meta:
21         abstract = True
22
23
24 class Order(models.Model):
25     pos_id = models.CharField(max_length=255)   # TODO: redundant?
26     customer_ip = models.GenericIPAddressField()
27     order_id = models.CharField(max_length=255, blank=True)
28
29     status = models.CharField(max_length=128, blank=True, choices=[
30         ('PENDING', 'Pending'),
31         ('WAITING_FOR_CONFIRMATION', 'Waiting for confirmation'),
32         ('COMPLETED', 'Completed'),
33         ('CANCELED', 'Canceled'),
34         ('REJECTED', 'Rejected'),
35     ])
36
37     class Meta:
38         abstract = True
39
40     # These need to be provided in a subclass.
41
42     def get_amount(self):
43         raise NotImplementedError()
44
45     def get_buyer(self):
46         raise NotImplementedError()
47
48     def get_continue_url(self):
49         raise NotImplementedError()
50     
51     def get_description(self):
52         raise NotImplementedError()
53
54     def get_products(self):
55         """ At least: name, unitPrice, quantity. """
56         return [
57             {
58                 'name': self.get_description(),
59                 'unitPrice': str(int(self.get_amount() * 100)),
60                 'quantity': '1',
61             },
62         ]
63
64     def is_recurring(self):
65         return False
66
67     def get_notify_url(self):
68         raise NotImplementedError
69
70     def status_updated(self):
71         pass
72
73     # 
74     
75     def get_pos(self):
76         return POSS[self.pos_id]
77
78     def get_representation(self, token=None):
79         rep = {
80             "notifyUrl": self.get_notify_url(),
81             "customerIp": self.customer_ip,
82             "merchantPosId": self.pos_id,
83             "currencyCode": self.get_pos().currency_code,
84             "totalAmount": str(int(self.get_amount() * 100)),
85             "extOrderId": "wolne-lektury-rcz-%d" % self.pk,
86
87             "buyer": self.get_buyer() or {},
88             "continueUrl": self.get_continue_url(),
89             "description": self.get_description(),
90             "products": self.get_products(),
91         }
92         if token:
93             token = self.get_card_token()
94             rep['recurring'] = 'STANDARD' if token.reusable_token else 'FIRST'
95             rep['payMethods'] = {
96                 "payMethod": {
97                     "type": "CARD_TOKEN",
98                     "value": token.reusable_token or token.disposable_token
99                 }
100             }
101         return rep
102
103     def put(self):
104         token = self.get_card_token() if self.is_recurring() else None
105         representation = self.get_representation(token)
106         try:
107             response = self.get_pos().request('POST', 'orders', representation)
108         except HTTPError as e:
109             resp = json.loads(e.read().decode('utf-8'))
110             if resp['status']['statusCode'] == 'ERROR_ORDER_NOT_UNIQUE':
111                 pass
112             else:
113                 raise
114
115         if token:
116             reusable_token = response.get('payMethods', {}).get('payMethod', {}).get('value', None)
117             if reusable_token:
118                 token.reusable_token = reusable_token
119                 token.save()
120             # else?
121
122         self.order_id = response['orderId']
123         self.save()
124
125         
126         return response.get('redirectUri', self.schedule.get_thanks_url())
127
128
129 class Notification(models.Model):
130     """ Add `order` FK to real Order model. """
131     body = models.TextField()
132     received_at = models.DateTimeField(auto_now_add=True)
133
134     class Meta:
135         abstract = True
136
137     def get_status(self):
138         return json.loads(self.body)['order']['status']
139
140     def apply(self):
141         status = self.get_status()
142         if self.order.status not in (status, 'COMPLETED'):
143             self.order.status = status
144             self.order.save()
145             self.order.status_updated()