Merge branch 'Ps' into 'master'

update 修改企业认证状态接口、企业搜索接口、企业主页接口

See merge request root/tfse_admin!2

TestManage/TestObj.py

@@ -27,7 +27,16 @@ class TestObj:
         """
         企业认证
         Returns:
-            info: 信息
+            info: 认证结果
         """
         info = company_verify_script(self.company, self.code, self.legal_person, self.cid)
         return info
+
+    def change_verified(self):
+        """
+        修改企业认证状态
+        Returns:
+            info: 修改结果
+        """
+        info = change_verified_script(self.cid)
+        return info

TestManage/db.py

@@ -13,7 +13,9 @@ def update_user_info_by_cid(cid, data):
         info: 信息
     """
     collection = client['用户']['用户信息']
-    collection.update_one({"企业ID": cid}, {"$set": data})
+    res = collection.update_one({"企业ID": cid}, {"$set": data})
+    info = res.raw_result['updatedExisting']
+    return info
 
 
 def find_user_info_by_cid(cid):

TestManage/routes.py

@@ -56,3 +56,21 @@ def enterprise_certification():
     info = test_obj.company_verify()
     status = 200 if info == '认证成功' else 210
     return {"info": info}, status
+
+
+@test_route.route('/change_verified', methods=['GET'])
+@api_secret
+def change_verified():
+    """
+    企业认证状态修改
+    Parameters:
+        -
+    Returns:
+        info: 修改结果结果
+    """
+    test_obj = TestObj()
+    test_obj.cid = request.args.get('cid')
+    info = test_obj.change_verified()
+    status = 200 if info else 210
+    res = '认证状态修改成功' if info else '认证状态修改失败'
+    return {"info": res}, status

TestManage/scripts.py

@@ -107,3 +107,23 @@ def company_verify_script(company, code, legal_person, cid):
             return '认证失败'
     except Exception:
         return '认证失败'
+
+
+def change_verified_script(cid):
+    """
+    将企业认证状态改为否
+    Parameters:
+        cid: 企业ID
+    Returns:
+        res: True 执行结果
+    """
+    # 更新用户认证状态
+    insert_dict = dict()
+    insert_dict['企业名称'] = None
+    insert_dict['统一社会信用代码'] = None
+    insert_dict['法人姓名'] = None
+    insert_dict['法人身份证'] = None
+    insert_dict['营业执照fid'] = None
+    insert_dict['身份证fid'] = None
+    info = update_user_info_by_cid(cid, {"企业名称": None, "已认证": "否", "认证信息": insert_dict})
+    return info

app.py

@@ -2,6 +2,7 @@ from flask import Flask
 from flask_cors import *
 
 from TestManage.routes import test_route
+from company.routes import company_route
 
 app = Flask(__name__)
 CORS(app, supports_credentials=True)
@@ -9,6 +10,7 @@ app.config['JSON_SORT_KEYS'] = False
 
 
 app.register_blueprint(test_route, url_prefix='/admin/test')
+app.register_blueprint(company_route, url_prefix='/admin/company')
 
 if __name__ == '__main__':
     app.run()

company/Company.py

@@ -0,0 +1,27 @@
+from company.scripts import *
+
+
+class Company:
+
+    def __init__(self):
+        self.rid = None
+        self.cid = None
+        self.query = None
+
+    def search_company(self):
+        """
+        企业搜索
+        Returns:
+            result: 查询结果
+        """
+        result = search_company_script(self.query)
+        return result
+
+    def company_index(self):
+        """
+        企业主页
+        Returns:
+            result: 企业信息查询结果
+        """
+        result = company_index_script(self.cid)
+        return result

company/db.py

@@ -0,0 +1,64 @@
+import re
+
+import pymongo
+
+client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
+
+
+def find_company_records(query):
+    """
+    查询企业信息
+    Parameters:
+        query: 查询条件
+    Returns:
+        record: 查询结果
+    """
+    collection = client['用户']['用户信息']
+    data = collection.find({"企业ID": query}, {'企业ID': True, '企业名称': True, '_id': False})
+    record = list(data)
+    if len(record) == 0:
+        data = collection.find({"企业名称": re.compile(query)}, {'企业ID': True, '企业名称': True, '_id': False})
+        record = list(data)
+    return record
+
+
+def find_user_info(cid):
+    """
+    查询用户信息
+    Parameters:
+        cid: 企业ID
+    Returns:
+        record: 查询结果
+    """
+    collection = client['用户']['用户信息']
+    data = collection.find({"企业ID": cid}, {'企业ID': True, '企业名称': True, '已认证': True, '_id': False})
+    record = list(data)
+    return record
+
+
+def find_company_basic_info(cid):
+    """
+    查询公司基本信息
+    Parameters:
+        cid: 企业ID
+    Returns:
+        record: 查询结果
+    """
+    collection = client['企业']['公司基本信息']
+    data = collection.find({"企业ID": cid}, {'_id': False})
+    record = list(data)
+    return record
+
+
+def find_company_evaluate_info(cid):
+    """
+    查询评价记录
+    Parameters:
+        cid: 企业ID
+    Returns:
+        record: 查询结果
+    """
+    collection = client['评价']['评价记录']
+    data = collection.find({"企业ID": cid}, {'_id': False, '企业ID': False})
+    record = list(data)
+    return record

company/routes.py

@@ -0,0 +1,40 @@
+from flask import Blueprint, request
+
+from common.scripts import api_secret
+from company.Company import Company
+
+company_route = Blueprint('company', __name__)
+
+
+@company_route.route('/search', methods=['POST'])
+@api_secret
+def search_company():
+    """
+    企业搜索
+    Parameters:
+        -
+    Returns:
+        info: 返回信息
+        result: 返回数据
+    """
+    company = Company()
+    company.query = request.json['query']
+    data = company.search_company()
+    return {"info": '查询成功', "result": data}, 200
+
+
+@company_route.route('/index', methods=['POST'])
+@api_secret
+def company_index():
+    """
+    企业主页
+    Parameters:
+        -
+    Returns:
+        info: 返回信息
+        result: 返回数据
+    """
+    company = Company()
+    company.cid = request.json['cid']
+    data = company.company_index()
+    return {"info": '企业信息查询成功', "result": data}, 200

company/scripts.py

@@ -0,0 +1,40 @@
+from company.db import *
+
+
+def search_company_script(query):
+    """
+    查询企业信息
+    Parameters:
+        query: 查询条件（企业名称/企业id）
+    Returns:
+        result: 返回数据
+    """
+    result = find_company_records(query)
+    return result
+
+
+def company_index_script(cid):
+    """
+    查询企业相关信息
+    Parameters:
+        cid: 企业ID
+    Returns:
+        result: 返回数据
+    """
+    user_info = find_user_info(cid)
+    basic_info = find_company_basic_info(cid)
+    evaluate_info = find_company_evaluate_info(cid)
+    result = dict()
+    if len(user_info) > 0:
+        result['企业名称'] = user_info[0]['企业名称']
+        result['企业ID'] = cid
+        result['已认证'] = user_info[0]['已认证']
+    if len(basic_info) > 0:
+        result['基本信息'] = dict()
+        result['基本信息']['工商信息'] = basic_info[0]['工商信息']
+        result['基本信息']['股东信息'] = basic_info[0]['股东信息']
+        result['基本信息']['主要成员'] = basic_info[0]['主要成员']
+    if len(evaluate_info) > 0:
+        result['评价记录'] = list()
+        result['评价记录'] = evaluate_info
+    return result