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