This commit is contained in:
王思川 2022-04-01 16:57:25 +08:00
parent 0bd59ae711
commit 223c059dc2
10 changed files with 106 additions and 214 deletions

View File

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

View File

@ -1,18 +1,14 @@
from Manage.company.db import *
from DBHelper.MongoHelper import MongoHelper
from Manage.company.company_db import *
from Manage.company.company_obj import SearchCompany
def search_company_script(query, skip, limit):
"""
查询企业信息
Parameters:
query: 查询条件企业名称/企业id
skip: 默认0
limit: 每页显示条数
Returns:
result: 返回数据
"""
result = find_company_records(query, skip, limit)
return result
class SearchCompanyImpl(SearchCompany):
""""""
db = MongoHelper("tfse_v0.21")
def conditions_search(self):
""""""
def company_index_script(cid):

View File

@ -0,0 +1,65 @@
from Manage.company.company_impl import *
class Company:
def __init__(self):
self.cid = None
self.name = None
self.email = None
self.verified = None
self.industry_l1 = None
self.industry_l2 = None
def check_obj(self, **kwargs):
"""检查对象字段"""
def dict_to_save(self, **kwargs):
"""存储对象"""
def dict_to_show(self, **kwargs):
"""显示对象"""
def company_index(self):
"""
企业主页
Returns:
result: 企业信息查询结果
"""
result = company_index_script(self.cid)
return result
def company_edit(self, data):
"""
企业信息编辑
Returns:
result: 企业信息编辑结果
"""
result = company_edit_script(self.cid, data)
return result
class SearchCompany(Company):
""""""
def __init__(self, search, sort, page_size, page_no):
super().__init__()
self.search = search
self.sort = sort
self.page_size = page_size
self.page_no = page_no
def conditions_search(self):
"""条件查询"""
def check_search(self):
""""""
def check_sort(self):
""""""
def check_page_size(self):
""""""
def check_page_no(self):
""""""

View File

@ -1,5 +1,5 @@
from flask import Blueprint, request
from Manage.company.Company import Company
from Manage.company.company_obj import Company
from user.user_auth import verify_token
company_route = Blueprint('company', __name__)
@ -16,12 +16,8 @@ def search_company(**kwargs):
info: 返回信息
result: 返回数据
"""
company = Company()
company.query = request.json['query']
skip = request.json['skip']
limit = request.json['limit']
data = company.search_company(skip, limit)
return {"info": '查询成功', "result": data}, 200
return {"info": '调整中'}, 200
@company_route.route('/index', methods=['POST'])

View File

@ -13,7 +13,7 @@ from Crypto.Cipher import PKCS1_v1_5 as PKCS1_cipher
from Board.board_db import FIND_DATA, INSERT_DATA
from TestCenter.tc_db import update_user_info_by_cid, find_user_info_by_cid, delete_data, find_data, update_data, \
delete_pdf, find_admin_data, insert_admin_data
delete_pdf
def search_evaluate_record_scripts(types):
@ -577,7 +577,3 @@ def new_test_company_impl(email, company, code, legal_person):
return company_verify_script(company, code, legal_person, cid)
return new_company_main_process()
if __name__ == '__main__':
delete_cid_cascade_scripts("qrSQXTqs", ['指标明细'])

View File

@ -1,6 +1,21 @@
from TestCenter.tc_impl import *
class CompanyVerify:
def __init__(self):
""""""
def company_verify(self):
"""
企业认证
Returns:
info: 认证结果
"""
info = company_verify_script(self.company, self.code, self.legal_person, self.cid)
return info
class TestObj:
def __init__(self):
@ -36,15 +51,6 @@ class TestObj:
info = delete_rid_cascade_scripts(self.rid, self.record, self.data)
return info
def company_verify(self):
"""
企业认证
Returns:
info: 认证结果
"""
info = company_verify_script(self.company, self.code, self.legal_person, self.cid)
return info
def change_verified(self):
"""
修改企业认证状态

View File

