re2o/cotisations/payment_methods/comnpay/models.py

from django.db import models
from django.shortcuts import render

from cotisations.models import Paiement
from cotisations.payment_methods.mixins import PaymentMethodMixin

from .aes_field import AESEncryptedField
from .views import comnpay


class ComnpayPayment(PaymentMethodMixin, models.Model):
    """
    The model allowing you to pay with COMNPAY.
    """
    payment = models.OneToOneField(
        Paiement,
        related_name='payment_method',
        editable=False
    )
    payment_credential = models.CharField(
        max_length=255,
        default='',
        blank=True
    )
    payment_pass = AESEncryptedField(
        max_length=255,
        null=True,
        blank=True,
    )

    def end_payment(self, invoice, request):
        invoice.valid = False
        invoice.save()
        content = comnpay(invoice, request)
        return render(request, 'cotisations/payment.html', content)