mirror of
https://gitee.com/dify_ai/dify.git
synced 2024-12-07 13:47:53 +08:00
33 lines
1019 B
Python
33 lines
1019 B
Python
|
import os
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
class OperationService:
|
||
|
|
base_url = os.environ.get('BILLING_API_URL', 'BILLING_API_URL')
|
||
|
|
secret_key = os.environ.get('BILLING_API_SECRET_KEY', 'BILLING_API_SECRET_KEY')
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _send_request(cls, method, endpoint, json=None, params=None):
|
||
|
|
headers = {
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
"Billing-Api-Secret-Key": cls.secret_key
|
||
|
|
}
|
||
|
|
|
||
|
|
url = f"{cls.base_url}{endpoint}"
|
||
|
|
response = requests.request(method, url, json=json, params=params, headers=headers)
|
||
|
|
|
||
|
|
return response.json()
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def record_utm(cls, tenant_id, args):
|
||
|
|
params = {
|
||
|
|
'tenant_id': tenant_id,
|
||
|
|
'utm_source': args['utm_source'],
|
||
|
|
'utm_medium': args['utm_medium'],
|
||
|
|
'utm_campaign': args['utm_campaign'],
|
||
|
|
'utm_content': args['utm_content'],
|
||
|
|
'utm_term': args['utm_term']
|
||
|
|
}
|
||
|
|
return cls._send_request('POST', '/tenant_utms', params=params)
|