import stripe
from django.http import HttpResponse
from django.core.mail import send_mail
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:
#
# - X Sort out the webhook
# - Associate the purchases with the users profile page
# - We need a profile page!
# - Fix the email and make it nice
# - 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
def create_line_items(resources):
price_objs = [r.price_obj.first() for r in resources]
stripe_price_ids = [p.stripe_price_id for p in price_objs]
return [{"quantity": 1, "price": s} for s in stripe_price_ids]
pass
class CreateCheckoutSessionView(View):
def post(self, request, *args, **kwargs):
#price = Price.objects.get(id=self.kwargs["pk"])
cart = ShoppingCart.objects.get(id=self.kwargs["pk"])
resources = [i.resource for i in cart.items.all()]
total = sum([r.price_obj.first().price for r in resources])
domain = "http://localhost:8000"
checkout_session = stripe.checkout.Session.create(
payment_method_types=["card"],
customer_email=request.user.email,
line_items=create_line_items(resources),
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):
resource = Resource.objects.get(id=91)
prices = Price.objects.filter(resource=resource)
context = super(ProductLandingPageView, self).get_context_data(**kwargs)
context.update({"resource": resource, "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 redirect("payments:cart_detail")
@login_required
def cart_detail(request):
cart, created = ShoppingCart.objects.get_or_create(user=request.user)
resources = [i.resource for i in cart.items.all()]
total = sum([r.price_obj.first().price for r in resources])
context = {
"cart": cart,
"resources": resources,
"total": total
}
return render(request, "payments/cart_detail.html", context)
# 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 Add the items to the users profile page
# This should be done before deleting the cart otherwise we have no record of what they bought
# TODO clear their shopping cart - we need to obtain their user ID
user = User.objects.get(email=customer_email)
ShoppingCart.objects.get(user=user).delete()
print(f"Deleted cart for {user}")
# TODO add the transaction to our local history? Or just use Stripe?
# TODO send an email to the customer
send_mail(
"Thank you for your purchase",
"You have bought something nice - enjoy it",
settings.DEFAULT_FROM_EMAIL,
[customer_email],
fail_silently=False,
)
return HttpResponse(status=200)
