Merge branch 'ps' into 'master'

Ps

See merge request root/tfse_admin!8

rating/Rating.py

@@ -6,6 +6,7 @@ class Rating:
     def __init__(self):
         self.query = None
         self.rid = None
+        self.industry = None
 
     def search_rating(self, skip, limit):
         """
@@ -76,3 +77,7 @@ class Rating:
         """
         result = report_create_script(self.rid)
         return result
+
+    def upload_macroscopic_report(self, file):
+        result = upload_macroscopic_report_script(self.industry, file)
+        return result

rating/db.py

@@ -144,5 +144,76 @@ def find_report_data(rid):
     return record
 
 
+def find_macroscopic_report_data(industry):
+    """
+    查询宏观报告数据
+    Parameters:
+        industry: 行业名称
+    Returns:
+        record: 报告数据
+    """
+    collection = client['宏观']['行业数据']
+    data = collection.find({"行业分类": industry}, {'_id': False})
+    record = list(data)
+    return record
+
+
+def insert_macroscopic_report_data(insert_data):
+    """
+    插入宏观报告数据
+    Parameters:
+        insert_data: 需要保存的数据
+    Returns:
+        result: 保存结果
+    """
+    collection = client['宏观']['行业数据']
+    res = collection.insert_one(insert_data)
+    info = res.acknowledged
+    return info
+
+
+def update_macroscopic_report_data(industry, fid):
+    """
+    更新宏观报告数据id
+    Parameters:
+        industry: 行业分类
+        fid: 文件id
+    Returns:
+        result: 更新结果
+    """
+    collection = client['宏观']['行业数据']
+    res = collection.update_one({"行业分类": industry}, {"$set": {"报告fid": fid}})
+    info = res.raw_result['updatedExisting']
+    return info
+
+
+def upload_report(file_name, files):
+    """
+       根据名称保存该企业报告
+    """
+    try:
+        database = '宏观'
+        collection = '宏观报告PDF'
+        fs = GridFS(client[database], collection)
+        pdf_id = fs.put(data=files.read(), content_type='application/pdf', filename=file_name)
+        return str(pdf_id)
+    except Exception:
+        return False
+
+
+def delete_macroscopic_pdf(fid):
+    """
+    根据id删除宏观报告
+    Parameters:
+        fid: 评价ID
+    Returns:
+        record: 删除结果
+    """
+    database = '模型'
+    collection = '宏观报告PDF'
+    fs = GridFS(client[database], collection)
+    fs.delete(ObjectId(fid))
+
+
 if __name__ == '__main__':
     delete_report_pdf('61de994660a518e91fbd0192')

rating/routes.py

@@ -130,3 +130,28 @@ def report_edit_save_delete():
         return {"info": "键值错误"}, 400
     except TypeError:
         return {"info": "参数错误"}, 400
+
+
+@rating_route.route('/upload_macroscopic_report', methods=['POST'])
+@api_secret
+def upload_report():
+    """
+    根据行业名称上传行业宏观报告
+    Parameters:
+        -
+    Returns:
+        info: 上传结果
+    """
+    try:
+        rating = Rating()
+        rating.industry = request.form['industry']
+        file = request.files['file']
+        result = rating.upload_macroscopic_report(file)
+        if result:
+            return {"info": "上传成功", "result": {"报告fid": result}}, 200
+        else:
+            return {"info": "上传失败"}, 210
+    except KeyError:
+        return {"info": "键值错误"}, 400
+    except TypeError:
+        return {"info": "参数错误"}, 400

rating/scripts.py

@@ -1,3 +1,4 @@
+import datetime
 import json
 
 import requests
@@ -124,3 +125,49 @@ def report_create_script(rid):
     if result['info'] == '生成报告成功':
         update_evaluation_records(rid, {"报告fid": result['result']['FileID']})
     return result
+
+
+def upload_macroscopic_report_script(industry, file):
+    """
+    根据行业保存PDF宏观报告
+    Parameters:
+        industry: 行业名称
+        file: pdf文件
+    Returns:
+        result: 生成结果
+    """
+
+    def save_report():
+        """
+        保存PDF宏观报告
+        Parameters:
+            -
+        Returns:
+            result: 生成结果
+        """
+        file_name = '{}_{}'.format(industry, datetime.datetime.now().strftime('%Y%m'))
+        file_id = upload_report(file_name, file)
+        return file_id
+
+    # 保存新宏观报告pdf文件
+    new_fid = save_report()
+    # 根据行业名称查询此行业是否存在宏观报告
+    macrosc = find_macroscopic_report_data(industry)
+    if macrosc:
+        # 使用新fid更新此条数据
+        upload_res = update_macroscopic_report_data(industry, new_fid)
+        if upload_res:
+            return new_fid
+        else:
+            return False
+    else:
+        # 不存在此条数据，就插入一条新数据
+        insert_data = dict()
+        insert_data['行业分类'] = industry
+        insert_data['报告fid'] = new_fid
+        save_res = insert_macroscopic_report_data(insert_data)
+        if save_res:
+            return new_fid
+        else:
+            return False
+