regluit/payment/views.py

"""
external library imports
"""
import logging
import traceback
import uuid

from decimal import Decimal as D
from unittest import TestResult

"""
django imports
"""
from django.conf import settings
from django.contrib.auth.models import User
from django.contrib.sites.models import RequestSite
from django.core.urlresolvers import reverse
from django.http import (
    HttpResponse,
    HttpRequest,
    HttpResponseRedirect,
    HttpResponseBadRequest
)
from django.shortcuts import render_to_response
from django.template import RequestContext
from django.test.utils import setup_test_environment
from django.views.decorators.csrf import csrf_exempt
from django.views.generic.edit import FormView
from django.views.generic.base import TemplateView

"""
regluit imports
"""
from regluit.core.models import Campaign, Wishlist
from regluit.payment.forms import StripePledgeForm
from regluit.payment.manager import PaymentManager
from regluit.payment.models import Transaction
from regluit.payment.parameters import *
from regluit.payment.stripelib import STRIPE_PK
from regluit.payment.tests import PledgeTest, AuthorizeTest
from regluit.utils.localdatetime import now

logger = logging.getLogger(__name__)

# parameterize some test recipients
TEST_RECEIVERS = ['seller_1317463643_biz@gmail.com', 'buyer5_1325740224_per@gmail.com']
#TEST_RECEIVERS = ['seller_1317463643_biz@gmail.com', 'Buyer6_1325742408_per@gmail.com']
#TEST_RECEIVERS = ['glueja_1317336101_biz@gluejar.com', 'rh1_1317336251_biz@gluejar.com', 'RH2_1317336302_biz@gluejar.com']

'''
http://BASE/querycampaign?id=2

Example that returns a summary total for a campaign
'''
def queryCampaign(request):
    
    id = request.GET['id']
    campaign = Campaign.objects.get(id=id)
    
    p = PaymentManager()
    
    # transactions = p.query_campaign(campaign)
    
    total = p.query_campaign(campaign, summary=True)
    
    return HttpResponse(str(total))

'''
http://BASE/testexecute?campaign=2

Example that executes a set of transactions that are pre-approved
'''
def testExecute(request):
    
    p = PaymentManager()
    
    if 'campaign' in request.GET.keys():
        campaign_id = request.GET['campaign']
        campaign = Campaign.objects.get(id=int(campaign_id))
    else:
        campaign = None
        
    output = ''
        
    if campaign:
        result = p.execute_campaign(campaign)
        
        for t in result:
            output += str(t)
            logger.info(str(t))
            
    else:
        transactions = Transaction.objects.filter(status=TRANSACTION_STATUS_ACTIVE)
        
        for t in transactions:
            
            # Note, set this to 1-5 different receivers with absolute amounts for each
            receiver_list = [{'email':TEST_RECEIVERS[0], 'amount':float(t.amount) * 0.80}, 
                            {'email':TEST_RECEIVERS[1], 'amount':float(t.amount) * 0.20}]
    
            p.execute_transaction(t, receiver_list)
            output += str(t)
            logger.info(str(t))
        
    return HttpResponse(output)
        

'''
http://BASE/testauthorize?amount=20

Example that initiates a pre-approval for a set amount
'''   
def testAuthorize(request):
    
    p = PaymentManager()
    
    if 'campaign' in request.GET.keys():
        campaign_id = request.GET['campaign']
    else:
        campaign_id = None
        
    if 'amount' in request.GET.keys():
        amount = float(request.GET['amount'])
    else:
        return HttpResponse("Error, no amount in request")
        
    
    # Note, set this to 1-5 different receivers with absolute amounts for each
    receiver_list = [{'email': TEST_RECEIVERS[0], 'amount':20.00}, 
                     {'email': TEST_RECEIVERS[1], 'amount':10.00}]
        
    if campaign_id:
        campaign = Campaign.objects.get(id=int(campaign_id))
        t, url = p.authorize('USD', TARGET_TYPE_CAMPAIGN, amount, campaign=campaign, return_url=None, list=None, user=None)
    
    else:
        t, url = p.authorize('USD', TARGET_TYPE_NONE, amount, campaign=None, return_url=None, list=None, user=None)
    
    if url:
        logger.info("testAuthorize: " + url)
        return HttpResponseRedirect(url)
    
    else:
        response = t.error
        logger.info("testAuthorize: Error " + str(t.error))
        return HttpResponse(response)
 
