def view_registration_page():
if current_user.is_authenticated:
return redirect('/')
form = RegistrationForm()
if form.validate_on_submit():
user = User(form.username.data, form.email.data, pw=form.password.data)
create_user(user)
login_user(user)
return redirect('/')
return render_template("registration.html", title="Register", form=form)
def save_user(self):
have_user = User.objects.filter(open_id=self.openid)
if have_user.exists():
# 已注册过
return self.flush_session()
sess = gen_hash()
redis_session.set(sess, self.openid, ex=259200)
user = User(open_id=self.openid)
user.save()
# 自动为用户生成Profile
# Profile(open_id=user).save()
# 注册成功，分配cookie
return {
'sess': sess,
'code': ASEC.REG_SUCCESS,
'message': ASEC.getMessage(ASEC.REG_SUCCESS)
}
def register_view():
if request.method == 'GET':
return render_template('register.html')
else:
# 获取数据
uname = request.values.get("uname")
# 从数据库查询用户
user = User.query.filter_by(name=uname).first()
if not user:
pwd = request.values.get("pwd")
# 对密码进行MD5加密
pwd = data_to_md5(pwd)
# 储存用户信息到数据库
user = User(name=uname, password=pwd, regis_date=datetime.now())
db.session.add(user)
db.session.commit()
return jsonify({"info": 'ok'})
else:
return jsonify({"info": "用户名已存在"})
def register_info():
if request.method == 'POST':
image_code = session.get('image_code')
i_code = request.form.get('i_code')
if image_code == i_code:
mobile = request.form.get('mobile')
password = request.form.get('password')
user = User.query.filter(User.phone == mobile).first()
if user:
return jsonify({'code': USER_REGISTER_NOT_PHONE})
if not re.match(r'(13\d|14[5|7]|15\d|166|17[3|6|7]|18\d)\d{8}$',
mobile):
return jsonify({'code': USER_REGISTER_PHONE_RE})
new_user = User()
new_user.phone = mobile
new_user.password = password
new_user.add_update()
return jsonify({'code': OK})
else:
return jsonify({'code': USER_REGISTER_IMAGE_CODE})
def register_insert(company_id, channel_id, gmt_create, mobile_platform):
"""
@:param mobile_platform : 所属客户端0:ios, 1:android 2 api
直接插库进行模拟用户注册
"""
# 创建user
# 查询渠道
channel = convert.query(models.Channel, id=channel_id).one()
user_cid = cid_generate()
user_mobile = fakerInstance.phone_number()
user_id = my_snow().get_next_id()
user_name = create_name_by_cid(user_cid)
user_password = hashlib.md5(
str(user_mobile).encode('utf-8')).hexdigest()[3:19]
user_detail_address = fakerInstance.address()
user_channel_name = channel.name
print("register_insert user_cid", user_cid)
print("register_insert user_mobile", user_mobile)
print("register_insert user_id", user_id)
user = User()
user.gmt_create = gmt_create
user.gmt_modified = gmt_create
user.is_deleted = 0
user.id = user_id
user.company_id = company_id
user.cid = user_cid
user.mobile = user_mobile
user.name = user_name
user.password = user_password
user.register_channel = user_channel_name
user.register_channel_id = channel_id
user.mobile_md5 = hashlib.md5(
str(user_mobile).encode('utf-8')).hexdigest()[3:19]
convert.add_one(user)
# 创建user_detail
user_detail = UserDetail()
user_detail.id = my_snow().get_next_id()
user_detail.gmt_create = gmt_create
user_detail.gmt_modified = gmt_create
user_detail.is_deleted = 0
user_detail.cert_item = None
user_detail.user_id = user_id
user_detail.company_id = company_id
user_detail.cid = user_cid
user_detail.mobile = user_mobile
user_detail.name = user_name
user_detail.password = user_password
user_detail.channel_id = channel_id
user_detail.channel_name = user_channel_name
user_detail.province = split_address(user_detail_address).get("province")
user_detail.city = split_address(user_detail_address).get("city")
user_detail.address = user_detail_address
user_detail.gender = check_sex(user_cid)
user_detail.nationality = "汉"
user_detail.birthday = create_birthday
user_detail.marital_status = None
user_detail.income_level = None
user_detail.education_level = None
user_detail.bank_account_id = None
user_detail.category = None
user_detail.mobile_platform = mobile_platform
user_detail.operator_id = None
user_detail.first_apply_time = gmt_create
user_detail.first_deduct_time = None
user_detail.loan_amount = None
user_detail.raise_amount = None
user_detail.emrg_contact_name_a = None
user_detail.emrg_contact_name_b = None
user_detail.emrg_contact_mobile_a = None
user_detail.emrg_contact_mobile_b = None
user_detail.idcard_front_img = None
user_detail.idcard_back_img = None
user_detail.thirdparty_data = None
user_detail.raise_comment = None
user_detail.update_count = None
user_detail.name_mirror = None
user_detail.raise_amount_sign = None
convert.add_one(user_detail)
# 模拟event_id
now = time.time()
_event_id = str(company_id) + "_" + str(channel_id) + "_" + str(
int(now * 1000))
print("register_insert _event_id", _event_id)
# 模拟token 取手机号的md5加密截取部分
_token = user_password
print("register_insert _event_id", _token)
return dict(mobile=user_mobile, event_id=_event_id, token=_token)
