Implement subscription management and ordering

core/lib/customers.py

@@ -1,5 +1,10 @@
+import logging
+
 import stripe
 
+logger = logging.getLogger(__name__)
+
+
 def expand_name(first_name, last_name):
     """
     Convert two name variables into one.
@@ -13,6 +18,7 @@ def expand_name(first_name, 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.
@@ -23,7 +29,7 @@ def get_or_create(email, first_name, last_name):
     matching_customers = stripe.Customer.list(email=email, limit=2)
     if len(matching_customers) == 2:
         # Something is horribly wrong
-        print("Two customers match!")
+        logger.error(f"Two customers found for email {email}")
         raise Exception(f"Two customers found for email {email}")
 
     elif len(matching_customers) == 1:
@@ -41,9 +47,11 @@ def get_or_create(email, first_name, last_name):
         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
 
+
 def update_customer_fields(stripe_id, email=None, first_name=None, last_name=None):
     """
     Update the customer fields in Stripe.
@@ -52,7 +60,9 @@ def update_customer_fields(stripe_id, email=None, first_name=None, last_name=Non
     if email:
         print("Email modified")
         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:
         print("Name modified")
         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}")

core/lib/products.py

@@ -0,0 +1,16 @@
+from core.models import Plan
+
+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 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