接口设置模块访问权限

Controller/RentController.py

@@ -3,13 +3,15 @@ import datetime
 import json
 
 import requests
-from flask import Blueprint, jsonify, request
+from flask import Blueprint, jsonify, request, g
+from pymongo import MongoClient
 from sqlalchemy import or_
 from sqlalchemy.exc import SQLAlchemyError
 
-from Config.base_config import ACTION_PWD
+from Config.base_config import ACTION_PWD, MONGO_DATABASE_URI
 from Libs.ErrorTips import BASE_RESPONSE, REFUND_NOT_RENT_INFO, REFUND_BACK_TIME_ERROR, REFUND_NOT_PRODUCTION_INFO, \
-    REFUND_MONEY_IS_ZERO, ACTION_CODE_ERROR, PARAMETER_ERROR, MACHINE_NOT_EXIST_ERROR
+    REFUND_MONEY_IS_ZERO, ACTION_CODE_ERROR, PARAMETER_ERROR, MACHINE_NOT_EXIST_ERROR, BASE_SUCCESS_RESPONSE, \
+    AGNET_MODULES_ERROR
 from Libs.Helper import Helper
 from Libs.Logger import logger
 from Model.Base import db
@@ -768,6 +770,17 @@ def get_machine_price():
 
 @route_rent.route('/multi_return', methods=['POST'])
 def run_multi_return():
+    agent_id = g.user.id
+    platform = g.platform
+
+    mongodatabase = MongoClient(MONGO_DATABASE_URI).get_database("suishenwan")
+    agent_modules = mongodatabase.get_collection("agent_modules")
+    result = agent_modules.find_one({"agent_id": agent_id, "platform": platform})
+    if not result:
+        return BASE_SUCCESS_RESPONSE(**AGNET_MODULES_ERROR)
+    if not result.get("module_list", {}).get("my_account", {}).get("child", {}).get("manual_return", None):
+        return BASE_SUCCESS_RESPONSE(**AGNET_MODULES_ERROR)
+
     json_data = request.get_json()
     rent_datas = json_data.get("rent_datas", [])
     back_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")