添加修改报告名称

app/controller/report.py

@@ -17,7 +17,7 @@ from flask import request, render_template, g
 from flask import make_response
 
 from app.api.engine import config
-from app.service.report_service import get_report_list, delete_report, get_one_report, edit_report, make_report
+from app.service.report_service import get_report_list, delete_report, get_one_report, edit_report, make_report, edit_report_name
 from app.service.template_manage_service import get_make_report_data
 from app.utils.auth import login_require
 from app.utils.format_transfer import npEncoder
@@ -64,8 +64,19 @@ class ReportHandlers(Resource):
 
     @login_require
     def put(self, id):
-        """."""
-        pass
+        """编辑报告名称."""
+        self.parser.add_argument('type', type=int, required=True, help='1为持仓报告2为诊断报告3周期报告')
+        self.parser.add_argument('id', type=int, required=True, help='报告id不能为空')
+        self.parser.add_argument('name', type=str, required=True, help='报告名称不能为空')
+        args = self.parser.parse_args()
+        args['ifa_id'] = g.ifa_id
+        data = edit_report_name(args)
+        resp = {
+            "statusCode": "0000",
+            "message": "成功",
+            "attributes": data
+        }
+        return resp
 
     @login_require
     def delete(self, id):

app/service/report_service.py

@@ -106,6 +106,37 @@ def get_one_report(args):
         return data[0]
     return False
 
+def edit_report_name(args):
+    """编辑报告"""
+    id = args.get('id')
+    type = args.get('type')
+    ifa_id = args.get('ifa_id')
+    name = args.get('name')
+    res = []
+    with TAMP_SQL(tamp_diagnose_app_engine) as tamp_diagnose_app:
+        tamp_diagnose_session = tamp_diagnose_app.session
+        if type == 1:
+            res = tamp_diagnose_session.query(HoldReport).filter(and_(
+                HoldReport.id == id,
+                HoldReport.ifa_id == ifa_id,
+            ))
+        elif type == 2:
+            res = tamp_diagnose_session.query(HoldDiagnoseReport).filter(and_(
+                HoldDiagnoseReport.id == id,
+                HoldDiagnoseReport.ifa_id == ifa_id,
+            ))
+        elif type == 3:
+            res = tamp_diagnose_session.query(PeriodicReport).filter(and_(
+                PeriodicReport.id == id,
+                PeriodicReport.ifa_id == ifa_id,
+            ))
+        else:
+            return False
+        r = res.update({
+            'name': name,
+            'update_time': datetime.datetime.now()
+        })
+    return True
 
 def edit_report(args):
     """编辑报告"""