'''
http://BASE/testcancel?transaction=2

Example that cancels a preapproved transaction
'''    
def testCancel(request):
    
    if "transaction" not in request.GET.keys():
        return HttpResponse("No Transaction in Request")
    
    t = Transaction.objects.get(id=int(request.GET['transaction']))
    p = PaymentManager()
    if p.cancel_transaction(t):
        return HttpResponse("Success")
    else:
        message = "Error: " + t.error
        return HttpResponse(message)
    
'''
http://BASE/testrefund?transaction=2

Example that refunds a transaction
'''    
def testRefund(request):
    
    if "transaction" not in request.GET.keys():
        return HttpResponse("No Transaction in Request")
    
    t = Transaction.objects.get(id=int(request.GET['transaction']))
    p = PaymentManager()
    if p.refund_transaction(t):
        return HttpResponse("Success")
    else:
        if t.error:
            message = "Error: " + t.error
        else:
            message = "Error"
            
        return HttpResponse(message)
    
'''
http://BASE/testmodify?transaction=2

Example that modifies the amount of a transaction
'''    
def testModify(request):
    
    if "transaction" not in request.GET.keys():
        return HttpResponse("No Transaction in Request")
    
    if "amount" in request.GET.keys():
        amount = float(request.GET['amount'])
    else:
        amount = 200.0
    
    t = Transaction.objects.get(id=int(request.GET['transaction']))
    p = PaymentManager()
        
    status, url = p.modify_transaction(t, amount, return_url=None)
    
    if url:
        logger.info("testModify: " + url)
        return HttpResponseRedirect(url)
    
    if status:
        return HttpResponse("Success")
    else:
        return HttpResponse("Error")
    
    
'''
http://BASE/testfinish?transaction=2

Example that finishes a delayed chained transaction
'''    
def testFinish(request):
    
    if "transaction" not in request.GET.keys():
        return HttpResponse("No Transaction in Request")
    
    t = Transaction.objects.get(id=int(request.GET['transaction']))
    p = PaymentManager()
    if p.finish_transaction(t):
        return HttpResponse("Success")
    else:
        message = "Error: " + t.error
        return HttpResponse(message)


def runTests(request):

    try:
        # Setup the test environement.  We need to run these tests on a live server
        # so our code can receive IPN notifications from paypal
        setup_test_environment()
        result = TestResult()
        
        # Run the authorize test
        test = AuthorizeTest('test_authorize')
        test.run(result)   

        output = "Tests Run: " + str(result.testsRun) + str(result.errors) + str(result.failures)
        logger.info(output)
    
        return HttpResponse(output)
    
    except:
        traceback.print_exc()
    
@csrf_exempt
def handleIPN(request, module):
    # Handler for IPN /webhook notifications
    
    p = PaymentManager()
    logger.info(str(request.POST))
    return p.processIPN(request, module)
    
def paymentcomplete(request):
    # pick up all get and post parameters and display
    output = "payment complete"
    output += request.method + "\n" + str(request.GET.items()) + "\n" + str(request.POST.items())
    return HttpResponse(output)

def checkStatus(request):
    # Check the status of all PAY transactions and flag any errors
    p = PaymentManager()
    error_data = p.checkStatus()
        
    return HttpResponse(error_data, mimetype="text/xml")

# https://raw.github.com/agiliq/merchant/master/example/app/views.py

def _render(request, template, template_vars={}):
    return render_to_response(template, template_vars, RequestContext(request))
    
class StripeView(FormView):
    template_name="stripe.html"
    form_class = StripePledgeForm

    def get_context_data(self, **kwargs):
        
        context = super(StripeView, self).get_context_data(**kwargs)
    
        context.update({
                'STRIPE_PK':STRIPE_PK
                })
        return context
    
    def form_valid(self, form):
        stripeToken = form.cleaned_data["stripeToken"]
        # e.g., tok_0C0k4jG5B2Oxox
        # 
        return HttpResponse("stripeToken: {0}".format(stripeToken))