def reg(request):
#注册用户
#对应的网页模版地址为 mysite/templates/mysite/reg.html
if chkAdminCookies(request) == None:
return msg(request, "mysite:index", "无权限查看！")
if request.method == 'POST':
form = RegForm(request.POST)
if form.is_valid():
username = request.POST['username'].strip()
idsn = request.POST['idsn'].strip()
phone = request.POST['phone'].strip()
type = int(request.POST['type'])
qry_usrs = User.objects.all().filter(idsn = idsn)
if len(qry_usrs) != 0:
qry_usrs[0].username = username
qry_usrs[0].phone = phone
qry_usrs[0].save()
msg(request, "mysite:reg", '用户已存在并重新修改！')
newusr = User(username = username, idsn = idsn, phone = phone, type = type, suffix = idsn[14:18])
newusr.save()
return msg(request, "mysite:reg", "注册成功！")
return msg(request, "mysite:reg", "关键项未填!")
form = RegForm()
usrs = User.objects.all()
c = {
'form': form,
'usrs': usrs,
}
return render_to_response('mysite/reg.html', c, context_instance=RequestContext(request))
class Restrict(TestCase):
def setUp(self):
self.my_admin_data = {'username': '******', 'password': '******'}
self.my_admin = User(username=self.my_admin_data['username'],
is_staff=True,
is_superuser=True)
self.my_admin.set_password(self.my_admin_data['password'])
self.my_admin.save()
def test_get_as_anonymous_user(self):
response = self.client.get(reverse(f'{APP_NAME}:restrict'))
self.assertEqual(response.status_code, 302)
def test_get_as_anonymous_user_following_redirects(self):
response = self.client.get(reverse(f'{APP_NAME}:restrict'),
follow=True)
self.assertEqual(response.status_code, 200)
def test_get_as_superuser(self):
self.client.login(username=self.my_admin_data['username'],
password=self.my_admin_data['password'])
response = self.client.get(reverse(f'{APP_NAME}:restrict'))
self.assertEqual(response.status_code, 200)
def handlereg(request):
c={}
username=request.POST['username']
password=request.POST['password']
email=request.POST['email']
if check_format(request) == 0:
return msg(request,"mysite:reg","注册格式不对！")
exist_user_list = User.objects.all().filter(
username=username
)
if len(exist_user_list) != 0:
return msg(request,"mysite:reg","用户已经存在.")
exist_user_list = User.objects.all().filter(
email=email
)
if len(exist_user_list) != 0:
return msg(request,"mysite:reg","邮箱已经存在.")
user = User(username=username,password=password,email=email)
user.save()
promo = Promo(user=user)
promo.save()
exist_ip_list = Ip.objects.all().filter(ipaddr=get_client_ip(request))
if len(exist_ip_list) !=0 and exist_ip_list[0].reged==False:
theboss = exist_ip_list[0].p_user
theboss.promo.invites += 1
theboss.promo.save()
theboss.balance += 10
theboss.save()
exist_ip_list[0].reged=True
exist_ip_list[0].save()
request.session['username'] = user.username
request.session['password'] = user.password
return msg(request,'mysite:account', "注册成功！")
def setUp(self):
self.my_admin_data = {'username': '******', 'password': '******'}
self.my_admin = User(username=self.my_admin_data['username'],
is_staff=True,
is_superuser=True)
self.my_admin.set_password(self.my_admin_data['password'])
self.my_admin.save()
def checkUser(self, username, realname, email):
user = self.findUser(username)
if user:
return user
parts = realname.split()
if len(parts) == 1:
first_name = parts[0]
last_name = ""
else:
first_name = parts[0]
last_name = parts[1]
user = User(username=username, email=email,
first_name=first_name, last_name=last_name)
user.save()
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data.lower(), email=form.email.data.lower())
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Вы успешно зарегистрировались!', 'success')
return redirect(url_for('index'))
return render_template('register.html', title='Регистрация', form=form)
def register():
form = RegistrationForm()
if form.validate_on_submit():
if form.username.data != db.session.query(
User.username).filter_by(username=form.username.data).scalar():
if form.email.data != db.session.query(
User.email).filter_by(email=form.email.data).scalar():
user = User(email=form.email.data,
username=form.username.data,
password=form.password.data,
is_active=False)
db.session.add(user)
db.session.commit()
token = generate_confirmation_token(user.email)
confirm_url = url_for('users.confirm_email',
token=token,
_external=True)
html = render_template('activate.html',
confirm_url=confirm_url)
subject = "Please confirm your email"
send_email(form.email.data, subject, html)
#login_user(user)
flash('A confirmation email has been sent via email.',
'success')
return redirect(url_for('users.login'))
else:
flash("Email Already Registered")
else:
flash("Username Already Exists")
return render_template('register.html', form=form)
def register():
form = UserForm(request.form)
if request.method == "POST" and form.validate():
user = User.query.filter_by(email=form.email.data).first()
if user:
flash(u"该邮箱已被注册过了，请登录", "warning")
return redirect(url_for("user.login_phone"))
else:
user = User(email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
login_user(user)
flash(u"注册成功，系统已为您自动登录", "success")
return redirect(request.args.get("next") or url_for("index"))
return render_template("user/register.html", form=form)
def hello(request):
now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
name = '%s' % ('show')
users = User.objects.all().filter(user_name='admin').values()
# 获取model执行的sql语句
sql = users.query
'''
SELECT
`mysite_user`.`id`,
`mysite_user`.`user_id`,
`mysite_user`.`user_name`,
`mysite_user`.`email`,
`mysite_user`.`cellphone`
FROM
`mysite_user`
WHERE
`mysite_user`.`user_name` = username
'''
user = User()
user.user_name = 'admin'
user.user_id = 'admin001'
user.email = '*****@*****.**'
user.cellphone = '13500008888'
user.password = md5('password')
md5str = md5('123456')
# user.save()
# request =' %s', % (request)
host = request.get_host()
return TemplateResponse(request, 'hello.html', locals())
class NoticeAdminOnly(TestCase):
def setUp(self):
self.my_admin_data = {'username': '******', 'password': '******'}
self.my_admin = User(username=self.my_admin_data['username'],
is_staff=True,
is_superuser=True)
self.my_admin.set_password(self.my_admin_data['password'])
self.my_admin.save()
def test_get_as_anonymous_user(self):
response = self.client.get(reverse(f'{APP_NAME}:notice_admin_only'))
self.assertEqual(response.status_code, 200)
def test_get_as_superuser(self):
self.client.login(username=self.my_admin_data['username'],
password=self.my_admin_data['password'])
response = self.client.get(reverse(f'{APP_NAME}:notice_admin_only'))
self.assertEqual(response.status_code, 200)
def reg(request):
#注册用户
#对应的网页模版地址为 mysite/templates/mysite/reg.html
if request.method == 'POST':
username = request.POST['username']
password = request.POST['password']
form = RegForm(request.POST)
if form.is_valid():
qry_usrs = User.objects.all().filter(username=username)
if len(qry_usrs) != 0:
return HttpResponse('用户已存在！')
newusr = User(
username=request.POST['username'], password=request.POST['password'])
newusr.save()
return HttpResponseRedirect(reverse('mysite:reg'))
form = RegForm()
usrs = User.objects.all()
c = {
'form': form,
'usrs': usrs,
}
return render_to_response('mysite/reg.html', c, context_instance=RequestContext(request))
def register():
if current_user.is_authenticated:
return redirect(url_for("home"))
form = RegistrationForm()
if form.validate_on_submit():
h_pass = bcrybt.generate_password_hash(
form.password.data).decode("utf-8")
user = User(name=form.username.data,
password=h_pass,
email=form.email.data)
db.session.add(user)
db.session.commit()
flash(f'Account created for {form.username.data}!', 'success')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def reset_token(token):
if current_user.is_authenticated:
return redirect(url_for("home"))
user = User.verify(token)
if user is None:
flash("this token is invalid", "warning")
return redirect(url_for("reset_password"))
form = Reset_password()
if form.validate_on_submit():
h_pass = bcrybt.generate_password_hash(
form.password.data).decode("utf-8")
user.password = h_pass
db.session.commit()
flash('ur password has been updated!', 'success')
return redirect(url_for('login'))
return render_template("reset_token.html",
title="reset password",
form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for('home'))
form = RegistrationForm()
if form.validate_on_submit():
hashed = bcrypt.generate_password_hash(
form.password.data).decode('utf-8')
user = User(username=form.username.data,
email=form.email.data,
password=hashed)
db.session.add(user)
db.session.commit()
flash('Sua conta foi criada com sucesso.', 'success')
return redirect(url_for('login'))
popular = Post.query.order_by(desc(Post.total_score)).limit(3)
return render_template('register.html',
title='Register',
form=form,
popular_posts=popular)
def register():
if current_user.is_authenticated:
return redirect(url_for('home'))
else:
form = RegistrationForm()
if form.validate_on_submit():
hashed_password = bcrypt.generate_password_hash(
form.password.data).decode('UTF-8')
user = User(
first_name=form.first_name.data,
last_name=form.last_name.data,
email=form.email.data,
password=hashed_password,
)
db.session.add(user)
db.session.commit()
flash(f'Your account has been created! You now able to log in',
'success')
return redirect(url_for('login'))
else:
return render_template('register.html',
title='Register',
form=form)
def userReg(request):
if request.method == 'POST':
verify_code = request.session['verify_code']
input_verify = request.POST['verify']
returndata = {}
# 检查验证码
if verify_code.lower() != input_verify.lower():
returndata = {'code': 100, 'msg': '验证码错误'}
# 检查用户名是否存在
userinfo = User.objects.filter(
user_name=request.POST['username']).exists()
if userinfo:
returndata = {'code': 101, 'msg': '用户名已存在'}
if returndata:
return HttpResponse(json.dumps(returndata), 'application/json')
user = User()
user.user_name = request.POST['username']
user.password = md5(request.POST['password'])
user.save()
returndata = {'code': 200, 'msg': '注册成功'}
return HttpResponse(json.dumps(returndata), 'application/json')
else:
return TemplateResponse(request, 'reg.html')
def forward(weixinhao):
weixin = Weixin.query.filter_by(name=weixinhao).first_or_404()
if verification(request, weixin.random):
msg = parse_msg(request.data)
openid = msg["FromUserName"]
if is_subscribe_event(msg):
user = User.query.filter(db.and_(User.weixinhao==weixinhao, User.openid==openid)).first()
if user:
user.state = True
else:
user = User(weixinhao=weixinhao, openid=openid)
db.session.add(user)
db.session.commit()
return response_text_msg(openid, weixinhao, welcome_text)
if is_unsubscribe_event(msg):
user = User.query.filter(db.and_(User.weixinhao==weixinhao, User.openid==openid)).first_or_404()
user.state = False
db.session.add(user)
db.session.commit()
if is_text_msg(msg):
content = msg["Content"]
#request for help
if content[0] == u"?" or content[0] == u"？ ":
return response_text_msg(openid, weixinhao, help_text)
#request for login
if content[0] == u"#":
random = msg["Content"][1:]
# if random in mc:
# mc[random] = openid
# return response_text_msg(openid, weixinhao, u"登陆验证成功")
# else:
# return response_text_msg(openid, weixinhao, u"未知随机码")
# #request for disease
# if content[0:2] == u"疾病" or content[0] == u"j" or content[0] == u"J":
# if u" " in content:
# diseases = []
# disease = Disease.query.filter_by(name=content.split(" ")[1]).first()
# if disease:
# diseases.append(disease)
# else:
# diseases = Disease.query.filter(Disease.name.like("%"+content.split(" ")[1]+"%")).limit(10).all()
# items = []
# if diseases:
# for disease in diseases:
# items.append({
# "title": disease.name,
# "description": disease_text % (disease.name, disease.people, disease.bodypart, disease.symptom, disease.department),
# "url": "http://xianglian.sinaapp.com" + url_for("disease.detail", diseaseId=disease.id)
# })
# return response_news_msg(openid, weixinhao, items)
# else:
# return response_text_msg(openid, weixinhao, not_find_text)
# else:
# return response_text_msg(openid, weixinhao, help_text)
return response_text_msg(openid, weixinhao, help_text)
return response_text_msg(openid, weixinhao, help_text)
return "message processing fail"
