mirror of
http://112.124.100.131/huang.ze/ebiz-dify-ai.git
synced 2025-12-10 03:16:51 +08:00
remove stripe and anthropic. (#1746)
This commit is contained in:
Garfield Dai
@@ -10,7 +10,7 @@ class BillingService:
|
||||
def get_info(cls, tenant_id: str):
|
||||
params = {'tenant_id': tenant_id}
|
||||
|
||||
billing_info = cls._send_request('GET', '/info', params=params)
|
||||
billing_info = cls._send_request('GET', '/subscription/info', params=params)
|
||||
|
||||
return billing_info
|
||||
|
||||
@@ -18,16 +18,26 @@ class BillingService:
|
||||
def get_subscription(cls, plan: str,
|
||||
interval: str,
|
||||
prefilled_email: str = '',
|
||||
user_name: str = '',
|
||||
tenant_id: str = ''):
|
||||
params = {
|
||||
'plan': plan,
|
||||
'interval': interval,
|
||||
'prefilled_email': prefilled_email,
|
||||
'user_name': user_name,
|
||||
'tenant_id': tenant_id
|
||||
}
|
||||
return cls._send_request('GET', '/subscription', params=params)
|
||||
return cls._send_request('GET', '/subscription/payment-link', params=params)
|
||||
|
||||
@classmethod
|
||||
def get_model_provider_payment_link(cls,
|
||||
provider_name: str,
|
||||
tenant_id: str,
|
||||
account_id: str):
|
||||
params = {
|
||||
'provider_name': provider_name,
|
||||
'tenant_id': tenant_id,
|
||||
'account_id': account_id
|
||||
}
|
||||
return cls._send_request('GET', '/model-provider/payment-link', params=params)
|
||||
|
||||
@classmethod
|
||||
def get_invoices(cls, prefilled_email: str = ''):
|
||||
@@ -45,10 +55,3 @@ class BillingService:
|
||||
response = requests.request(method, url, json=json, params=params, headers=headers)
|
||||
|
||||
return response.json()
|
||||
|
||||
@classmethod
|
||||
def process_event(cls, event: dict):
|
||||
json = {
|
||||
"content": event,
|
||||
}
|
||||
return cls._send_request('POST', '/webhook/stripe', json=json)
|
||||
|
||||
@@ -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
|
||||
Reference in New Issue
Block a user