保存综合填报数据

input/InputForm.py

@@ -1,7 +1,7 @@
 from input.scripts import *
 
 
-class Input:
+class InputForm:
 
     def __init__(self):
         self.rid = None
@@ -20,24 +20,26 @@ class Input:
 
     def start_general_rating_result(self):
         """
-        开始企业综合测评
+        开始企业综合评价
         Parameters:
             param desc
         Returns:
-            res desc
+            info 信息
+            result 综合评价填报数据
         """
         info, result = start_general_rating_by_cid(self.cid)
         return info, result
 
-    def save_input_general(self):
+    def save_input_general(self, input_form):
         """
-        Notes
+        保存综合评价填报数据
         Parameters:
-            param: desc
+            input_form: desc
         Returns:
             res: desc
         """
-        pass
+        save_general_input_form(self.cid, input_form)
+        return "保存成功"
 
     def exec_general_rating(self):
         """

input/db.py

@@ -53,3 +53,16 @@ def insert_general_input(data):
     """
     collection = client['评价']['综合评价填报']
     collection.insert_one(data)
+
+
+def update_general_input(rid, data):
+    """
+    更新综合填报数据
+    Parameters:
+        rid: 评价ID
+        data: 综合填报数据
+    Returns:
+        res: desc
+    """
+    collection = client['评价']['综合评价填报']
+    collection.update_one({"评价ID": rid}, {"$set": data})

input/routes.py

@@ -1,7 +1,7 @@
 from flask import Blueprint, request, Response
 
 from common.auth import check_block, verify_token
-from input.Input import Input
+from input.InputForm import InputForm
 
 input_route = Blueprint('input', __name__)
 
@@ -17,7 +17,7 @@ def rating_records(**kwargs):
         info 信息
         result 记录
     """
-    inp = Input()
+    inp = InputForm()
     inp.cid = kwargs['cid']
     result = inp.rating_records_result()
     status = 200 if result else 210
@@ -35,7 +35,23 @@ def start_general_rating(**kwargs):
         info 信息
         result 记录
     """
-    inp = Input()
+    input_form = InputForm()
-    inp.cid = kwargs['cid']
+    input_form.cid = kwargs['cid']
-    info, result = inp.start_general_rating_result()
+    info, result = input_form.start_general_rating_result()
     return {"info": info, "result": result}, 200
+
+
+@input_route.route('/save_general_input_form', methods=['POST'])
+@verify_token
+def save_general_input_form(**kwargs):
+    """
+    保存综合填报数据
+    Parameters:
+        cid 企业ID
+    Returns:
+        info: 信息
+    """
+    input_form = InputForm()
+    input_form.cid = kwargs['cid']
+    info = input_form.save_input_general(request.json['input_form'])
+    return {"info": info}, 200

input/scripts.py

@@ -79,22 +79,24 @@ def start_general_rating_by_cid(cid):
         return "该企业未认证", {}
 
     # 根据企业ID和当前年度查询是否存在当年的填报记录
-    inputs = find_rating_inputs({"企业ID": cid, "评价年度": get_current_year()})
+    exist_input_form = find_rating_inputs({"企业ID": cid, "评价年度": get_current_year()})
 
     # 若存在填报记录，根据评价ID查询评价记录该评价是否完成
-    if inputs:
+    if exist_input_form:
         # 已完成 本年度不能再填报
-        if find_rating_records({"评价ID": inputs[0]["评价ID"]})[0]['进行状态'] == "完成":
+        if find_rating_records({"评价ID": exist_input_form[0]["评价ID"]})[0]['进行状态'] == "完成":
             return "本年度已评价", {}
         # 未完成 返回已经填报保存的数据
         else:
-            inputs[0].pop('评价ID')
+            exist_input_form[0].pop('评价ID')
-            inputs[0].pop('企业ID')
+            exist_input_form[0].pop('企业ID')
-            return "继续填报", inputs[0]
+            return "继续填报", exist_input_form[0]
 
     # 不存在填报记录 新生成填报记录、评价记录 返回空填报记录对象
     else:
+        # 新生成填报ID
         new_rid = gen_new_rid()
+        # 生成报告期
         report_dates = make_3_report_dates()
 
         # 新填报记录
@@ -123,3 +125,16 @@ def start_general_rating_by_cid(cid):
         insert_rating_records(rating)
 
         return "开始填报", return_input
+
+
+def save_general_input_form(cid, input_form):
+    """
+    保存综合信用评价填报
+    Parameters:
+        cid: 企业ID
+        input_form: 填报表单
+    Returns:
+        res: desc
+    """
+    rid = find_rating_records({"企业ID": cid, "进行状态": "进行"})[0]['评价ID']
+    update_general_input(rid, input_form)