Begin implementing billing
parent
0937f7299a
commit
ac4c248175
GPG Key ID:
5ACFCEED46C0904F
@ -1,4 +1,11 @@
|
|||||||
# from lago_python_client import Client
|
from django.conf import settings
|
||||||
# from django.conf import settings
|
from lago_python_client import Client
|
||||||
|
from lago_python_client.models import (
|
||||||
|
Charge,
|
||||||
|
Charges,
|
||||||
|
Customer,
|
||||||
|
CustomerBillingConfiguration,
|
||||||
|
Plan,
|
||||||
|
)
|
||||||
|
|
||||||
# client = Client(api_key=settings.LAGO_API_KEY, api_url=settings.LAGO_URL)
|
client = Client(api_key=settings.LAGO_API_KEY, api_url=settings.LAGO_URL)
|
||||||
|
|||||||
@ -1,65 +1,65 @@
|
|||||||
import logging
|
# import logging
|
||||||
|
|
||||||
import stripe
|
# import stripe
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
# logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def expand_name(first_name, last_name):
|
# def expand_name(first_name, last_name):
|
||||||
"""
|
# """
|
||||||
Convert two name variables into one.
|
# Convert two name variables into one.
|
||||||
Last name without a first name is ignored.
|
# Last name without a first name is ignored.
|
||||||
"""
|
# """
|
||||||
name = None
|
# name = None
|
||||||
if first_name:
|
# if first_name:
|
||||||
name = first_name
|
# name = first_name
|
||||||
# We only want to put the last name if we have a first name
|
# # We only want to put the last name if we have a first name
|
||||||
if last_name:
|
# if last_name:
|
||||||
name += f" {last_name}"
|
# name += f" {last_name}"
|
||||||
return name
|
# return name
|
||||||
|
|
||||||
|
|
||||||
def get_or_create(email, first_name, last_name):
|
# def get_or_create(email, first_name, last_name):
|
||||||
"""
|
# """
|
||||||
Get a customer ID from Stripe if one with the given email exists.
|
# Get a customer ID from Stripe if one with the given email exists.
|
||||||
Create a customer if one does not.
|
# Create a customer if one does not.
|
||||||
Raise an exception if two or more customers matching the given email exist.
|
# Raise an exception if two or more customers matching the given email exist.
|
||||||
"""
|
# """
|
||||||
# Let's see if we're just missing the ID
|
# # Let's see if we're just missing the ID
|
||||||
matching_customers = stripe.Customer.list(email=email, limit=2)
|
# matching_customers = stripe.Customer.list(email=email, limit=2)
|
||||||
if len(matching_customers) == 2:
|
# if len(matching_customers) == 2:
|
||||||
# Something is horribly wrong
|
# # Something is horribly wrong
|
||||||
logger.error(f"Two customers found for email {email}")
|
# logger.error(f"Two customers found for email {email}")
|
||||||
raise Exception(f"Two customers found for email {email}")
|
# raise Exception(f"Two customers found for email {email}")
|
||||||
|
|
||||||
elif len(matching_customers) == 1:
|
# elif len(matching_customers) == 1:
|
||||||
# We found a customer. Let's copy the ID
|
# # We found a customer. Let's copy the ID
|
||||||
customer = matching_customers["data"][0]
|
# customer = matching_customers["data"][0]
|
||||||
customer_id = customer["id"]
|
# customer_id = customer["id"]
|
||||||
return customer_id
|
# return customer_id
|
||||||
|
|
||||||
else:
|
# else:
|
||||||
# We didn't find anything. Create the customer
|
# # We didn't find anything. Create the customer
|
||||||
|
|
||||||
# Create a name, since we have 2 variables which could be null
|
# # Create a name, since we have 2 variables which could be null
|
||||||
name = expand_name(first_name, last_name)
|
# name = expand_name(first_name, last_name)
|
||||||
cast = {"email": email}
|
# cast = {"email": email}
|
||||||
if name:
|
# if name:
|
||||||
cast["name"] = name
|
# cast["name"] = name
|
||||||
customer = stripe.Customer.create(**cast)
|
# customer = stripe.Customer.create(**cast)
|
||||||
logger.info(f"Created new Stripe customer {customer.id} with email {email}")
|
# logger.info(f"Created new Stripe customer {customer.id} with email {email}")
|
||||||
|
|
||||||
return customer.id
|
# return customer.id
|
||||||
|
|
||||||
|
|
||||||
def update_customer_fields(stripe_id, email=None, first_name=None, last_name=None):
|
# def update_customer_fields(stripe_id, email=None, first_name=None, last_name=None):
|
||||||
"""
|
# """
|
||||||
Update the customer fields in Stripe.
|
# Update the customer fields in Stripe.
|
||||||
"""
|
# """
|
||||||
if email:
|
# if email:
|
||||||
stripe.Customer.modify(stripe_id, email=email)
|
# stripe.Customer.modify(stripe_id, email=email)
|
||||||
logger.info(f"Modified Stripe customer {stripe_id} to have email {email}")
|
# logger.info(f"Modified Stripe customer {stripe_id} to have email {email}")
|
||||||
if first_name or last_name:
|
# if first_name or last_name:
|
||||||
name = expand_name(first_name, last_name)
|
# name = expand_name(first_name, last_name)
|
||||||
stripe.Customer.modify(stripe_id, name=name)
|
# stripe.Customer.modify(stripe_id, name=name)
|
||||||
logger.info(f"Modified Stripe customer {stripe_id} to have email {name}")
|
# logger.info(f"Modified Stripe customer {stripe_id} to have email {name}")
|
||||||
|
|||||||
@ -1,21 +1,21 @@
|
|||||||
from asgiref.sync import sync_to_async
|
# from asgiref.sync import sync_to_async
|
||||||
|
|
||||||
from core.models import Plan
|
# from core.models import Plan
|
||||||
|
|
||||||
|
|
||||||
async def assemble_plan_map(product_id_filter=None):
|
# async def assemble_plan_map(product_id_filter=None):
|
||||||
"""
|
# """
|
||||||
Get all the plans from the database and create an object Stripe wants.
|
# Get all the plans from the database and create an object Stripe wants.
|
||||||
"""
|
# """
|
||||||
line_items = []
|
# line_items = []
|
||||||
for plan in await sync_to_async(list)(Plan.objects.all()):
|
# for plan in await sync_to_async(list)(Plan.objects.all()):
|
||||||
if product_id_filter:
|
# if product_id_filter:
|
||||||
if plan.product_id != product_id_filter:
|
# if plan.product_id != product_id_filter:
|
||||||
continue
|
# continue
|
||||||
line_items.append(
|
# line_items.append(
|
||||||
{
|
# {
|
||||||
"price": plan.product_id,
|
# "price": plan.product_id,
|
||||||
"quantity": 1,
|
# "quantity": 1,
|
||||||
}
|
# }
|
||||||
)
|
# )
|
||||||
return line_items
|
# return line_items
|
||||||
|
|||||||
@ -1,11 +0,0 @@
|
|||||||
from django import template
|
|
||||||
|
|
||||||
register = template.Library()
|
|
||||||
|
|
||||||
|
|
||||||
@register.filter
|
|
||||||
def has_plan(user, plan_name):
|
|
||||||
if not hasattr(user, "plans"):
|
|
||||||
return False
|
|
||||||
plan_list = [plan.name for plan in user.plans.all()]
|
|
||||||
return plan_name in plan_list
|
|
||||||
@ -1,104 +1,104 @@
|
|||||||
import logging
|
# import logging
|
||||||
from datetime import datetime
|
# from datetime import datetime
|
||||||
|
|
||||||
import stripe
|
# import stripe
|
||||||
from django.conf import settings
|
# from django.conf import settings
|
||||||
from django.http import HttpResponse, JsonResponse
|
# from django.http import HttpResponse, JsonResponse
|
||||||
from django.views.decorators.csrf import csrf_exempt
|
# from django.views.decorators.csrf import csrf_exempt
|
||||||
from rest_framework.parsers import JSONParser
|
# from rest_framework.parsers import JSONParser
|
||||||
from rest_framework.views import APIView
|
# from rest_framework.views import APIView
|
||||||
|
|
||||||
from core.models import Plan, Session, User
|
# from core.models import Plan, Session, User
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
# logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class Callback(APIView):
|
# class Callback(APIView):
|
||||||
parser_classes = [JSONParser]
|
# parser_classes = [JSONParser]
|
||||||
|
|
||||||
# TODO: make async
|
# # TODO: make async
|
||||||
@csrf_exempt
|
# @csrf_exempt
|
||||||
def post(self, request):
|
# def post(self, request):
|
||||||
payload = request.body
|
# payload = request.body
|
||||||
sig_header = request.META["HTTP_STRIPE_SIGNATURE"]
|
# sig_header = request.META["HTTP_STRIPE_SIGNATURE"]
|
||||||
try:
|
# try:
|
||||||
stripe.Webhook.construct_event(
|
# stripe.Webhook.construct_event(
|
||||||
payload, sig_header, settings.STRIPE_ENDPOINT_SECRET
|
# payload, sig_header, settings.STRIPE_ENDPOINT_SECRET
|
||||||
)
|
# )
|
||||||
except ValueError:
|
# except ValueError:
|
||||||
# Invalid payload
|
# # Invalid payload
|
||||||
logger.error("Invalid payload")
|
# logger.error("Invalid payload")
|
||||||
return HttpResponse(status=400)
|
# return HttpResponse(status=400)
|
||||||
except stripe.error.SignatureVerificationError:
|
# except stripe.error.SignatureVerificationError:
|
||||||
# Invalid signature
|
# # Invalid signature
|
||||||
logger.error("Invalid signature")
|
# logger.error("Invalid signature")
|
||||||
return HttpResponse(status=400)
|
# return HttpResponse(status=400)
|
||||||
|
|
||||||
if request.data is None:
|
# if request.data is None:
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
if "type" in request.data.keys():
|
# if "type" in request.data.keys():
|
||||||
rtype = request.data["type"]
|
# rtype = request.data["type"]
|
||||||
if rtype == "checkout.session.completed":
|
# if rtype == "checkout.session.completed":
|
||||||
session = request.data["data"]["object"]["id"]
|
# session = request.data["data"]["object"]["id"]
|
||||||
subscription_id = request.data["data"]["object"]["subscription"]
|
# subscription_id = request.data["data"]["object"]["subscription"]
|
||||||
session_map = Session.objects.get(session=session)
|
# session_map = Session.objects.get(session=session)
|
||||||
if not session_map:
|
# if not session_map:
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
user = session_map.user
|
# user = session_map.user
|
||||||
session_map.subscription_id = subscription_id
|
# session_map.subscription_id = subscription_id
|
||||||
session_map.save()
|
# session_map.save()
|
||||||
|
|
||||||
if rtype == "customer.subscription.updated":
|
# if rtype == "customer.subscription.updated":
|
||||||
stripe_id = request.data["data"]["object"]["customer"]
|
# stripe_id = request.data["data"]["object"]["customer"]
|
||||||
if not stripe_id:
|
# if not stripe_id:
|
||||||
logging.error("No stripe id")
|
# logging.error("No stripe id")
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
user = User.objects.get(stripe_id=stripe_id)
|
# user = User.objects.get(stripe_id=stripe_id)
|
||||||
# ssubscription_active
|
# # ssubscription_active
|
||||||
subscription_id = request.data["data"]["object"]["id"]
|
# subscription_id = request.data["data"]["object"]["id"]
|
||||||
sessions = Session.objects.filter(user=user)
|
# sessions = Session.objects.filter(user=user)
|
||||||
session = None
|
# session = None
|
||||||
for session_iter in sessions:
|
# for session_iter in sessions:
|
||||||
if session_iter.subscription_id == subscription_id:
|
# if session_iter.subscription_id == subscription_id:
|
||||||
session = session_iter
|
# session = session_iter
|
||||||
if not session:
|
# if not session:
|
||||||
logging.error(
|
# logging.error(
|
||||||
f"No session found for subscription id {subscription_id}"
|
# f"No session found for subscription id {subscription_id}"
|
||||||
)
|
# )
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
# query Session objects
|
# # query Session objects
|
||||||
# iterate and check against product_id
|
# # iterate and check against product_id
|
||||||
session.request = request.data["request"]["id"]
|
# session.request = request.data["request"]["id"]
|
||||||
product_id = request.data["data"]["object"]["plan"]["id"]
|
# product_id = request.data["data"]["object"]["plan"]["id"]
|
||||||
plan = Plan.objects.get(product_id=product_id)
|
# plan = Plan.objects.get(product_id=product_id)
|
||||||
if not plan:
|
# if not plan:
|
||||||
logging.error(f"Plan not found: {product_id}")
|
# logging.error(f"Plan not found: {product_id}")
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
session.plan = plan
|
# session.plan = plan
|
||||||
session.save()
|
# session.save()
|
||||||
|
|
||||||
elif rtype == "payment_intent.succeeded":
|
# elif rtype == "payment_intent.succeeded":
|
||||||
customer = request.data["data"]["object"]["customer"]
|
# customer = request.data["data"]["object"]["customer"]
|
||||||
user = User.objects.get(stripe_id=customer)
|
# user = User.objects.get(stripe_id=customer)
|
||||||
if not user:
|
# if not user:
|
||||||
logging.error(f"No user found for customer: {customer}")
|
# logging.error(f"No user found for customer: {customer}")
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
session = Session.objects.get(request=request.data["request"]["id"])
|
# session = Session.objects.get(request=request.data["request"]["id"])
|
||||||
|
|
||||||
user.plans.add(session.plan)
|
# user.plans.add(session.plan)
|
||||||
user.last_payment = datetime.utcnow()
|
# user.last_payment = datetime.utcnow()
|
||||||
user.save()
|
# user.save()
|
||||||
|
|
||||||
elif rtype == "customer.subscription.deleted":
|
# elif rtype == "customer.subscription.deleted":
|
||||||
customer = request.data["data"]["object"]["customer"]
|
# customer = request.data["data"]["object"]["customer"]
|
||||||
user = User.objects.get(stripe_id=customer)
|
# user = User.objects.get(stripe_id=customer)
|
||||||
if not user:
|
# if not user:
|
||||||
logging.error(f"No user found for customer {customer}")
|
# logging.error(f"No user found for customer {customer}")
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
product_id = request.data["data"]["object"]["plan"]["id"]
|
# product_id = request.data["data"]["object"]["plan"]["id"]
|
||||||
plan = Plan.objects.get(product_id=product_id)
|
# plan = Plan.objects.get(product_id=product_id)
|
||||||
user.plans.remove(plan)
|
# user.plans.remove(plan)
|
||||||
user.save()
|
# user.save()
|
||||||
else:
|
# else:
|
||||||
return JsonResponse({"success": False}, status=500)
|
# return JsonResponse({"success": False}, status=500)
|
||||||
return JsonResponse({"success": True})
|
# return JsonResponse({"success": True})
|
||||||
|
|||||||
Loading…
Reference in New Issue