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')