quay/data/billing.py

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,
  },
]


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 = {}

    @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)