def new_users_same_password():
user_1 = User(username='******')
user_2 = User(username='******')
user_1.set_password('test_password')
user_2.set_password('test_password')
yield user_1, user_2
db.session.delete(user_1)
db.session.delete(user_2)
db.session.commit()
def register(request):
if request.method == "POST":
username = request.POST['username']
email = request.POST['email']
password = request.POST['password']
confirmpassword = request.POST['confirmpassword']
# password = request.POST['username']
if len(username) > 10:
messages.error(request, "Username must be under the 10 characters")
return redirect(register)
if password != confirmpassword:
messages.error(request,
"new password or confirm password not match")
return redirect(register)
else:
password = make_password(password)
user = User(username=username, email=email, password=password)
subject = "Greeting"
message = "successfully account created"
to = email
res = send_mail(subject, message, settings.EMAIL_HOST_USER, [to])
if res == 1:
print("mail send successfully")
else:
print("mail not sent")
user.save()
return redirect(login)
return render(request, "myapp/register.html")
def main():
from myapp.models import User
i = 0
all = []
for line in f:
i += 1
data = ''
try:
data = json.loads(line)
except:
print i
error.write(str(i) + '\n')
error.flush()
break
all.append(
User(user_name=data['user_name'],
id=data['id'],
nick_name=data['nick_name'],
signup_time=data['signup_time'],
ban=data['ban'],
avater=data['avater']))
if i % 5000 == 0:
print i
User.objects.bulk_create(all)
all = []
User.objects.bulk_create(all)
f.close()
error.close()
def seed_all():
from database import db
from myapp.models import User
# Create a user
db.session.add(User(email="*****@*****.**", password="******"))
db.session.commit()
def InsertUser(request):
try:
user=User()
user.title=request.POST['title']
photo=request.FILES.get('mypic',None)
user.photo=photo.name
if not photo:
return HttpResponse("没有上传图片")
filename = str(photo.name)
destination = open("./static/myapp/pics/" + filename, "wb+")
for chunk in photo.chunks():
destination.write(chunk)
destination.close()
# 执行图片缩放
im = Image.open("./static/myapp/pics/" + filename)
# 缩放到75*75（缩放后的宽高比例不变）
im.thumbnail((192, 120))
# 把缩放后的图片用jpeg格式保存
im.save("./static/myapp/pics/s_" + filename, None)
user.createtime=datetime.now()
user.save()
context = {"info": "添加成功"}
except:
context = {"info": "添加失败"}
return render(request,'myapp/info.html',context)
def test_user_to_json(self):
# No database dependencies here :)
user = User(first_name='blah', username='******', id='blah')
json_dict = user.to_json()
self.assertEqual(user.first_name, json_dict['first_name'])
self.assertEqual(user.username, json_dict['username'])
self.assertEqual(user.id, json_dict['id'])
def register(username, passwd):
if re.search(disallowed_chars,
username) or len(username) > 20 or get_user(username):
raise Forbidden
user = User(username, passwd)
session.commit()
return user
def register(request):
if request.method == "GET":
return render(request, 'register.html')
elif request.method == "POST":
account = request.POST.get('account')
password = request.POST.get('password')
username = request.POST.get('nickname')
passwd = request.POST.get('passwd')
verifycode = request.POST.get('verifycode')
verifycode = verifycode.upper()
print(print("verifycode:{}".format(verifycode)))
print(print("rand_str:{}".format(rand_str)))
print(verifycode == rand_str)
if verifycode != rand_str:
return HttpResponse('验证码错误')
if (passwd == password) and password and username and (verifycode
== rand_str):
try:
user = User()
user.name = username
user.password = generate_password(password)
user.token = uuid.uuid3(uuid.uuid4(), 'username')
user.account = account
user.save()
response = redirect('paixienet:mainweb')
response.set_cookie('token', user.token)
return response
except Exception as e:
return HttpResponse('注册失败')
else:
return HttpResponse('注册失败')
def add_user(request):
if request.method == 'POST':
try:
payload = json.loads(request.body)
username = payload['username']
email = payload['email']
password = payload['password']
avatar = payload['avatar']
user_id = uuid.uuid4().hex[:10]
hashed_password = make_password(password)
user = User(user_id, username, email, hashed_password, avatar)
user_query = User.objects.filter(username=username).exists()
email_query = User.objects.filter(email=email).exists()
if user_query:
response = json.dumps([{'error': 'username exists'}])
elif email_query:
response = json.dumps([{'error': 'email exists'}])
else:
user.save()
response = json.dumps([{
'success': 'User added successfully!'
}])
except:
response = json.dumps([{'error': 'User could not be added!'}])
return HttpResponse(response, content_type='text/json')
else:
response = json.dumps([{'error': 'User could not be added!'}])
return HttpResponse(response, status='404', content_type='text/json')
def registerok(request):
email = request.POST.get("text1")
psd = request.POST.get("password")
print(psd)
sender = '*****@*****.**'
key = request.POST.get("key")
request.session['key'] = key
print(key)
url = "http://127.0.0.1/registerlock?username=%s&key=%s" % (email, key)
# stmp.stmpemail(sender,email,url)
subject = '旺享笔记注册'
message = '感谢您注册旺享笔记,这是您的注册激活链接:%s' % url
print(message)
sender = '*****@*****.**'
receivers = [email]
try:
send_mail(subject, message, sender, receivers, fail_silently=False)
print("发送成功!")
except:
return render(request, 'registerfail.html')
user = User(username=email,
psd=keyrandom.psdmd5(psd),
islock=True,
time=timezone.now())
user.save()
return render(request, 'registerok.html')
def signup():
if request.method == "POST":
parsejson = request.get_json()
newuser_data = {
"email": parsejson["email"],
"username": parsejson["username"],
"password": parsejson["password"],
"confirmpassword": parsejson["confirmpassword"]
}
errors = signup_validator(newuser_data)
if errors is not None:
for x in errors.values():
return x
checkuser = User.query.filter_by(
username=newuser_data["username"]).first()
if checkuser is not None:
return jsonify({"Error": "username already exist"
}), HTTP_400_BAD_REQUEST
psw_hash = bcrypt.generate_password_hash(
newuser_data['password']).decode("utf8")
newuser = User(email=newuser_data["email"],
username=newuser_data["username"],
password=psw_hash)
db.session.add(newuser)
db.session.commit()
return jsonify({"Message": "registration successful"}), HTTP_200_OK
def register():
if current_user.is_authenticated:
return redirect(
url_for('main.home')
) #si le user est authentifié il est renvoyé à la page d'acceuil
form = RegistrationForm()
if form.validate_on_submit(
): #si il valide son enregistrement, ses donnnées sont envoyées à la DB
hashed_password = bcrypt.generate_password_hash(
form.password.data).decode('utf-8')
user = User(username=form.username.data,
email=form.email.data,
password=hashed_password,
admin=True,
confirmed=True,
publisher=False)
db.session.add(user)
db.session.commit()
token = user.generate_confirmation_token()
send_mail(user.email,
'Confirm your Account',
'registration_confirmation',
user=user,
token=token)
flash(
'A confirmation mail has been sent to you ! Check your mail box and confirm your account please !',
'success')
return redirect(url_for('users.login'))
return render_template('register.html', title='Register', form=form)
def registe(request):
if request.method == 'GET':
return render(request, 'mine/registe.html')
elif request.method == 'POST':
user = User()
user.account = request.POST.get('account')
user.password = genarate_password(request.POST.get('password'))
user.name = request.POST.get('name')
user.phone = request.POST.get('phone')
user.addr = request.POST.get('addr')
# 头像
img_name = user.account + '.png'
imagePath = os.path.join(settings.MEDIA_ROOT, img_name)
file = request.FILES.get('icon')
with open(imagePath, 'wb') as fp:
for data in file.chunks():
fp.write(data)
user.img = img_name
user.token = str(uuid.uuid5(uuid.uuid4(), 'register'))
user.save()
# 状态保持
request.session['token'] = user.token
# 重定向
return redirect('myapp:mine')
def register(request):
role = Role.objects.get(name="user")
if request.method == "POST":
print("hello")
# first_name = request.POST.get('first_name')
# last_name = request.POST.get('last_name')
email = request.POST.get('email')
password = request.POST.get('password')
confirm_passsword = request.POST.get('confirm_passsword')
role_id = request.POST.get('role_id')
roleid = Role.objects.get(pk=role_id)
if password != confirm_passsword:
print(password)
messages.error(request,
"new password or confirm password not match")
return redirect(register)
else:
password = make_password(password)
user = User(username=email,
email=email,
password=password,
role_id=roleid)
user.save()
return redirect(login)
print("hiiii")
return render(request, 'user/registration.html', {'role': role})
def init_database():
# Create the database and the database table
db.create_all()
# Insert user data
user1 = User(username='******', plaintext_password='******')
user2 = User(email='anotheruser', plaintext_password='******')
db.session.add(user1)
db.session.add(user2)
# Commit the changes for the users
db.session.commit()
yield db # this is where the testing happens!
db.drop_all()
def test_user_set_password():
user = User(username="******")
assert user.salt == None
assert user.password_hash == None
user.set_password("test_password")
assert user.salt
assert user.password_hash and user.password_hash != "test_password"
def register():
error = None
#
# if User.query.filter_by(email=request.form['email']).count()==0:
# user = User(
# email=request.form["email"],
# username=request.form["username"],
# password=request.form["password"],
# register_time=datetime.utcnow()
# )
# db.session.add(user)
# db.session.commit()
# return redirect(url_for('auth.login'))
# else:
# return jsonify({'error':'email already been use!'})
form = RegistrationForm()
if form.validate_on_submit():
user = User(email=form.email.data,
username=form.username.data,
password=form.password.data,
register_time=datetime.utcnow())
db.session.add(user)
db.session.commit()
flash('You can now login.')
return redirect(url_for('main.login'))
return render_template('auth/register.html', form=form)
def listuser(request):
# return render(request, 'template3.html', {'data': User.objects.all()})
if request.method == 'GET':
userid = request.GET.get('userid','')
if userid != "":
# User.objects.all().get(userid = userid).delete()
User.objects.all().filter(userid = userid)[0].delete()
return redirect('/listuser')
q = request.GET.get('q', '')
data = User.objects.all()
if q != '' :
data = User.objects.all().filter(name__contains = q)
return render(request, 'template2.html', {'data': data})
else:
Userid = request.POST.get('Userid')
Name = request.POST.get('Name')
Age = int(request.POST.get('Age'))
Hobby = request.POST.get('Hobby')
user = User(userid = Userid, name = Name, age = Age, hobby = Hobby)
user.save()
return redirect('/listuser')
def admin_user(db):
user = User(username='******', email='*****@*****.**', password='******')
db.session.add(user)
db.session.commit()
return user
def create_user():
if 'username' not in request.json or 'first_name' not in request.json:
return Response('Missing required arguments', status=400)
user = User(first_name=request.json['first_name'],
username=request.json['username'])
DB.session.add(user)
DB.session.commit()
return jsonify(user.to_json())
def register_api():
if request.method == "GET":
#此处可以重定向到登录页面
return "注册成功"
else:
#解析参数
params = request.form
u_phone = params.get("u_phone")
u_name = params.get("u_name")
u_password = params.get("u_password")
email = params.get("email")
u_icon = request.form.get("u_icon", None)
#校验参数
if u_name and u_password and u_phone and len(u_name) > 3 and len(
str(u_phone)) == 11:
#校验手机号
enc_phone = User.query.filter(User.u_phone == u_phone).all()
#若为True 则手机号注册过
if len(enc_phone) == 0:
# 给密码加密
enc_pwd_str = enc_pwd(u_password)
#实例化
u = User()
u.u_name = u_name
u.u_password = enc_pwd_str
#校验用户头像，若为None，则没上传
if u_icon == None:
return "没上传头像，请上传头像"
else:
#生成唯一头像名字
file_name = create_rand_str() + ".jpg"
#拼接文保存路径
file_path = os.path.join(BASE_DIR,
'static/icon' + file_name)
#保存文件
file = open(file_path, "wb")
for i in u_icon.chunks():
u.u_icon = file_path
#保存文件路径
#将用户对象保存到数据库
db.session.add(u)
db.session.commit()
res = send_mail(email, request.host)
# 设置缓存
cache.set(res, u.id, 60 * 60)
data = {"code": SUCCESS, "msg": '注册成功', "data": []}
return jsonify(data)
else:
#手机号已被注册
return jsonify({
"code": FAIL,
"msg": "手机号已被注册，请换个手机号重新注册",
"data": []
})
else:
#参数不合适
return jsonify({"code": NOT_LOGIN, "msg": "参数不对，请核对", "data": []})
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
return redirect(url_for('login'))
return render_template('register.html', title="Kaydol", form=form)
def signup():
if session.get("user_name"):
flash(
f"You are already a user {session.get('user_name')}. You need to logout if you want to create a new account!!",
category="success")
return redirect(url_for("mytodos"))
if request.method == "POST":
form_return_data = [
request.form.get("username"),
request.form.get("email"),
request.form.get("password"),
request.form.get("repeatpassword")
]
email_list = []
for i in User.query.all():
email_list.append(i.get("email"))
if request.form.get("email") not in email_list:
if request.form.get("password") == request.form.get(
"repeatpassword"):
user_list = []
for i in User.query.all():
user_list.append(i.get("name"))
if request.form.get("username") not in user_list:
try:
new_user = User(name=request.form.get("username"),
email=request.form.get("email"),
password="")
new_user.set_password(request.form.get("password"))
db.session.add(new_user)
db.session.commit()
session['redirect_to'] = "home"
return redirect(url_for("login"))
except Exception as e:
print(e)
flash(
"some error happened while trying to create your User Model!! Try Again",
category="model")
else:
flash("your username is already taken!!", category="user")
else:
flash("there was some error in your password!!",
category="error")
flash("your passwords don't match!!", category="password")
else:
flash("Sorry but there was some error with your email!!",
category="error")
return render_template("signup.html", form_data=form_return_data)
return render_template("signup.html")
def test_user_set_salt():
user = User(username="******")
assert user.salt == None
user._set_salt()
assert user.salt != None
old_salt = user.salt
user._set_salt()
new_salt = user.salt
assert old_salt != new_salt
def registerPage(request):
if request.method == "GET":
return render(request, "register.html")
else:
email1 = request.POST["email"]
password1 = make_password(request.POST["password"])
b = User(email=email1, password=password1)
b.save()
return redirect("/login")
def import_reviews_users():
print('Indexing reviews... look at progress:')
aux = find_url_user_review()
User.objects.all().delete()
Review.objects.all().delete()
reviews = []
users = []
for line in aux:
try:
name = line[0]
photo = line[1]
users.append(User(name=name, photo=photo))
except:
e = sys.exc_info()[0]
print("Error when creating a review or user: {0}".format(e))
names = []
usersnew = []
for user in users:
if user.name in names:
names = names
else:
names.append(user.name)
usersnew.append(user)
User.objects.bulk_create(usersnew)
for line in aux:
try:
rate = line[2]
visit_date = line[3]
title = line[4]
description = line[5]
restaurant_name = line[6]
user_name = line[0]
if user_name == "jcarazo":
st = "falla"
restaurant_id = Restaurant.objects.filter(
name=restaurant_name)[0].id
print("Restaurante id" + str(restaurant_id))
user_id = User.objects.filter(name=user_name)[0].id
print("User id" + str(user_id))
reviews.append(
Review(user_id=user_id,
restaurant_id=restaurant_id,
rate=rate,
visit_date=visit_date,
title=title,
description=description))
except:
e = sys.exc_info()
print("Error when creating a review or user: {0}".format(e))
Review.objects.bulk_create(reviews)
def create(self):
form = Registration(request.form)
if request.method == 'POST' and form.validate():
user = User(username=form.username.data,
password=generate_password_hash(form.password.data))
db.session.add(user)
db.session.commit()
flash('You are registered.')
return redirect('/users/login')
return render_template('users/create.html', form=form)
def new_user(create_db):
user = User(username='******')
user.set_password('test_password')
db.session.add(user)
db.session.commit()
yield user
db.session.delete(user)
db.session.commit()
def register():
form = RegisterForm()
# On GET request, return false
if form.validate_on_submit():
flash('Registered visitor: {}'.format(form.user_name.data))
visitor = User(user_name = form.user_name.data)
session.add(visitor)
session.commit()
return redirect(url_for('index'))
return render_template('register.html', form=form)
def signup():
if request.method == 'POST':
if User.query.filter_by(
uname=request.form['username']).first() is None:
db.session.add(User(request.form['username'], request.form['pwd']))
db.session.commit()
return redirect(url_for('signin'))
else:
return '0'
else:
return render_template('signup.html')
