From 9a6633bbd145652e85aed92bd92b2d369a71ee2d Mon Sep 17 00:00:00 2001
From: Nick Sergeant <nick@nicksergeant.com>
Date: Fri, 30 May 2014 14:30:35 -0400
Subject: [PATCH] Support custom plans via URL.

---
 templates/pro.html | 6 +++---
 views.py           | 8 ++++----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/templates/pro.html b/templates/pro.html
index e77248a..1b62084 100644
--- a/templates/pro.html
+++ b/templates/pro.html
@@ -33,7 +33,7 @@
       <strong>Your card was declined.</strong> You have not been charged. Please try again.
     </div>
   {% endif %}
-  <form class="form-horizontal static-box" id="pro-signup" method="post" action="/pro/complete/">
+  <form class="form-horizontal static-box" id="pro-signup" method="post" action="/pro/complete/{% if 'plan' in request.GET %}?plan={{ request.GET.plan }}{% endif %}">
     <fieldset>
       <div class="info">
         Subscribe for just <span>$5/mo</span>, <span>$49/yr</span>, or <span>$149 once</span>.
@@ -49,8 +49,8 @@
         <label class="control-label" for="name">Payment plan:</label>
         <div class="controls">
           <select name="plan" type="text" class="input-medium" id="plan">
-            <option value="monthly">$5/month</option>
-            <option value="yearly">$49/year</option>
+            <option value="snipt-monthly">$5/month</option>
+            <option value="snipt-yearly">$49/year</option>
             <option value="onetime">$149 once</option>
           </select>
         </div>
diff --git a/views.py b/views.py
index cd9dd02..3c09fe4 100644
--- a/views.py
+++ b/views.py
@@ -119,10 +119,10 @@ def pro_complete(request):
         # Recurring plans.
         else:
 
-            if request.POST['plan'] == 'monthly':
-                plan = 'snipt-monthly'
-            elif request.POST['plan'] == 'yearly':
-                plan = 'snipt-yearly'
+            if 'plan' in request.GET:
+                plan = request.GET['plan']
+            else:
+                plan = request.POST['plan']
 
             try:
                 customer = stripe.Customer.create(card=token,