pbp_com/apps/checkout/integrations/gc_integration.py

""" Google checkout integration for django-merchant """
import logging

from billing.integrations.google_checkout_integration import (
    GoogleCheckoutIntegration as Integration)
from billing.models import GCNewOrderNotification
from billing import signals

from store_order.models import Order, OrderPayment

LOGGER = logging.getLogger('gc_integration')

BUTTON_SANDBOX_URL_SSL = 'https://sandbox.google.com/checkout/buttons/checkout.gif?merchant_id=%(merchant_id)s&w=%(width)s&h=%(height)s&style=white&variant=text&loc=en_US'
BUTTON_URL_SSL = 'https://checkout.google.com/checkout/buttons/checkout.gif?merchant_id=%(merchant_id)s&w=%(width)s&h=%(height)s&style=white&variant=text&loc=en_US'

class GcIntegration(Integration):

    # remove once this is fixed in django-merchant
    def button_image_url(self):
        params = {"merchant_id": self.merchant_id,
                  "width": self.button_width,
                  "height": self.button_height}
        if self.test_mode:
            return BUTTON_SANDBOX_URL_SSL % params
        return BUTTON_URL_SSL % params

    def gc_order_state_change_notification(self, request):
        """ used when a the GC order changed state """
        LOGGER.debug("[gc_order_state_change_notification] new update")
        post_data = request.POST.copy()
        #TODO, clean this up, only for testing.
        LOGGER.debug("[gc_order_state_change_notification] data = {0}".format(
            post_data))
        try:
            order = GCNewOrderNotification.objects.get(
                google_order_number=post_data['google-order-number'])
        except GCNewOrderNotification.DoesNotExist:
            LOGGER.info("[gc_order_state_change_notification] object not found!")
            return

        order.financial_order_state = post_data['new-financial-order-state']

        try:
            order_payments = order.order_payment.all()
            if order_payments:
                order_payment = order_payments[0]
                orig_order = order_payment.order
            else:
                orig_order = None
        except Exception as exp:
            LOGGER.error("[gc_order_state_change_notification] Error = {0}".format(exp))
            orig_order = None

        # Send success signal when order state is charged
        if order.financial_order_state == 'CHARGED':
            signals.transaction_was_successful.send(sender=self.__class__,
                                                    type="purchase",
                                                    response=order)
            if orig_order:
                LOGGER.debug("[gc_order_state_change_notification] status changed to processed")
                orig_order.status = Order.PROCESSED
                orig_order.save()
                # Ideally we'd call this in a signal receiver above, but
                # I wasn't able to get the signal to work.
                from storefront.subscriptions import send_subscription_email
                send_subscription_email(orig_order)

        elif order.financial_order_state in ['CANCELLED', 'CANCELLED_BY_GOOGLE']:
            if orig_order:
                LOGGER.debug("[gc_order_state_change_notification] status changed to cancelled")
                orig_order.status = Order.CANCELLED
                orig_order.save()

        order.fulfillment_order_state = post_data['new-fulfillment-order-state']

        if order.fulfillment_order_state == 'DELIVERED':
            if orig_order:
                LOGGER.debug("[gc_order_state_change_notification] status changed to shipped")
                orig_order.status = Order.SHIPPED
                orig_order.save()

        order.save()

    def gc_new_order_notification(self, request):
        """ new GC order coming in"""
        LOGGER.debug("New GC Order")
        post_data = request.POST.copy()
        data = {}
        resp_fields = {
            "_type": "notify_type",
            "serial-number": "serial_number",
            "google-order-number": "google_order_number",
            "buyer-id": "buyer_id",
            "buyer-shipping-address.contact-name": "shipping_contact_name",
            "buyer-shipping-address.address1": "shipping_address1",
            "buyer-shipping-address.address2": "shipping_address2",
            "buyer-shipping-address.city": "shipping_city",
            "buyer-shipping-address.postal-code": "shipping_postal_code",
            "buyer-shipping-address.region": "shipping_region",
            "buyer-shipping-address.country-code": "shipping_country_code",
            "buyer-shipping-address.email": "shipping_email",
            "buyer-shipping-address.company-name": "shipping_company_name",
            "buyer-shipping-address.fax": "shipping_fax",
            "buyer-shipping-address.phone": "shipping_phone",
            "buyer-billing-address.contact-name": "billing_contact_name",
            "buyer-billing-address.address1": "billing_address1",
            "buyer-billing-address.address2": "billing_address2",
            "buyer-billing-address.city": "billing_city",
            "buyer-billing-address.postal-code": "billing_postal_code",
            "buyer-billing-address.region": "billing_region",
            "buyer-billing-address.country-code": "billing_country_code",
            "buyer-billing-address.email": "billing_email",
            "buyer-billing-address.company-name": "billing_company_name",
            "buyer-billing-address.fax": "billing_fax",
            "buyer-billing-address.phone": "billing_phone",
            "buyer-marketing-preferences.email-allowed": "marketing_email_allowed",
            "order-adjustment.total-tax": "total_tax",
            "order-adjustment.total-tax.currency": "total_tax_currency",
            "order-adjustment.adjustment-total": "adjustment_total",
            "order-adjustment.adjustment-total.currency": "adjustment_total_currency",
            "order-total": "order_total",
            "order-total.currency": "order_total_currency",
            "financial-order-state": "financial_order_state",
            "fulfillment-order-state": "fulfillment_order_state",
            "timestamp": "timestamp",
            "shopping-cart.merchant-private-data": "private_data",
        }

        for (key, val) in resp_fields.iteritems():
            data[val] = post_data.get(key, '')

        data['num_cart_items'] = len(post_data.getlist('shopping-cart.items'))
        data['cart_items'] = self.gc_cart_items_blob(post_data)
        resp = GCNewOrderNotification.objects.create(**data)

        try:
            order_id = int(resp.private_data)
            order = Order.objects.get(pk=order_id)
            order_payment = OrderPayment()
            order_payment.order = order
            order_payment.google_payment = resp
            order_payment.save()
            #order.orderpayment = order_payment
            order.status = Order.SUBMITTED
            #TODO: Make sure status is set correctly.
            order.save()
            LOGGER.debug("New GC Order saved!")
        except Exception as exc:
            #TODO log error, send email.
            LOGGER.error("[gc_new_order_notification] Error = {0}".format(exc))