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()