@ -7,6 +7,18 @@ from user.user_auth import verify_token
test_route = Blueprint('test', __name__)
@test_route.route('/enterprise_certification', methods=['POST'])
@api_secret
def enterprise_certification():
"""测试用企业认证"""
test_obj = TestObj()
test_obj.company = request.json['company']
test_obj.code = request.json['code']
test_obj.legal_person = request.json['legal_person']
test_obj.cid = request.json['cid']
return {"info": "调整中"}, 200
@test_route.route('/search_evaluate_record', methods=['POST'])
@api_secret
def search_evaluate_record_route():
@ -79,26 +91,6 @@ def generate_rid_information_route():
return {"info": info}, status
@test_route.route('/enterprise_certification', methods=['POST'])
@api_secret
def enterprise_certification():
"""
管理端企业认证接口
Parameters:
-
Returns:
-
"""
test_obj = TestObj()
test_obj.company = request.json['company']
test_obj.code = request.json['code']
test_obj.legal_person = request.json['legal_person']
test_obj.cid = request.json['cid']
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():

2
app.py
View File

@ -5,7 +5,7 @@ from Board.board_routes import board_route
from Manage.industry.industry_routes import industry_route
from Model.model_routes import model_route
from TestCenter.tc_routes import test_route
from Manage.company.routes import company_route
from Manage.company.company_routes import company_route
from Manage.rating.routes import rating_route
from user.user_routes import user_route
from Manage.macro.routes import macro_route

View File

@ -1,123 +0,0 @@
import pymongo
MG_USER = "root"
MG_PWD = "RYIHrqml#LSW6#!*"
MG_HOST = "116.63.130.34"
MG_PORT = "27020"
CLIENT = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(MG_USER, MG_PWD, MG_HOST, MG_PORT))
def INSERT_DATA(param1, param2, param3):
"""
插入一条数据
Parameters:
param1: str 数据库
param2: str 数据集
param3: obj 插入数据
Returns:
True 执行成功
False 执行失败
"""
collection = CLIENT[param1][param2]
collection.insert_one(param3)
return True
def DELETE_DATA(param1, param2, param3):
"""
根据查询条件删除一条文档
Parameters:
param1: str 数据库
param2: str 数据集
param3: obj 查询条件
Returns:
True 执行完成
"""
collection = CLIENT[param1][param2]
collection.delete_one(param3)
return True
def UPDATE_DATA(param1, param2, param3, param4):
"""
Parameters:
param1: str 数据库
param2: str 数据集
param3: obj 查询条件
param4: obj 插入数据
Returns:
True 执行完成
"""
collection = CLIENT[param1][param2]
collection.update_one(param3, param4)
return True
def UPSERT_DATA(param1, param2, param3, param4):
"""
Parameters:
param1: str 数据库
param2: str 数据集
param3: obj 查询条件
param4: obj 插入数据
Returns:
True 执行完成
"""
collection = CLIENT[param1][param2]
collection.update_one(param3, {"$set": param4}, upsert=True)
return True
def FIND_DATA(param1, param2, param3):
"""
根据查询条件查询符合条件的所有数据
Parameters:
param1: str 数据库
param2: str 数据集
param3: obj 查询条件
Returns:
records: list 符合条件的数据
"""
collection = CLIENT[param1][param2]
records = list(collection.find(param3, {'_id': False}))
return records
def FIND_DATA_PAGE(param1, param2, param3, param4, param5):
"""
根据查询条件查询符合条件的所有数据
Args:
param1: str 数据库
param2: str 数据集
param3: obj 查询条件
param4: skip 页码
param5: limit 每页条数
Returns:
records: list 符合条件的数据
"""
collection = CLIENT[param1][param2]
param4 = (param4 - 1) * param5
count = collection.find(param3).count()
records = list(collection.find(param3, {'_id': False}).skip(param4).limit(param5))
result = {
"total": count,
"data": records
}
return result
def UPDATE_INFO(param1, param2, param3, param4):
"""
更新信息
Parameters:
param1: str 数据库
param2: str 数据集
param3: obj 查询条件
param4: obj 插入数据
Returns:
record: 编辑结果
"""
collection = CLIENT[param1][param2]
res = collection.update_one(param3, {"$set": param4})
info = res.raw_result['updatedExisting']
return info