import stripe
from django.http import HttpResponse
from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.shortcuts import get_object_or_404
from django.shortcuts import redirect
from django.shortcuts import render
from django.views import View
from django.views.generic import TemplateView
from django.views.decorators.csrf import csrf_exempt
from alphabetlearning.resources.models import Resource
from alphabetlearning.users.models import User
from .models import CartItem
from .models import Price
from .models import Product
from .models import ShoppingCart
# TODO get the cart integrated with Stripe
# Steps to convert our Cart into something that can be used with Stripe:
#
# - Sort out the webhook
# - Associate each of our resources with a Product item
# - this should be done in the create resource page
# - or we can do it manually for the time being
# - Check that we can add resources to the cart
# - When the user hits the cart page, each associated product is pulled from the database
# - prices are extracted from the products accordingly
# - use the form from the landingpage.html as the checkout button to call the create-checkout-session
# - Add delete buttons the checkout page, etc
stripe.api_key = settings.STRIPE_SECRET_KEY
class CreateCheckoutSessionView(View):
def post(self, request, *args, **kwargs):
price = Price.objects.get(id=self.kwargs["pk"])
domain = "http://localhost:8000"
checkout_session = stripe.checkout.Session.create(
payment_method_types=["card"],
line_items=[
{
"price": price.stripe_price_id,
"quantity": 1,
},
],
mode="payment",
success_url=domain + "/payments/success/",
cancel_url=domain + "/payments/cancel/",
)
return redirect(checkout_session.url, code=303)
class SuccessView(TemplateView):
template_name = "payments/success.html"
class CancelView(TemplateView):
template_name = "payments/cancel.html"
class ProductLandingPageView(TemplateView):
template_name = "payments/landingpage.html"
def get_context_data(self, **kwargs):
product = Product.objects.get(name="Worksheet 1")
prices = Price.objects.filter(product=product)
context = super(ProductLandingPageView, self).get_context_data(**kwargs)
context.update({"product": product, "prices": prices})
return context
@login_required
def add_to_cart(request, resource_id):
resource = get_object_or_404(Resource, id=resource_id)
cart, created = ShoppingCart.objects.get_or_create(user=request.user)
cart_item, created = CartItem.objects.get_or_create(cart=cart, resource=resource)
# cart_item.quantity += 1
cart_item.save()
return render(request, "payments/cart_detail.html", {"cart": cart})
@login_required
def cart_detail(request):
cart, created = ShoppingCart.objects.get_or_create(user=request.user)
return render(request, "payments/cart_detail.html", {"cart": cart})
# def cart_detail(request):
# cart, created = ShoppingCart.objects.get_or_create(user=request.user)
# return render(request, "payments/cart_detail.html", {"cart": cart})
@login_required
def checkout(request):
cart = ShoppingCart.objects.get(user=request.user)
total = sum(item.get_total_price() for item in cart.items.all())
if request.method == "POST":
# Create Stripe PaymentIntent
intent = stripe.PaymentIntent.create(
amount=int(total * 100), # Stripe amount is in cents
currency="usd",
automatic_payment_methods={"enabled": True},
)
# Redirect to Stripe checkout or handle payment confirmation
return render(request, "cart/checkout_success.html")
return render(request, "cart/checkout.html", {"cart": cart, "total": total})
@csrf_exempt
def stripe_webhook(request):
payload = request.body
sig_header = request.META["HTTP_STRIPE_SIGNATURE"]
event = None
# verify that the request came from STRIPE
try:
event = stripe.Webhook.construct_event(payload, sig_header, settings.STRIPE_WEBHOOK_SECRET)
except ValueError as e:
return HttpResponse(status=400)
except stripe.error.SignatureVerificationError as e:
return HttpResponse(status=400)
if event["type"] == "checkout.session.completed":
session = event["data"]["object"]
customer_email = session["customer_details"]["email"]
payment_intent = session["payment_intent"]
# TODO send an email to the customer
print("Here we send an email to the customer")
return HttpResponse(status=200)
