From 37fe78b690bd2e90d92b1f6bd44ff49eea10e87a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E6=80=9D=E5=B7=9D?= Date: Wed, 27 Apr 2022 15:02:43 +0800 Subject: [PATCH] =?UTF-8?q?=E7=94=A8=E6=88=B7=E6=A8=A1=E5=9D=97=E9=87=8D?= =?UTF-8?q?=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- CompanyUser/CompanyUserImpl.py | 564 --------------------- CompanyUser/CompanyUserObj.py | 188 ------- CompanyUser/CompanyUserRoute.py | 203 -------- CompanyUser/__init__.py | 0 Implements/User/RealCompanyVerifyImpl.py | 258 ++++++++++ ObjectsInProject/User/RealCompanyVerify.py | 93 ++++ Routes/User/TFSECompanyUserRoute.py | 31 +- 7 files changed, 357 insertions(+), 980 deletions(-) delete mode 100644 CompanyUser/CompanyUserImpl.py delete mode 100644 CompanyUser/CompanyUserObj.py delete mode 100644 CompanyUser/CompanyUserRoute.py delete mode 100644 CompanyUser/__init__.py create mode 100644 Implements/User/RealCompanyVerifyImpl.py create mode 100644 ObjectsInProject/User/RealCompanyVerify.py diff --git a/CompanyUser/CompanyUserImpl.py b/CompanyUser/CompanyUserImpl.py deleted file mode 100644 index 5283ec6..0000000 --- a/CompanyUser/CompanyUserImpl.py +++ /dev/null @@ -1,564 +0,0 @@ -import json -import time -import random -import base64 -import requests -from urllib.request import Request, urlopen - -from werkzeug.security import check_password_hash -from itsdangerous import TimedJSONWebSignatureSerializer as Serializer - -from DBHelper.MongoHelper import MongoHelper -from Utils.ErrorUtil import ReturnConditionCheckFailed -from CompanyUser.CompanyUserObj import CompanyUser, VerifyInfo, EmailVerifyCodeRecord, VerifyMaterial, \ - CompanyVerifyThreeFactors - - -class CompanyUserImpl(CompanyUser): - """企业用户实现类""" - - db = MongoHelper("tfse_v0.21") - - def check_email_registered(self): - """检查邮箱已被注册""" - email = self.db.find_single_column( - "应用端", - "企业用户", - {"邮箱": self.email}, - "邮箱" - ) - if email is not None: - raise ReturnConditionCheckFailed("邮箱已被注册", 200) - - def check_email_not_registered(self): - """检查邮箱未被注册""" - email = self.db.find_single_column( - "应用端", - "企业用户", - {"邮箱": self.email}, - "邮箱" - ) - if email is None: - raise ReturnConditionCheckFailed("邮箱未被注册", 200) - - def check_vcode_correct(self, code): - """检查邮箱验证码是否正确""" - verify_data = self.db.find_single_data( - "应用端", - "邮箱验证码记录", - {"邮箱": self.email}, - ["验证码", "验证有效期"] - ) - - if not verify_data: - raise ReturnConditionCheckFailed("验证码错误", 200) - - if verify_data['验证码'] != code: - raise ReturnConditionCheckFailed("验证码错误", 200) - - if time.time() - verify_data['验证有效期'] > 300: - raise ReturnConditionCheckFailed('验证码过期', 200) - - def login(self): - """登录""" - - company_user = self.db.find_single_data( - "应用端", - "企业用户", - {"邮箱": self.email}, - ["企业ID", "企业名称", "邮箱", "密码", "已认证"] - ) - - def check_user_existed(): - """检查用户是否存在""" - if company_user is None: - raise ReturnConditionCheckFailed("不存在该用户", 200) - - def check_password_correct(): - """检查密码是否正确""" - if not check_password_hash(company_user['密码'], self.pwd): - raise ReturnConditionCheckFailed("登录密码错误", 200) - - def create_token(param): - """ - 创建token - param: 传入参数，用于创建token - return: 用户token - """ - secret_key = '0FTuOi^#Afx1@2@F' - token_expire = 14400 - s = Serializer(secret_key, expires_in=token_expire) - token = '' + s.dumps(param).decode('ascii') - return token - - def __main__(): - """""" - check_user_existed() - check_password_correct() - - self.cid = company_user['企业ID'] - self.name = company_user['企业名称'] - self.verify_status = company_user['已认证'] - - return_body = self.dict_to_show(columns=["企业名称", "认证状态", "token"]) - return_body['token'] = create_token({"cid": self.cid}) - - return return_body - - return __main__() - - def register(self, code): - """注册""" - - def make_new_cid(): - """生成新的企业ID，如果该ID存在，则重新生成""" - - def random_cid(num): - """随机企业ID""" - choices = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789' - salt = '' - for i in range(num): - salt += random.choice(choices) - return salt - - new_cid = random_cid(8) - - case = self.db.find_single_column( - "应用端", - "企业用户", - {"企业ID": new_cid}, - "企业ID" - ) is not None - while case: - new_cid = random_cid(8) - return new_cid - - def __main__(): - """""" - self.check_email_registered() - self.check_vcode_correct(code) - - self.cid = make_new_cid() - self.avatar_id = "623152edf36000004f00124e" - self.verify_status = "否" - self.verify_info = VerifyInfo() - - self.db.insert_single_data( - "应用端", - "企业用户", - self.dict_to_save() - ) - - self.db.delete_single_data( - "应用端", - "邮箱验证码记录", - {"邮箱": self.email} - ) - - __main__() - - def get_company_user_info(self): - """获取企业用户信息""" - company_user_data = self.db.find_single_data( - "应用端", - "企业用户", - {"企业ID": self.cid}, - ["企业ID", "企业名称", "邮箱", "已认证"] - ) - - self.name = company_user_data['企业名称'] - self.email = company_user_data['邮箱'] - self.verify_status = company_user_data['已认证'] - - return self.dict_to_show(columns=["企业ID", "企业名称", "邮箱", "认证状态"]) - - def get_avatar(self): - """获取用户头像""" - - self.avatar_id = self.db.find_single_column( - "应用端", - "企业用户", - {"企业ID": self.cid}, - "头像fid" - ) - - avatar = self.db.find_file( - "文件", - "企业用户头像", - self.avatar_id - ) - - return avatar - - def change_avatar(self, avatar_option): - """修改企业用户头像""" - avatars = { - "avatar_01": "623152edf36000004f00124e", - "avatar_02": "623152edf36000004f001250", - "avatar_03": "623152edf36000004f00124f", - "avatar_04": "623152edf36000004f001251", - "avatar_05": "623152edf36000004f001252", - "avatar_06": "623152edf36000004f001253" - } - self.avatar_id = avatars[avatar_option] - self.db.update_single_data( - "应用端", - "企业用户", - {"企业ID": self.cid}, - {"头像fid": self.avatar_id} - ) - - def change_password(self, code): - """修改登录密码""" - self.check_email_not_registered() - self.check_vcode_correct(code) - self.db.update_single_data( - "应用端", - "企业用户", - {"邮箱": self.email}, - self.dict_to_save(columns=["密码"]) - ) - - def change_email(self, code): - """更换邮箱""" - self.check_vcode_correct(code) - self.check_email_registered() - self.db.update_single_data( - "应用端", - "企业用户", - {"企业ID": self.cid}, - self.dict_to_save(columns=["邮箱"]) - ) - - -class VerifyMaterialImpl(VerifyMaterial): - """认证材料实现类""" - - db = MongoHelper("tfse_v0.21") - - @staticmethod - def read_img_by_base64(img_file): - """读取base64编码格式的图片""" - data = img_file.read() - try: - encode_str = str(base64.b64encode(data), 'utf-8') - except TypeError: - encode_str = base64.b64encode(data) - return encode_str - - def check_is_verified(self): - """检查企业是否已认证""" - self.verify_status = self.db.find_single_column( - "应用端", - "企业用户", - {"企业ID": self.cid}, - "已认证" - ) - if self.verify_status == "是": - raise ReturnConditionCheckFailed("企业已认证", 200) - - def upload_id_card(self): - """上传身份证""" - - def trans_date_format(_time): - """将解析出来的时间数据加上年月日""" - year = _time[:4] - month = _time[4:6] - day = _time[6:8] - return year + '.' + month + '.' + day - - def id_card_recognition(side): - """调用身份证图片识别接口""" - - url = 'http://dm-51.data.aliyun.com/rest/160601/ocr/ocr_idcard.json' - APPCODE = "6d6c3146677c40bf9e3e1e2aa73404cd" - - # 请求头 - headers = { - 'Authorization': 'APPCODE %s' % APPCODE, - 'Content-Type': 'application/json; charset=UTF-8' - } - - # 请求体 - body = { - "configure": {'side': side}, - "image": self.read_img_by_base64(self.image) - } - - try: - params = json.dumps(body).encode(encoding='UTF8') - req = Request(url, params, headers) - res = urlopen(req) - html = res.read() - result = json.loads(html.decode("utf8")) - - if result['is_fake']: - raise ReturnConditionCheckFailed("伪造身份证", 200) - - id_card = dict() - if side == 'face': - id_card['姓名'] = result['name'] - id_card['性别'] = result['sex'] - id_card['民族'] = result['nationality'] - id_card['出生'] = "{}年{}月{}日".format(result['birth'][:4], result['birth'][4:6], result['birth'][6:]) - id_card['住址'] = result['address'] - id_card['身份证号码'] = result['num'] - else: - id_card['签发机关'] = result['issue'] - id_card['有效期限'] = '{}-{}'.format(trans_date_format(result['start_date']), trans_date_format(result['end_date'])) - return id_card - except Exception: - raise ReturnConditionCheckFailed("身份证识别失败", 200) - - def id_card_recognition_result(): - """ - 身份证文本识别 - cid: 企业ID - image: 图片 - return: 识别结果 - """ - - # 身份证正面文本识别 - result = id_card_recognition(side='face') - - # 重置image读取位置，不然下一步读不出图片 - self.image.seek(0) - - # 更新企业认证中身份证的信息 - self.db.update_single_data( - "应用端", - "企业用户", - {"企业ID": self.cid}, - {"认证信息.法人姓名": result['姓名'], "认证信息.法人身份证": result['身份证号码']} - ) - - return {"姓名": result['姓名'], "身份证号码": result['身份证号码']} - - def __main__(): - """""" - self.check_is_verified() - return id_card_recognition_result() - - return __main__() - - def upload_business_license(self): - """上传营业执照""" - - def trans_date_format(_time): - """将解析出来的时间数据加上年月日""" - year = _time[:4] - month = _time[4:6] - day = _time[6:8] - return year + '年' + month + '月' + day + '日' - - def business_license_recognition(): - """调用营业执照识别接口""" - url = 'https://dm-58.data.aliyun.com/rest/160601/ocr/ocr_business_license.json' - app_code = "6d6c3146677c40bf9e3e1e2aa73404cd" - - headers = { - 'Authorization': 'APPCODE %s' % app_code, - 'Content-Type': 'application/json; charset=UTF-8' - } - - body = { - "image": self.read_img_by_base64(self.image) - } - - response = requests.post(url=url, data=json.dumps(body), headers=headers) - - try: - res = json.loads(response.text) - data = dict() - data['统一社会信用代码'] = res['reg_num'] - data['名称'] = res['name'] - data['注册资本'] = res['capital'] - data['类型'] = res['type'] - data['成立日期'] = trans_date_format(res['establish_date']) - data['法定代表人'] = res['person'] - data['经营范围'] = res['business'] - data['住所'] = res['address'] - return data - except Exception: - raise ReturnConditionCheckFailed("营业执照识别异常", 200) - - def business_license_recognition_result(): - """营业执照识别结果""" - - # 识别营业执照 - result = business_license_recognition() - - # 图片识别正确 更新企业认证中营业执照的信息 - self.db.update_single_data( - "应用端", - "企业用户", - {"企业ID": self.cid}, - {"认证信息.企业名称": result['名称'], "认证信息.统一社会信用代码": result['统一社会信用代码']} - ) - return {"公司名称": result['名称'], "统一社会信用代码": result['统一社会信用代码']} - - def _main__(): - """""" - self.check_is_verified() - return business_license_recognition_result() - - return _main__() - - -class CompanyVerifyThreeFactorsImpl(CompanyVerifyThreeFactors): - """企业认证三要素实现类""" - - db = MongoHelper("tfse_v0.21") - - def company_verify(self): - """企业认证""" - - def get_verify_info(): - """获取企业认证信息""" - record = self.db.find_single_data( - "应用端", - "企业用户", - {"企业ID": self.cid}, - ["认证信息.企业名称", "认证信息.法人姓名", "认证信息.统一社会信用代码"] - ) - - if not record: - raise ReturnConditionCheckFailed("企业认证信息异常", 200) - - verify_info = record['认证信息'] - self.code = verify_info['统一社会信用代码'] - self.name = verify_info['企业名称'] - self.legal_person = verify_info['法人姓名'] - - def check_verify_info_complete(): - """检查认证信息是否完整""" - if None in [self.code, self.name, self.legal_person]: - raise ReturnConditionCheckFailed("企业认证信息异常", 200) - - def check_three_elements_by_tyc(): - """调用天眼查企业三要素验证接口""" - token = "5407573b-e41e-4ab4-9f46-daa0032ad0a4" - base = "http://open.api.tianyancha.com/services/open/ic/verify/2.0?code={}&name={}&legalPersonName={}" - url = base.format(self.code, self.name, self.legal_person) - headers = {'Authorization': token} - res = json.loads(requests.get(url, headers=headers).text) - try: - if res['result']['result'] != 1: - raise ReturnConditionCheckFailed("企业认证未通过", 200) - except Exception: - raise ReturnConditionCheckFailed("认证接口异常", 200) - - def init_company_data(): - """初始化企业主页数据""" - init_url = "http://api.fecribd.com/etl_tfse/company/init_company_data" - init_token = "dmfd7FshT!5Wng9^gcCcQV7T6FBnVgl4" - init_headers = {'token': init_token, "Content-Type": "application/json;charset=UTF-8"} - data = json.dumps({"cid": self.cid, "company_name": self.name}) - response = requests.post(init_url, headers=init_headers, data=data) - - res = json.loads(response.text) - if res['info'] == "企业数据初始化成功": - raise ReturnConditionCheckFailed("企业数据初始化异常", 200) - else: - company_user = CompanyUser() - company_user.cid = self.cid - company_user.name = self.name - company_user.verify_status = "是" - - self.db.upsert_single_data( - "应用端", - "企业用户", - {"企业ID": self.cid}, - company_user.dict_to_save(columns=["企业名称", "已认证"]) - ) - - def __main__(): - get_verify_info() - check_verify_info_complete() - check_three_elements_by_tyc() - init_company_data() - - return __main__() - - -class EmailVerifyCodeRecordImpl(EmailVerifyCodeRecord): - """用户通知邮件实现类""" - - db = MongoHelper("tfse_v0.21") - - def make_verify_code(self): - """生成邮箱验证码""" - - def make_verify_code(): - """随机生成6位数字验证码""" - num_code = "" - for i in range(6): - ch = chr(random.randrange(ord('0'), ord('9') + 1)) - num_code += ch - return num_code - - self.v_code = make_verify_code() - self.v_period = time.time() + 300 - self.db.upsert_single_data( - "应用端", - "邮箱验证码记录", - {"邮箱": self.email}, - self.dict_to_save() - ) - - def transfer_email_api(self): - """发送邮件""" - # 邮箱服务接口 - email_api = 'http://116.63.130.34:30001' - - v_types = { - "register": "注册账号", - "resetpwd": "修改密码", - "resetemail": "修改邮箱" - } - - headers = {"Content-Type": "application/json;charset=UTF-8"} - data = { - "title": "【远东资信】{}".format(v_types[self.v_type]), - "sender": 'fecribd@fecr.com.cn', - "recipients": [self.email], - "msg_body": "您{}的验证码为 【{}】，5分钟内有效。".format(v_types[self.v_type], self.v_code) - } - - response = requests.post(url=email_api + '/send_mail', headers=headers, data=json.dumps(data)) - return response.text - - def check_email_registered(self): - """检查邮箱已被注册""" - email = self.db.find_single_column( - "应用端", - "企业用户", - {"邮箱": self.email}, - "邮箱" - ) - if email is not None: - raise ReturnConditionCheckFailed("邮箱已被注册", 200) - - def check_email_not_registered(self): - """检查邮箱未被注册""" - email = self.db.find_single_column( - "应用端", - "企业用户", - {"邮箱": self.email}, - "邮箱" - ) - if email is None: - raise ReturnConditionCheckFailed("邮箱未被注册", 200) - - def send_email(self): - """""" - - if self.v_type == 'register' or self.v_type == 'resetemail': - self.check_email_registered() - self.make_verify_code() - self.transfer_email_api() - - elif self.v_type == 'resetpwd': - self.check_email_not_registered() - self.make_verify_code() - self.transfer_email_api() diff --git a/CompanyUser/CompanyUserObj.py b/CompanyUser/CompanyUserObj.py deleted file mode 100644 index ddbcdab..0000000 --- a/CompanyUser/CompanyUserObj.py +++ /dev/null @@ -1,188 +0,0 @@ -from werkzeug.security import generate_password_hash - -from Utils.ValidateUtil import ValidateAttr, Validate - - -class VerifyInfo(object): - """认证信息""" - name = ValidateAttr(field='name', default=None, type=str, error_info='企业名称异常', error_code=200) - code = ValidateAttr(field='code', default=None, type=str, error_info='统一社会信用代码异常', error_code=200) - legal_person = ValidateAttr(field='legal_person', default=None, type=str, error_info='法人姓名异常', error_code=200) - legal_person_id = ValidateAttr(field='legal_person_id', default=None, type=str, error_info='法人身份证号异常', error_code=200) - - def dict_to_save(self, **kwargs): - """""" - _dict_ = { - "企业名称": self.name, - "统一社会信用代码": self.code, - "法人姓名": self.legal_person, - "法人身份证": self.legal_person_id, - } - - if 'columns' in kwargs: - _dict_ = {key: _dict_[key] for key in kwargs['columns']} - - return _dict_ - - -class CompanyUser(object): - """企业用户""" - cid = ValidateAttr(field='cid', type=str, length=8, error_info='企业ID异常', error_code=200) - name = ValidateAttr(field='name', type=str, default=None, error_info='企业名称异常', error_code=200) - email = ValidateAttr(field='email', func=Validate.email, error_info='邮箱格式异常', error_code=200) - pwd = ValidateAttr(field='pwd', func=Validate.password, error_info='密码强度异常', error_code=200) - avatar_id = ValidateAttr(field='avatar_id', type=str, length=24, error_info='头像ID异常', error_code=200) - verify_status = ValidateAttr(field='verify_status', type=str, in_list=["是", "否"], error_info='认证状态异常', error_code=200) - verify_info = ValidateAttr(field='verify_info', type=VerifyInfo, error_info='认证信息异常', error_code=200) - register_time = ValidateAttr(field='register_time', default=None, type=str, error_info='注册时间异常', error_code=200) - - def dict_to_show(self, **kwargs): - """""" - _dict_ = { - "企业ID": self.cid, - "企业名称": '企业用户' if self.name is None else self.name, - "邮箱": self.email, - "认证状态": "已认证" if self.verify_status == "是" else "未认证", - "token": None - } - - if 'columns' in kwargs: - _dict_ = {key: _dict_[key] for key in kwargs['columns']} - - return _dict_ - - def dict_to_save(self, **kwargs): - """存储对象""" - - # 对象字段与实例字段的映射 - keys_map = { - "cid": "企业ID", - "name": "企业名称", - "email": "邮箱", - "avatar_id": "头像fid", - "pwd": "密码", - "register_time": "注册时间", - "verify_status": "已认证", - "verify_info": "认证信息" - } - - # 实例一个存储对象 - _dict_ = dict() - - # 实例字段列表 - obj_keys = [key for key in self.__dict__.keys()] - - # 实例的值循环赋值给存储对象 - for key in obj_keys: - _dict_[keys_map[key]] = self.__dict__[key] - - # 存储对象子集 - if 'columns' in kwargs: - _dict_ = {key: _dict_[key] for key in kwargs['columns']} - - if '密码' in _dict_: - _dict_['密码'] = generate_password_hash(_dict_['密码']) - - return _dict_ - - def login(self): - """登录""" - - def register(self, **kwargs): - """注册""" - - def get_company_user_info(self): - """获取企业用户信息""" - - def get_avatar(self): - """获取用户头像""" - - def change_avatar(self, **kwargs): - """修改用户头像""" - - def change_password(self, **kwargs): - """修改登录密码""" - - def change_email(self, **kwargs): - """修改登录邮箱""" - - -class IDCardFront(object): - """身份证正面""" - name = ValidateAttr() - gender = ValidateAttr() - nationality = ValidateAttr() - birth = ValidateAttr() - address = ValidateAttr() - id_num = ValidateAttr() - - -class IDCardBack(object): - """身份证反面""" - issuing_authority = ValidateAttr() - validity_period = ValidateAttr() - - -class BusinessLicense(object): - """营业执照""" - reg_num = ValidateAttr() - name = ValidateAttr() - capital = ValidateAttr() - type = ValidateAttr() - establish_date = ValidateAttr() - person = ValidateAttr() - business = ValidateAttr() - address = ValidateAttr() - - -class VerifyMaterial(object): - """认证材料""" - - cid = ValidateAttr(field='cid', type=str, length=8, error_info='企业ID异常', error_code=200) - image = ValidateAttr(field='image', func=Validate.image, error_info='图片格式异常', error_code=200) - verify_status = ValidateAttr(field='verify_status', type=str, in_list=["是", "否"], error_info='认证状态异常', error_code=200) - - def upload_id_card(self): - """上传身份证""" - - def upload_business_license(self): - """上传营业执照""" - - -class CompanyVerifyThreeFactors(object): - """企业认证三要素""" - cid = ValidateAttr(field='cid', type=str, length=8) - code = ValidateAttr(field='code', type=str) - name = ValidateAttr(field='name', type=str) - legal_person = ValidateAttr(field='legal_person', type=str) - - fields_map = { - "cid": "企业ID", - "code": "统一社会信用代码", - "name": "企业名称", - "legal_person": "法人姓名", - } - - def company_verify(self): - """企业认证""" - - -class EmailVerifyCodeRecord(object): - """用户邮件通知""" - - email = ValidateAttr(field='email', func=Validate.email, error_info='邮箱格式异常', error_code=200) - v_period = ValidateAttr(field='v_period', type=float, error_info='验证有效期格式异常', error_code=200) - v_code = ValidateAttr(field='v_code', type=str, length=6, error_info='验证码格式异常', error_code=200) - v_type = ValidateAttr(field="v_type", type=str, in_list=["register", "resetemail", "resetpwd"], error_info="邮件验证类型错误", error_code=200) - - def dict_to_save(self): - """""" - _dict_ = { - "邮箱": self.email, - "验证有效期": self.v_period, - "验证码": self.v_code - } - return _dict_ - - def send_email(self): - """发送邮件""" diff --git a/CompanyUser/CompanyUserRoute.py b/CompanyUser/CompanyUserRoute.py deleted file mode 100644 index 83e4c3a..0000000 --- a/CompanyUser/CompanyUserRoute.py +++ /dev/null @@ -1,203 +0,0 @@ -from flask import Blueprint, request, Response - -from common.rsa import decrypt_data -from Utils.AuthUtil import check_block, verify_token -from Utils.ErrorUtil import ReturnConditionCheckFailed -from CompanyUser.CompanyUserImpl import CompanyUserImpl, EmailVerifyCodeRecordImpl, VerifyMaterialImpl, \ - CompanyVerifyThreeFactorsImpl - -company_user_route = Blueprint('user', __name__) - - -@company_user_route.route('/login', methods=['POST']) -@check_block -def login(): - """企业用户登录""" - try: - company_user = CompanyUserImpl() - company_user.email = request.json['email'] - - try: - company_user.pwd = decrypt_data(encrypt_msg=request.json['pwd']) - except Exception: - return {"info": "登录密码错误"}, 200 - - result = company_user.login() - return {"info": "登陆成功", "result": result}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/register', methods=['POST']) -@check_block -def register(): - """企业用户注册""" - try: - req = request.json - company_user = CompanyUserImpl() - company_user.email = req['email'] - company_user.pwd = req['pwd'] - company_user.register(req['code']) - response = {"info": "注册成功"}, 200 - return response - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/user_info', methods=['GET']) -@verify_token -def user_info(**kwargs): - """获取企业用户信息""" - try: - company_user = CompanyUserImpl() - company_user.cid = kwargs['cid'] - result = company_user.get_company_user_info() - return {"info": "查询结果", "result": result} - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/get_avatar', methods=['GET']) -@verify_token -def get_avatar(**kwargs): - """获取头像""" - try: - company_user = CompanyUserImpl() - company_user.cid = kwargs['cid'] - response = Response(company_user.get_avatar(), content_type='image/jpeg') - return response - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/change_avatar', methods=['POST']) -@verify_token -def change_avatar(**kwargs): - """修改头像""" - try: - company_user = CompanyUserImpl() - company_user.cid = kwargs['cid'] - company_user.change_avatar(request.json['avatar_id']) - return {"info": "修改成功"}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/change_pwd', methods=['POST']) -@check_block -def change_pwd(): - """修改密码""" - try: - req = request.json - company_user = CompanyUserImpl() - company_user.email = req['email'] - company_user.pwd = decrypt_data(encrypt_msg=req['pwd']) - company_user.change_password(req['code']) - return {"info": "修改成功"}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/change_email', methods=['POST']) -@check_block -@verify_token -def change_email(**kwargs): - """更换账号邮箱""" - try: - req = request.json - company_user = CompanyUserImpl() - company_user.cid = kwargs['cid'] - company_user.email = req['email'] - company_user.change_email(req['code']) - return {"info": "修改成功"}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/verify_email', methods=['POST']) -@check_block -def verify_email(): - """发送邮箱验证码""" - try: - req = request.json - email_verify = EmailVerifyCodeRecordImpl() - email_verify.email = req['email'] - email_verify.v_type = req['v_type'] - email_verify.send_email() - return {"info": "邮件已发送"}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/upload_id_card', methods=['POST']) -@verify_token -def upload_id_card(**kwargs): - """上传身份证""" - try: - material = VerifyMaterialImpl() - material.cid = kwargs['cid'] - material.image = request.files['image'] - result = material.upload_id_card() - return {"info": "识别结果", "result": result}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/upload_business_license', methods=['POST']) -@verify_token -def upload_business_license(**kwargs): - """上传营业执照""" - try: - material = VerifyMaterialImpl() - material.cid = kwargs['cid'] - material.image = request.files['image'] - result = material.upload_business_license() - return {"info": "识别结果", "result": result}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - -@company_user_route.route('/company_verify', methods=['GET']) -@verify_token -def company_verify_route(**kwargs): - """企业认证""" - try: - company_verify = CompanyVerifyThreeFactorsImpl() - company_verify.cid = kwargs['cid'] - company_verify.company_verify() - return {"info": "认证成功"}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 diff --git a/CompanyUser/__init__.py b/CompanyUser/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/Implements/User/RealCompanyVerifyImpl.py b/Implements/User/RealCompanyVerifyImpl.py new file mode 100644 index 0000000..e84e433 --- /dev/null +++ b/Implements/User/RealCompanyVerifyImpl.py @@ -0,0 +1,258 @@ +import base64 +import json +from urllib.request import urlopen, Request + +import requests + +from DBHelper.MongoHelper import MongoHelper +from ObjectsInProject.User.RealCompanyVerify import UploadMaterial, RealCompanyVerify +from ObjectsInProject.User.TFSECompanyUser import TFSECompanyUser +from Utils.ErrorUtil import ReturnConditionCheckFailed + + +class UploadMaterialImpl(UploadMaterial): + """认证材料实现类""" + + db = MongoHelper("tfse_v0.21") + + @staticmethod + def read_img_by_base64(img_file): + """读取base64编码格式的图片""" + data = img_file.read() + try: + encode_str = str(base64.b64encode(data), 'utf-8') + except TypeError: + encode_str = base64.b64encode(data) + return encode_str + + def check_is_verified(self): + """检查企业是否已认证""" + self.verify_status = self.db.find_single_column( + "应用端", + "企业用户", + {"企业ID": self.cid}, + "已认证" + ) + if self.verify_status == "是": + raise ReturnConditionCheckFailed(failed_info="企业已认证", status_code=200) + + def upload_id_card(self): + """上传身份证""" + + def trans_date_format(_time): + """将解析出来的时间数据加上年月日""" + year = _time[:4] + month = _time[4:6] + day = _time[6:8] + return year + '.' + month + '.' + day + + def id_card_recognition(side): + """调用身份证图片识别接口""" + + url = 'http://dm-51.data.aliyun.com/rest/160601/ocr/ocr_idcard.json' + APPCODE = "6d6c3146677c40bf9e3e1e2aa73404cd" + + # 请求头 + headers = { + 'Authorization': 'APPCODE %s' % APPCODE, + 'Content-Type': 'application/json; charset=UTF-8' + } + + # 请求体 + body = { + "configure": {'side': side}, + "image": self.read_img_by_base64(self.image) + } + + try: + params = json.dumps(body).encode(encoding='UTF8') + req = Request(url, params, headers) + res = urlopen(req) + html = res.read() + result = json.loads(html.decode("utf8")) + + if result['is_fake']: + raise ReturnConditionCheckFailed("伪造身份证", 200) + + id_card = dict() + if side == 'face': + id_card['姓名'] = result['name'] + id_card['性别'] = result['sex'] + id_card['民族'] = result['nationality'] + id_card['出生'] = "{}年{}月{}日".format(result['birth'][:4], result['birth'][4:6], result['birth'][6:]) + id_card['住址'] = result['address'] + id_card['身份证号码'] = result['num'] + else: + id_card['签发机关'] = result['issue'] + id_card['有效期限'] = '{}-{}'.format(trans_date_format(result['start_date']), trans_date_format(result['end_date'])) + return id_card + except Exception: + raise ReturnConditionCheckFailed("身份证识别失败", 200) + + def id_card_recognition_result(): + """ + 身份证文本识别 + cid: 企业ID + image: 图片 + return: 识别结果 + """ + + # 身份证正面文本识别 + result = id_card_recognition(side='face') + + # 重置image读取位置，不然下一步读不出图片 + self.image.seek(0) + + # 更新企业认证中身份证的信息 + self.db.update_single_data( + "应用端", + "企业用户", + {"企业ID": self.cid}, + {"认证信息.法人姓名": result['姓名'], "认证信息.法人身份证": result['身份证号码']} + ) + + return {"姓名": result['姓名'], "身份证号码": result['身份证号码']} + + def __main__(): + self.check_is_verified() + return id_card_recognition_result() + + return __main__() + + def upload_business_license(self): + """上传营业执照""" + + def trans_date_format(_time): + """将解析出来的时间数据加上年月日""" + year = _time[:4] + month = _time[4:6] + day = _time[6:8] + return year + '年' + month + '月' + day + '日' + + def business_license_recognition(): + """调用营业执照识别接口""" + url = 'https://dm-58.data.aliyun.com/rest/160601/ocr/ocr_business_license.json' + app_code = "6d6c3146677c40bf9e3e1e2aa73404cd" + + headers = { + 'Authorization': 'APPCODE %s' % app_code, + 'Content-Type': 'application/json; charset=UTF-8' + } + + body = { + "image": self.read_img_by_base64(self.image) + } + + response = requests.post(url=url, data=json.dumps(body), headers=headers) + + try: + res = json.loads(response.text) + data = dict() + data['统一社会信用代码'] = res['reg_num'] + data['名称'] = res['name'] + data['注册资本'] = res['capital'] + data['类型'] = res['type'] + data['成立日期'] = trans_date_format(res['establish_date']) + data['法定代表人'] = res['person'] + data['经营范围'] = res['business'] + data['住所'] = res['address'] + return data + except Exception: + raise ReturnConditionCheckFailed("营业执照识别异常", 200) + + def business_license_recognition_result(): + """营业执照识别结果""" + + # 识别营业执照 + result = business_license_recognition() + + # 图片识别正确 更新企业认证中营业执照的信息 + self.db.update_single_data( + "应用端", + "企业用户", + {"企业ID": self.cid}, + {"认证信息.企业名称": result['名称'], "认证信息.统一社会信用代码": result['统一社会信用代码']} + ) + return {"公司名称": result['名称'], "统一社会信用代码": result['统一社会信用代码']} + + def _main_(): + self.check_is_verified() + return business_license_recognition_result() + + return _main_() + + +class RealCompanyVerifyImpl(RealCompanyVerify): + """企业认证三要素实现类""" + + db = MongoHelper("tfse_v0.21") + + def verify(self): + """企业认证""" + + def get_verify_info(): + """获取企业认证信息""" + record = self.db.find_single_data( + "应用端", + "企业用户", + {"企业ID": self.cid}, + ["认证信息.企业名称", "认证信息.法人姓名", "认证信息.统一社会信用代码"] + ) + + if not record: + raise ReturnConditionCheckFailed("企业认证信息异常", 200) + + verify_info = record['认证信息'] + self.code = verify_info['统一社会信用代码'] + self.name = verify_info['企业名称'] + self.legal_person = verify_info['法人姓名'] + + def check_verify_info_complete(): + """检查认证信息是否完整""" + if None in [self.code, self.name, self.legal_person]: + raise ReturnConditionCheckFailed("企业认证信息异常", 200) + + def check_three_elements_by_tyc(): + """调用天眼查企业三要素验证接口""" + token = "5407573b-e41e-4ab4-9f46-daa0032ad0a4" + base = "http://open.api.tianyancha.com/services/open/ic/verify/2.0?code={}&name={}&legalPersonName={}" + url = base.format(self.code, self.name, self.legal_person) + headers = {'Authorization': token} + res = json.loads(requests.get(url, headers=headers).text) + try: + if res['result']['result'] != 1: + raise ReturnConditionCheckFailed("企业认证未通过", 200) + except Exception: + raise ReturnConditionCheckFailed("认证接口异常", 200) + + def init_company_data(): + """初始化企业主页数据""" + init_url = "http://api.fecribd.com/etl_tfse/company/init_company_data" + init_token = "dmfd7FshT!5Wng9^gcCcQV7T6FBnVgl4" + init_headers = {'token': init_token, "Content-Type": "application/json;charset=UTF-8"} + data = json.dumps({"cid": self.cid, "company_name": self.name}) + response = requests.post(init_url, headers=init_headers, data=data) + + res = json.loads(response.text) + if res['info'] == "企业数据初始化成功": + raise ReturnConditionCheckFailed("企业数据初始化异常", 200) + else: + company_user = TFSECompanyUser() + company_user.cid = self.cid + company_user.name = self.name + company_user.verify_status = "是" + + self.db.upsert_single_data( + "应用端", + "企业用户", + {"企业ID": self.cid}, + company_user.dict_keys_toggle(columns=["企业名称", "已认证"]) + ) + + def __main__(): + get_verify_info() + check_verify_info_complete() + check_three_elements_by_tyc() + init_company_data() + + return __main__() diff --git a/ObjectsInProject/User/RealCompanyVerify.py b/ObjectsInProject/User/RealCompanyVerify.py new file mode 100644 index 0000000..415b1a4 --- /dev/null +++ b/ObjectsInProject/User/RealCompanyVerify.py @@ -0,0 +1,93 @@ +from Utils.ObjUtil import SpecObject +from Utils.ValidateUtil import ValidateAttr, Validate + + +class IDCardFront(SpecObject): + """身份证正面""" + name = ValidateAttr(field='name', type=str) + gender = ValidateAttr(field='gender', in_list=['男', '女']) + nationality = ValidateAttr(field='nationality', type=str) + birth_day = ValidateAttr(field='birth_day', func=Validate.date_format) + address = ValidateAttr(field='address', type=str) + id_num = ValidateAttr(field='id_num', type=str) + + fields_map = { + '姓名': 'name', + '性别': 'gender', + '民族': 'nationality', + '出生': 'birth_day', + '住址': 'address', + '身份证号码': 'id_num' + } + + +class IDCardBack(SpecObject): + """身份证反面""" + issuing_authority = ValidateAttr(field='issuing_authority', type=str) + validity_period = ValidateAttr(field='validity_period', type=str) + + fields_map = { + "签发机关": "issuing_authority", + "有效期限": "validity_period" + } + + +class BusinessLicense(SpecObject): + """营业执照""" + reg_num = ValidateAttr(field='reg_num', type=str) + name = ValidateAttr(field='name', type=str) + capital = ValidateAttr(field='capital', type=str) + type = ValidateAttr(field='type', type=str) + establish_date = ValidateAttr(field='establish_date', type=str) + person = ValidateAttr(field='person', type=str) + business = ValidateAttr(field='business', type=str) + address = ValidateAttr(field='address', type=str) + + fields_map = { + '统一社会信用代码': "reg_num", + '名称': "name", + '注册资本': "capital", + '类型': "type", + '成立日期': "establish_date", + '法定代表人': "person", + '经营范围': "business", + '住所': "address" + } + + +class UploadMaterial(SpecObject): + """上传材料""" + + cid = ValidateAttr(field='cid', type=str, length=8) + image = ValidateAttr(field='image', func=Validate.image, mark='图片格式') + verify_status = ValidateAttr(field='verify_status', in_list=["是", "否"], mark='认证状态') + + fields_map = { + "cid": "企业ID", + "image": "图片", + "verify_status": "已认证" + } + + def upload_id_card(self): + """上传身份证""" + + def upload_business_license(self): + """上传营业执照""" + + +class RealCompanyVerify(object): + """企业认证三要素""" + cid = ValidateAttr(field='cid', type=str, length=8) + code = ValidateAttr(field='code', type=str) + name = ValidateAttr(field='name', type=str) + legal_person = ValidateAttr(field='legal_person', type=str) + + fields_map = { + "cid": "企业ID", + "code": "统一社会信用代码", + "name": "企业名称", + "legal_person": "法人姓名", + } + + def verify(self): + """企业认证""" diff --git a/Routes/User/TFSECompanyUserRoute.py b/Routes/User/TFSECompanyUserRoute.py index f4e06b7..c1e8f29 100644 --- a/Routes/User/TFSECompanyUserRoute.py +++ b/Routes/User/TFSECompanyUserRoute.py @@ -1,11 +1,10 @@ from flask import Blueprint, request, Response +from Implements.User.RealCompanyVerifyImpl import UploadMaterialImpl, RealCompanyVerifyImpl from Implements.User.TFSECompanyUserImpl import TFSECompanyUserImpl from common.rsa import decrypt_data from Utils.AuthUtil import check_block, verify_token from Utils.ErrorUtil import ReturnConditionCheckFailed -from CompanyUser.CompanyUserImpl import EmailVerifyCodeRecordImpl, VerifyMaterialImpl, \ - CompanyVerifyThreeFactorsImpl company_user_route = Blueprint('user', __name__) @@ -139,30 +138,12 @@ def change_email(**kwargs): return {"info": "参数异常"}, 400 -@company_user_route.route('/verify_email', methods=['POST']) -@check_block -def verify_email(): - """发送邮箱验证码""" - try: - req = request.json - email_verify = EmailVerifyCodeRecordImpl() - email_verify.email = req['email'] - email_verify.v_type = req['v_type'] - email_verify.send_email() - return {"info": "邮件已发送"}, 200 - except ReturnConditionCheckFailed as e: - e.log_error() - return {"info": e.failed_info}, e.status_code - except KeyError: - return {"info": "参数异常"}, 400 - - @company_user_route.route('/upload_id_card', methods=['POST']) @verify_token def upload_id_card(**kwargs): """上传身份证""" try: - material = VerifyMaterialImpl() + material = UploadMaterialImpl() material.cid = kwargs['cid'] material.image = request.files['image'] result = material.upload_id_card() @@ -179,7 +160,7 @@ def upload_id_card(**kwargs): def upload_business_license(**kwargs): """上传营业执照""" try: - material = VerifyMaterialImpl() + material = UploadMaterialImpl() material.cid = kwargs['cid'] material.image = request.files['image'] result = material.upload_business_license() @@ -196,9 +177,9 @@ def upload_business_license(**kwargs): def company_verify_route(**kwargs): """企业认证""" try: - company_verify = CompanyVerifyThreeFactorsImpl() - company_verify.cid = kwargs['cid'] - company_verify.company_verify() + impl = RealCompanyVerifyImpl() + impl.cid = kwargs['cid'] + impl.verify() return {"info": "认证成功"}, 200 except ReturnConditionCheckFailed as e: e.log_error()