remove stripe and anthropic. (#1746)
Garfield Dai
2 years ago
committed by
GitHub
parent
4c639961f5
commit
6b499b9a16
@ -1,61 +0,0 @@
|
||||
import logging
|
||||
|
||||
import stripe
|
||||
from flask import request, current_app
|
||||
from flask_restful import Resource
|
||||
|
||||
from controllers.console import api
|
||||
from controllers.console.setup import setup_required
|
||||
from controllers.console.wraps import only_edition_cloud
|
||||
from services.provider_checkout_service import ProviderCheckoutService
|
||||
|
||||
|
||||
class StripeWebhookApi(Resource):
|
||||
@setup_required
|
||||
@only_edition_cloud
|
||||
def post(self):
|
||||
payload = request.data
|
||||
sig_header = request.headers.get('STRIPE_SIGNATURE')
|
||||
webhook_secret = current_app.config.get('STRIPE_WEBHOOK_SECRET')
|
||||
|
||||
try:
|
||||
event = stripe.Webhook.construct_event(
|
||||
payload, sig_header, webhook_secret
|
||||
)
|
||||
except ValueError as e:
|
||||
# Invalid payload
|
||||
return 'Invalid payload', 400
|
||||
except stripe.error.SignatureVerificationError as e:
|
||||
# Invalid signature
|
||||
return 'Invalid signature', 400
|
||||
|
||||
# Handle the checkout.session.completed event
|
||||
if event['type'] == 'checkout.session.completed':
|
||||
logging.debug(event['data']['object']['id'])
|
||||
logging.debug(event['data']['object']['amount_subtotal'])
|
||||
logging.debug(event['data']['object']['currency'])
|
||||
logging.debug(event['data']['object']['payment_intent'])
|
||||
logging.debug(event['data']['object']['payment_status'])
|
||||
logging.debug(event['data']['object']['metadata'])
|
||||
|
||||
session = stripe.checkout.Session.retrieve(
|
||||
event['data']['object']['id'],
|
||||
expand=['line_items'],
|
||||
)
|
||||
|
||||
logging.debug(session.line_items['data'][0]['quantity'])
|
||||
|
||||
# Fulfill the purchase...
|
||||
provider_checkout_service = ProviderCheckoutService()
|
||||
|
||||
try:
|
||||
provider_checkout_service.fulfill_provider_order(event, session.line_items)
|
||||
except Exception as e:
|
||||
|
||||
logging.debug(str(e))
|
||||
return 'success', 200
|
||||
|
||||
return 'success', 200
|
||||
|
||||
|
||||
api.add_resource(StripeWebhookApi, '/webhook/stripe')
|
||||
@ -1,6 +0,0 @@
|
||||
import stripe
|
||||
|
||||
|
||||
def init_app(app):
|
||||
if app.config.get('STRIPE_API_KEY'):
|
||||
stripe.api_key = app.config.get('STRIPE_API_KEY')
|
||||
@ -1,174 +0,0 @@
|
||||
import datetime
|
||||
import logging
|
||||
|
||||
import stripe
|
||||
from flask import current_app
|
||||
|
||||
from core.model_providers.model_provider_factory import ModelProviderFactory
|
||||
from extensions.ext_database import db
|
||||
from models.account import Account
|
||||
from models.provider import ProviderOrder, ProviderOrderPaymentStatus, ProviderType, Provider, ProviderQuotaType
|
||||
|
||||
|
||||
class ProviderCheckout:
|
||||
def __init__(self, stripe_checkout_session):
|
||||
self.stripe_checkout_session = stripe_checkout_session
|
||||
|
||||
def get_checkout_url(self):
|
||||
return self.stripe_checkout_session.url
|
||||
|
||||
|
||||
class ProviderCheckoutService:
|
||||
def create_checkout(self, tenant_id: str, provider_name: str, account: Account) -> ProviderCheckout:
|
||||
# check provider name is valid
|
||||
model_provider_rules = ModelProviderFactory.get_provider_rules()
|
||||
if provider_name not in model_provider_rules:
|
||||
raise ValueError(f'provider name {provider_name} is invalid')
|
||||
|
||||
model_provider_rule = model_provider_rules[provider_name]
|
||||
|
||||
# check provider name can be paid
|
||||
self._check_provider_payable(provider_name, model_provider_rule)
|
||||
|
||||
# get stripe checkout product id
|
||||
paid_provider = self._get_paid_provider(tenant_id, provider_name)
|
||||
model_provider_class = ModelProviderFactory.get_model_provider_class(provider_name)
|
||||
model_provider = model_provider_class(provider=paid_provider)
|
||||
payment_info = model_provider.get_payment_info()
|
||||
if not payment_info:
|
||||
raise ValueError(f'provider name {provider_name} not support payment')
|
||||
|
||||
payment_product_id = payment_info['product_id']
|
||||
payment_min_quantity = payment_info['min_quantity']
|
||||
payment_max_quantity = payment_info['max_quantity']
|
||||
|
||||
# create provider order
|
||||
provider_order = ProviderOrder(
|
||||
tenant_id=tenant_id,
|
||||
provider_name=provider_name,
|
||||
account_id=account.id,
|
||||
payment_product_id=payment_product_id,
|
||||
quantity=1,
|
||||
payment_status=ProviderOrderPaymentStatus.WAIT_PAY.value
|
||||
)
|
||||
|
||||
db.session.add(provider_order)
|
||||
db.session.flush()
|
||||
|
||||
line_item = {
|
||||
'price': f'{payment_product_id}',
|
||||
'quantity': payment_min_quantity
|
||||
}
|
||||
|
||||
if payment_min_quantity > 1 and payment_max_quantity != payment_min_quantity:
|
||||
line_item['adjustable_quantity'] = {
|
||||
'enabled': True,
|
||||
'minimum': payment_min_quantity,
|
||||
'maximum': payment_max_quantity
|
||||
}
|
||||
|
||||
try:
|
||||
# create stripe checkout session
|
||||
checkout_session = stripe.checkout.Session.create(
|
||||
line_items=[
|
||||
line_item
|
||||
],
|
||||
mode='payment',
|
||||
success_url=current_app.config.get("CONSOLE_WEB_URL")
|
||||
+ f'?provider_name={provider_name}&payment_result=succeeded',
|
||||
cancel_url=current_app.config.get("CONSOLE_WEB_URL")
|
||||
+ f'?provider_name={provider_name}&payment_result=cancelled',
|
||||
automatic_tax={'enabled': True},
|
||||
)
|
||||
except Exception as e:
|
||||
logging.exception(e)
|
||||
raise ValueError(f'provider name {provider_name} create checkout session failed, please try again later')
|
||||
|
||||
provider_order.payment_id = checkout_session.id
|
||||
db.session.commit()
|
||||
|
||||
return ProviderCheckout(checkout_session)
|
||||
|
||||
def fulfill_provider_order(self, event, line_items):
|
||||
provider_order = db.session.query(ProviderOrder) \
|
||||
.filter(ProviderOrder.payment_id == event['data']['object']['id']) \
|
||||
.first()
|
||||
|
||||
if not provider_order:
|
||||
raise ValueError(f'provider order not found, payment id: {event["data"]["object"]["id"]}')
|
||||
|
||||
if provider_order.payment_status != ProviderOrderPaymentStatus.WAIT_PAY.value:
|
||||
raise ValueError(
|
||||
f'provider order payment status is not wait pay, payment id: {event["data"]["object"]["id"]}')
|
||||
|
||||
provider_order.transaction_id = event['data']['object']['payment_intent']
|
||||
provider_order.currency = event['data']['object']['currency']
|
||||
provider_order.total_amount = event['data']['object']['amount_subtotal']
|
||||
provider_order.payment_status = ProviderOrderPaymentStatus.PAID.value
|
||||
provider_order.paid_at = datetime.datetime.utcnow()
|
||||
provider_order.updated_at = provider_order.paid_at
|
||||
|
||||
# update provider quota
|
||||
provider = db.session.query(Provider).filter(
|
||||
Provider.tenant_id == provider_order.tenant_id,
|
||||
Provider.provider_name == provider_order.provider_name,
|
||||
Provider.provider_type == ProviderType.SYSTEM.value,
|
||||
Provider.quota_type == ProviderQuotaType.PAID.value
|
||||
).first()
|
||||
|
||||
if not provider:
|
||||
raise ValueError(f'provider not found, tenant id: {provider_order.tenant_id}, '
|
||||
f'provider name: {provider_order.provider_name}')
|
||||
|
||||
model_provider_class = ModelProviderFactory.get_model_provider_class(provider_order.provider_name)
|
||||
model_provider = model_provider_class(provider=provider)
|
||||
payment_info = model_provider.get_payment_info()
|
||||
|
||||
quantity = line_items['data'][0]['quantity']
|
||||
|
||||
if not payment_info:
|
||||
increase_quota = 0
|
||||
else:
|
||||
increase_quota = int(payment_info['increase_quota']) * quantity
|
||||
|
||||
if increase_quota > 0:
|
||||
provider.quota_limit += increase_quota
|
||||
provider.is_valid = True
|
||||
|
||||
db.session.commit()
|
||||
|
||||
def _check_provider_payable(self, provider_name: str, model_provider_rule: dict):
|
||||
if ProviderType.SYSTEM.value not in model_provider_rule['support_provider_types']:
|
||||
raise ValueError(f'provider name {provider_name} not support payment')
|
||||
|
||||
if 'system_config' not in model_provider_rule:
|
||||
raise ValueError(f'provider name {provider_name} not support payment')
|
||||
|
||||
if 'supported_quota_types' not in model_provider_rule['system_config']:
|
||||
raise ValueError(f'provider name {provider_name} not support payment')
|
||||
|
||||
if 'paid' not in model_provider_rule['system_config']['supported_quota_types']:
|
||||
raise ValueError(f'provider name {provider_name} not support payment')
|
||||
|
||||
def _get_paid_provider(self, tenant_id: str, provider_name: str):
|
||||
paid_provider = db.session.query(Provider) \
|
||||
.filter(
|
||||
Provider.tenant_id == tenant_id,
|
||||
Provider.provider_name == provider_name,
|
||||
Provider.provider_type == ProviderType.SYSTEM.value,
|
||||
Provider.quota_type == ProviderQuotaType.PAID.value,
|
||||
).first()
|
||||
|
||||
if not paid_provider:
|
||||
paid_provider = Provider(
|
||||
tenant_id=tenant_id,
|
||||
provider_name=provider_name,
|
||||
provider_type=ProviderType.SYSTEM.value,
|
||||
quota_type=ProviderQuotaType.PAID.value,
|
||||
quota_limit=0,
|
||||
quota_used=0,
|
||||
)
|
||||
db.session.add(paid_provider)
|
||||
db.session.commit()
|
||||
|
||||
return paid_provider
|
||||
Loading…
Reference in New Issue