mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-04-22 05:39:42 +08:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from flask import request
|
|
from flask_login import current_user # type: ignore
|
|
from flask_restful import Resource, reqparse # type: ignore
|
|
|
|
from libs.helper import extract_remote_ip
|
|
from libs.login import login_required
|
|
from services.billing_service import BillingService
|
|
|
|
from .. import api
|
|
from ..wraps import account_initialization_required, only_edition_cloud, setup_required
|
|
|
|
|
|
class ComplianceApi(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@only_edition_cloud
|
|
def get(self):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("doc_name", type=str, required=True, location="args")
|
|
args = parser.parse_args()
|
|
|
|
ip_address = extract_remote_ip(request)
|
|
device_info = request.headers.get("User-Agent", "Unknown device")
|
|
|
|
return BillingService.get_compliance_download_link(
|
|
doc_name=args.doc_name,
|
|
account_id=current_user.id,
|
|
tenant_id=current_user.current_tenant_id,
|
|
ip=ip_address,
|
|
device_info=device_info,
|
|
)
|
|
|
|
|
|
api.add_resource(ComplianceApi, "/compliance/download")
|