From 4ed1344d51acfd6d9c21f81bc14daea6ba285c2b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=80=9D=E5=B7=9D?= <sichanwong0527@gmail.com>
Date: Fri, 17 Dec 2021 17:15:20 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=BC=E5=90=88=E8=AF=84=E4=BB=B7=E5=88=86?=
 =?UTF-8?q?=E6=9E=90=20=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 company/CompanyIndex.py | 10 +++++++++-
 company/db.py           | 14 ++++++++++++++
 company/routes.py       | 16 ++++++++++++++++
 company/scripts.py      |  7 ++++++-
 4 files changed, 45 insertions(+), 2 deletions(-)

diff --git a/company/CompanyIndex.py b/company/CompanyIndex.py
index 8462b31..975f073 100644
--- a/company/CompanyIndex.py
+++ b/company/CompanyIndex.py
@@ -1,4 +1,4 @@
-from company.scripts import basic_info_script
+from company.scripts import basic_info_script, general_rating_script
 
 
 class CompanyIndex:
@@ -12,3 +12,11 @@ class CompanyIndex:
             return "查询成功", result[0]
         else:
             return "无数据", {}
+
+    def general_rating_obj(self):
+        result = general_rating_script(self.cid)
+        if result:
+            return "查询成功", result[0]
+        else:
+            return "无数据", {}
+
diff --git a/company/db.py b/company/db.py
index 3918941..1862e8a 100644
--- a/company/db.py
+++ b/company/db.py
@@ -15,3 +15,17 @@ def find_basic_info(param):
     data = collection.find(param, {'_id': False})
     record = list(data)
     return record
+
+
+def find_general_rating_info(param):
+    """
+    Notes
+    Parameters:
+        param: desc
+    Returns:
+        res: desc
+    """
+    collection = client['企业']['综合评价分析']
+    data = collection.find(param, {'_id': False})
+    record = list(data)
+    return record
diff --git a/company/routes.py b/company/routes.py
index f02c6f8..b24ebde 100644
--- a/company/routes.py
+++ b/company/routes.py
@@ -20,3 +20,19 @@ def basic_info_route(**kwargs):
     company_index.cid = kwargs['cid']
     info, result = company_index.basic_info_obj()
     return {"info": info, "result": result}
+
+
+@company_route.route('/general_rating', methods=['GET'])
+@verify_token
+def general_rating_route(**kwargs):
+    """
+
+    Parameters:
+        param desc
+    Returns:
+        res: desc
+    """
+    company_index = CompanyIndex()
+    company_index.cid = kwargs['cid']
+    info, result = company_index.general_rating_obj()
+    return {"info": info, "result": result}
diff --git a/company/scripts.py b/company/scripts.py
index cc36042..1bbebf5 100644
--- a/company/scripts.py
+++ b/company/scripts.py
@@ -1,6 +1,11 @@
-from company.db import find_basic_info
+from company.db import find_basic_info, find_general_rating_info
 
 
 def basic_info_script(cid):
     data = find_basic_info({"企业ID": cid})
     return data
+
+
+def general_rating_script(cid):
+    data = find_general_rating_info({"企业ID": cid})
+    return data