mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
485 lines
13 KiB
Python
485 lines
13 KiB
Python
from typing import Any, Dict
|
|
import stripe
|
|
|
|
from datetime import datetime, timedelta
|
|
from calendar import timegm
|
|
|
|
from util.morecollections import AttrDict
|
|
|
|
PLANS = [
|
|
# Deprecated Plans (2013-2014)
|
|
{
|
|
"title": "Micro",
|
|
"price": 700,
|
|
"privateRepos": 5,
|
|
"stripeId": "micro",
|
|
"audience": "For smaller teams",
|
|
"bus_features": False,
|
|
"deprecated": True,
|
|
"free_trial_days": 14,
|
|
"superseded_by": "personal-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Basic",
|
|
"price": 1200,
|
|
"privateRepos": 10,
|
|
"stripeId": "small",
|
|
"audience": "For your basic team",
|
|
"bus_features": False,
|
|
"deprecated": True,
|
|
"free_trial_days": 14,
|
|
"superseded_by": "bus-micro-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Yacht",
|
|
"price": 5000,
|
|
"privateRepos": 20,
|
|
"stripeId": "bus-coreos-trial",
|
|
"audience": "For small businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 180,
|
|
"superseded_by": "bus-small-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Personal",
|
|
"price": 1200,
|
|
"privateRepos": 5,
|
|
"stripeId": "personal",
|
|
"audience": "Individuals",
|
|
"bus_features": False,
|
|
"deprecated": True,
|
|
"free_trial_days": 14,
|
|
"superseded_by": "personal-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Skiff",
|
|
"price": 2500,
|
|
"privateRepos": 10,
|
|
"stripeId": "bus-micro",
|
|
"audience": "For startups",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 14,
|
|
"superseded_by": "bus-micro-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Yacht",
|
|
"price": 5000,
|
|
"privateRepos": 20,
|
|
"stripeId": "bus-small",
|
|
"audience": "For small businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 14,
|
|
"superseded_by": "bus-small-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Freighter",
|
|
"price": 10000,
|
|
"privateRepos": 50,
|
|
"stripeId": "bus-medium",
|
|
"audience": "For normal businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 14,
|
|
"superseded_by": "bus-medium-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Tanker",
|
|
"price": 20000,
|
|
"privateRepos": 125,
|
|
"stripeId": "bus-large",
|
|
"audience": "For large businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 14,
|
|
"superseded_by": "bus-large-30",
|
|
"plans_page_hidden": False,
|
|
},
|
|
# Deprecated plans (2014-2017)
|
|
{
|
|
"title": "Personal",
|
|
"price": 1200,
|
|
"privateRepos": 5,
|
|
"stripeId": "personal-30",
|
|
"audience": "Individuals",
|
|
"bus_features": False,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "personal-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Skiff",
|
|
"price": 2500,
|
|
"privateRepos": 10,
|
|
"stripeId": "bus-micro-30",
|
|
"audience": "For startups",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "bus-micro-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Yacht",
|
|
"price": 5000,
|
|
"privateRepos": 20,
|
|
"stripeId": "bus-small-30",
|
|
"audience": "For small businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "bus-small-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Freighter",
|
|
"price": 10000,
|
|
"privateRepos": 50,
|
|
"stripeId": "bus-medium-30",
|
|
"audience": "For normal businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "bus-medium-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Tanker",
|
|
"price": 20000,
|
|
"privateRepos": 125,
|
|
"stripeId": "bus-large-30",
|
|
"audience": "For large businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "bus-large-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Carrier",
|
|
"price": 35000,
|
|
"privateRepos": 250,
|
|
"stripeId": "bus-xlarge-30",
|
|
"audience": "For extra large businesses",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "bus-xlarge-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Huge",
|
|
"price": 65000,
|
|
"privateRepos": 500,
|
|
"stripeId": "bus-500-30",
|
|
"audience": "For huge business",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "bus-500-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Huuge",
|
|
"price": 120000,
|
|
"privateRepos": 1000,
|
|
"stripeId": "bus-1000-30",
|
|
"audience": "For the SaaS savvy enterprise",
|
|
"bus_features": True,
|
|
"deprecated": True,
|
|
"free_trial_days": 30,
|
|
"superseded_by": "bus-1000-2018",
|
|
"plans_page_hidden": False,
|
|
},
|
|
# Active plans (as of Dec 2017)
|
|
{
|
|
"title": "Open Source",
|
|
"price": 0,
|
|
"privateRepos": 0,
|
|
"stripeId": "free",
|
|
"audience": "Committment to FOSS",
|
|
"bus_features": False,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Developer",
|
|
"price": 1500,
|
|
"privateRepos": 5,
|
|
"stripeId": "personal-2018",
|
|
"audience": "Individuals",
|
|
"bus_features": False,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Micro",
|
|
"price": 3000,
|
|
"privateRepos": 10,
|
|
"stripeId": "bus-micro-2018",
|
|
"audience": "For startups",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Small",
|
|
"price": 6000,
|
|
"privateRepos": 20,
|
|
"stripeId": "bus-small-2018",
|
|
"audience": "For small businesses",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Medium",
|
|
"price": 12500,
|
|
"privateRepos": 50,
|
|
"stripeId": "bus-medium-2018",
|
|
"audience": "For normal businesses",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Large",
|
|
"price": 25000,
|
|
"privateRepos": 125,
|
|
"stripeId": "bus-large-2018",
|
|
"audience": "For large businesses",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "Extra Large",
|
|
"price": 45000,
|
|
"privateRepos": 250,
|
|
"stripeId": "bus-xlarge-2018",
|
|
"audience": "For extra large businesses",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "XXL",
|
|
"price": 85000,
|
|
"privateRepos": 500,
|
|
"stripeId": "bus-500-2018",
|
|
"audience": "For huge business",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "XXXL",
|
|
"price": 160000,
|
|
"privateRepos": 1000,
|
|
"stripeId": "bus-1000-2018",
|
|
"audience": "For the SaaS savvy enterprise",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "XXXXL",
|
|
"price": 310000,
|
|
"privateRepos": 2000,
|
|
"stripeId": "bus-2000-2018",
|
|
"audience": "For the SaaS savvy big enterprise",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
{
|
|
"title": "XXXXXL",
|
|
"price": 2170000,
|
|
"privateRepos": 15000,
|
|
"stripeId": "price_1LRztA2OoNF1TIf0SvSrz106",
|
|
"audience": "For the SaaS savvy very big enterprise",
|
|
"bus_features": True,
|
|
"deprecated": False,
|
|
"free_trial_days": 30,
|
|
"superseded_by": None,
|
|
"plans_page_hidden": False,
|
|
},
|
|
]
|
|
|
|
|
|
def get_plan(plan_id):
|
|
"""
|
|
Returns the plan with the given ID or None if none.
|
|
"""
|
|
for plan in PLANS:
|
|
if plan["stripeId"] == plan_id:
|
|
return plan
|
|
|
|
return None
|
|
|
|
|
|
class FakeSubscription(AttrDict):
|
|
@classmethod
|
|
def build(cls, data, customer):
|
|
data = AttrDict.deep_copy(data)
|
|
data["customer"] = customer
|
|
return cls(data)
|
|
|
|
def delete(self):
|
|
self.customer.subscription = None
|
|
|
|
|
|
class FakeStripe(object):
|
|
class Customer(AttrDict):
|
|
FAKE_PLAN = AttrDict(
|
|
{
|
|
"id": "bus-small",
|
|
}
|
|
)
|
|
|
|
FAKE_SUBSCRIPTION = AttrDict(
|
|
{
|
|
"plan": FAKE_PLAN,
|
|
"current_period_start": timegm(datetime.utcnow().utctimetuple()),
|
|
"current_period_end": timegm(
|
|
(datetime.utcnow() + timedelta(days=30)).utctimetuple()
|
|
),
|
|
"trial_start": timegm(datetime.utcnow().utctimetuple()),
|
|
"trial_end": timegm((datetime.utcnow() + timedelta(days=30)).utctimetuple()),
|
|
}
|
|
)
|
|
|
|
FAKE_CARD = AttrDict(
|
|
{
|
|
"id": "card123",
|
|
"name": "Joe User",
|
|
"type": "Visa",
|
|
"last4": "4242",
|
|
"exp_month": 5,
|
|
"exp_year": 2016,
|
|
}
|
|
)
|
|
|
|
FAKE_CARD_LIST = AttrDict(
|
|
{
|
|
"data": [FAKE_CARD],
|
|
}
|
|
)
|
|
|
|
ACTIVE_CUSTOMERS: Dict[str, Any] = {}
|
|
|
|
@property
|
|
def card(self):
|
|
return self.get("new_card", None)
|
|
|
|
@card.setter
|
|
def card(self, card_token):
|
|
self["new_card"] = card_token
|
|
|
|
@property
|
|
def plan(self):
|
|
return self.get("new_plan", None)
|
|
|
|
@plan.setter
|
|
def plan(self, plan_name):
|
|
self["new_plan"] = plan_name
|
|
|
|
def save(self):
|
|
if self.get("new_card", None) is not None:
|
|
raise stripe.error.CardError(
|
|
"Test raising exception on set card.", self.get("new_card"), 402
|
|
)
|
|
if self.get("new_plan", None) is not None:
|
|
if self.subscription is None:
|
|
self.subscription = FakeSubscription.build(self.FAKE_SUBSCRIPTION, self)
|
|
self.subscription.plan.id = self.get("new_plan")
|
|
|
|
@classmethod
|
|
def retrieve(cls, stripe_customer_id):
|
|
if stripe_customer_id in cls.ACTIVE_CUSTOMERS:
|
|
cls.ACTIVE_CUSTOMERS[stripe_customer_id].pop("new_card", None)
|
|
cls.ACTIVE_CUSTOMERS[stripe_customer_id].pop("new_plan", None)
|
|
return cls.ACTIVE_CUSTOMERS[stripe_customer_id]
|
|
else:
|
|
new_customer = cls(
|
|
{
|
|
"default_card": "card123",
|
|
"cards": AttrDict.deep_copy(cls.FAKE_CARD_LIST),
|
|
"id": stripe_customer_id,
|
|
}
|
|
)
|
|
new_customer.subscription = FakeSubscription.build(
|
|
cls.FAKE_SUBSCRIPTION, new_customer
|
|
)
|
|
cls.ACTIVE_CUSTOMERS[stripe_customer_id] = new_customer
|
|
return new_customer
|
|
|
|
class Invoice(AttrDict):
|
|
@staticmethod
|
|
def list(customer, count):
|
|
return AttrDict(
|
|
{
|
|
"data": [],
|
|
}
|
|
)
|
|
|
|
|
|
class Billing(object):
|
|
def __init__(self, app=None):
|
|
self.app = app
|
|
if app is not None:
|
|
self.state = self.init_app(app)
|
|
else:
|
|
self.state = None
|
|
|
|
def init_app(self, app):
|
|
billing_type = app.config.get("BILLING_TYPE", "FakeStripe")
|
|
|
|
if billing_type == "Stripe":
|
|
billing = stripe
|
|
stripe.api_key = app.config.get("STRIPE_SECRET_KEY", None)
|
|
|
|
elif billing_type == "FakeStripe":
|
|
billing = FakeStripe
|
|
|
|
else:
|
|
raise RuntimeError("Unknown billing type: %s" % billing_type)
|
|
|
|
# register extension with app
|
|
app.extensions = getattr(app, "extensions", {})
|
|
app.extensions["billing"] = billing
|
|
return billing
|
|
|
|
def __getattr__(self, name):
|
|
return getattr(self.state, name, None)
|