source: main/waeup.ikoba/branches/uli-payments/src/waeup/ikoba/payments/demo_provider.py @ 12737

import grok
from waeup.ikoba.interfaces import MessageFactory as _
from waeup.ikoba.browser.layout import IkobaEditFormPage, action
from waeup.ikoba.payments.interfaces import (
    IPaymentGatewayService, IPayment, IPayable, IPayer, IPayee,)
from waeup.ikoba.payments.payment import (
    Payment, get_payment, find_payable_from_payable_id,
    PaymentProviderServiceBase)


grok.templatedir('templates')


WARN_FINAL_SUBMIT = _(
    'You can not edit your contract after final submission. '
    'Do you really want to submit?'
)


class DemoCreditcardPaymentService(PaymentProviderServiceBase):
    """A demo payment gateway service.

    This one supports credit card payments.
    """
    grok.implements(IPaymentGatewayService)
    grok.name('demo_creditcard')

    title = _(u'Credit Card (Demo Payments)')

    def create_payment(self, payer, payable,  payee=None):
        """Create a payment.
        """
        if not IPayer.providedBy(payer):
            payer = IPayer(payer)
        if not IPayable.providedBy(payable):
            payable = IPayable(payable)
        if (payee is not None) and (not IPayee.providedBy(payee)):
            payee = IPayee(payee)
        payment = Payment(payer, payable, payee)
        payment.gateway_service = 'demo_creditcard'  # must be grok.name above
        return payment

    def next_step(self, payment_id):
        """Tell where to go next.

        Returns (context, view_name). Both may be none.
        """
        payment = get_payment(payment_id)
        if payment is None:
            return None, None
        return payment, 'demo_cc1'


class CreditCardStep1(IkobaEditFormPage):
    grok.context(IPayment)
    grok.name('demo_cc1')
    # XXX: Use own permissions for payments
    grok.require('waeup.Authenticated')
    label = "Enter Credit Card Details"
    grok.template('demo_cc_step1')
    pnav = 4

    def update(self):
        pass

    @action(_('Authorize Payment'), warning=WARN_FINAL_SUBMIT,
            style="primary")
    def authorize(self, **data):
        print "AUTH!"

    @action(_('Cancel'))
    def cancel(self, **data):
        """Go back to the payable (if possible) or site home.
        """
        payable_id = getattr(self.context, 'payable_id', '')
        payed_item = find_payable_from_payable_id(payable_id)
        self.flash(_("Payment cancelled."))
        if payed_item is not None:
            # remove context/payment?
            target = self.url(payed_item)
        else:
            target = grok.getSite()
        self.redirect(self.url(target))