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 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 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 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 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():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
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()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def create_admin(create_db):
username = "******"
password = "******"
new_user = User(username=username)
new_user.set_password(password)
db.session.add(new_user)
db.session.commit()
yield new_user
db.session.delete(new_user)
db.session.commit()
def register():
if current_user.is_authenticated:
return redirect(url_for('home'))
form = RegisterForm()
if form.validate_on_submit():
user = User(username=form.username.data,
email=form.email.data,
is_admin=True)
user.set_password(form.password.data)
# veritabanına kayıt için session'a kullanıcıyı ekle
db.session.add(user)
# veritabanına gönder
db.session.commit()
return redirect(url_for('login'))
return render_template('register.html', title="Kaydol", form=form)
def post(self):
# 解析json数据
args = user_parser.parse_args()
# 写入数据库
user = User(**args)
passwd = gen_password(10)
user.set_password(passwd)
db.session.add(user)
db_commit()
send_password(user, passwd)
return {
'id': user.id,
'account': user.account,
'name': user.name,
'is_super': user.is_super,
'is_active': user.is_active,
'email': user.email,
'phone': user.phone,
'remarks': user.remarks
}
from getpass import getpass
import os
import sys
from myapp import create_app
from myapp.models import User, db
app = create_app()
with app.app_context():
username = os.getenv('ADMIN_USER')
password = os.getenv('ADMIN_PASSWORD')
if User.query.filter(User.username == username).count():
print("Admin already exist")
sys.exit(0)
new_user = User(username=username)
new_user.set_password(password)
db.session.add(new_user)
db.session.commit()
print(f"Admin added")
def test_password_hashing(self):
u = User(username='******')
u.set_password('cat')
self.assertFalse(u.check_password('dog'))
self.assertTrue(u.check_password('cat'))
def test_anonymous_user_permissions(self):
Role.insert_roles()
u = User(username='******', email='*****@*****.**')
u.set_password('cat')
self.assertFalse(u.can(Permission.ADMINISTER))
self.assertTrue(u.can(Permission.BUY))
def test_admin_role(self):
Role.insert_roles()
u = User(username='******', email='*****@*****.**')
u.set_password('cat')
self.assertEqual(u.role.name, 'Administrator')
