diff --git a/TestManage/TestObj.py b/TestManage/TestObj.py index 2171ab6..0edf0ba 100644 --- a/TestManage/TestObj.py +++ b/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 diff --git a/TestManage/db.py b/TestManage/db.py index 9550f47..7cb90f0 100644 --- a/TestManage/db.py +++ b/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): diff --git a/TestManage/routes.py b/TestManage/routes.py index 210db5a..5f83bbf 100644 --- a/TestManage/routes.py +++ b/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 diff --git a/TestManage/scripts.py b/TestManage/scripts.py index 2bcf63d..5663090 100644 --- a/TestManage/scripts.py +++ b/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 diff --git a/app.py b/app.py index 41ad445..e46abb1 100644 --- a/app.py +++ b/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() diff --git a/company/Company.py b/company/Company.py new file mode 100644 index 0000000..51b2a56 --- /dev/null +++ b/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 diff --git a/company/__init__.py b/company/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/company/db.py b/company/db.py new file mode 100644 index 0000000..d6a458d --- /dev/null +++ b/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 diff --git a/company/routes.py b/company/routes.py new file mode 100644 index 0000000..28fdb8c --- /dev/null +++ b/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 diff --git a/company/scripts.py b/company/scripts.py new file mode 100644 index 0000000..d59bbbe --- /dev/null +++ b/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