edan_f0/payment/models.py

from django.contrib.auth.models import User
from django.db import models
from django.utils.translation import ugettext as _


class Source(models.Model):
    """
    A source of payment for an order.  
    
    This is normally a credit card which has been pre-authed
    for the order amount, but some applications will allow orders to be paid for using multiple
    sources such as cheque, credit accounts, gift cards.  Each payment source will have its own
    entry.
    """
    order = models.ForeignKey('order.Order', related_name='sources')
    type = models.CharField(max_length=128)
    initial_amount = models.IntegerField()
    balance = models.IntegerField()
    reference = models.CharField(max_length=128, blank=True, null=True)
    
    def __unicode__(self):
        description = "Payment of %.2f from %s" % (self.initial_amount, self.type)
        if self.reference:
            description += " (reference: %s)" % self.reference
        return description
    
    
class Transaction(models.Model):
    """
    A transaction for payment sources which need a secondary 'transaction' to actually take the money
    
    This applies mainly to credit card sources which can be a pre-auth for the money.  A 'complete'
    needs to be run later to debit the money from the account.
    """
    source = models.ForeignKey('payment.Source', related_name='transactions')
    type = models.CharField(max_length=128, blank=True)
    delta_amount = models.FloatField()
    reference = models.CharField(max_length=128)
    transaction_date = models.DateField()
    
    def __unicode__(self):
        return "Transaction of %.2f" % self.delta_amount