from functools import wraps

import stripe
from flask import redirect, url_for, flash
from flask_login import current_user
from flask_babel import gettext as _


def subscription_required(f):
    """
    Ensure a user is subscribed, if not redirect them to the pricing table.

    :return: Function
    """
    @wraps(f)
    def decorated_function(*args, **kwargs):
        subscription = current_user.subscription
        if current_user.partnership_or_partnership_account.billing_type in ['invoice', 'partnership']:
            subscription = current_user.subscription
        if not subscription or subscription.status != 'active':
            flash(_('Your subscription has not been set up'), 'warning')
            return redirect(url_for('billing.pricing'))

        return f(*args, **kwargs)

    return decorated_function


def handle_stripe_exceptions(f):
    """
    Handle stripe exceptions so they do not throw 500s.

    :param f:
    :type f: Function
    :return: Function
    """
    @wraps(f)
    def decorated_function(*args, **kwargs):
        try:
            return f(*args, **kwargs)
        except stripe.error.CardError:
            flash(_('Sorry, your card was declined. Try again perhaps?'),
                  'danger')
            return redirect(url_for('user.settings'))
        except stripe.error.InvalidRequestError as e:
            flash(e, 'danger')
            return redirect(url_for('user.settings'))
        except stripe.error.AuthenticationError:
            flash(_('Authentication with our payment gateway failed.'),
                  'danger')
            return redirect(url_for('user.settings'))
        except stripe.error.APIConnectionError:
            flash(_(
                'Our payment gateway is experiencing connectivity issues'
                ', please try again.'), 'danger')
            return redirect(url_for('user.settings'))
        except stripe.error.StripeError:
            flash(_(
                'Our payment gateway is having issues, please try again.'),
                'danger')
            return redirect(url_for('user.settings'))

    return decorated_function