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