feat: utm supports. (#2181)

api/controllers/console/__init__.py

@@ -11,10 +11,13 @@ from .app import (advanced_prompt_template, annotation, app, audio, completion,
                   model_config, site, statistic)
 # Import auth controllers
 from .auth import activate, data_source_oauth, login, oauth
-from .billing import billing
 # Import datasets controllers
 from .datasets import data_source, datasets, datasets_document, datasets_segments, file, hit_testing
 # Import explore controllers
 from .explore import audio, completion, conversation, installed_app, message, parameter, recommended_app, saved_message
 # Import workspace controllers
 from .workspace import account, members, model_providers, models, tool_providers, workspace
+# Import billing controllers
+from .billing import billing
+# Import operation controllers
+from .operation import operation

api/controllers/console/operation/operation.py

@@ -0,0 +1,30 @@
+from flask_login import current_user
+from flask_restful import Resource, reqparse
+
+from controllers.console import api
+from controllers.console.setup import setup_required
+from controllers.console.wraps import account_initialization_required, only_edition_cloud
+from libs.login import login_required
+from services.operation_service import OperationService
+
+
+class TenantUtm(Resource):
+
+    @setup_required
+    @login_required
+    @account_initialization_required
+    @only_edition_cloud
+    def post(self):
+
+        parser = reqparse.RequestParser()
+        parser.add_argument('utm_source', type=str, required=True)
+        parser.add_argument('utm_medium', type=str, required=True)
+        parser.add_argument('utm_campaign', type=str, required=False, default='')
+        parser.add_argument('utm_content', type=str, required=False, default='')
+        parser.add_argument('utm_term', type=str, required=False, default='')
+        args = parser.parse_args()
+
+        return OperationService.record_utm(current_user.current_tenant_id, args)
+
+
+api.add_resource(TenantUtm, '/operation/utm')