def login(): if request.method == 'GET': return render_template('login.html') else: account = request.form['account'] password = request.form['password'] if len(account) == 0 | len(password) == 0: return render_template('login.html') cursor.execute('select account from coursefilemanagement.user') accounts = cursor.fetchall() for account in accounts: if request.form['account'] == account[0]: cursor.execute( 'select password,usertype from coursefilemanagement.user where account=%s', (account[0])) detail = cursor.fetchall() if request.form['password'] == detail[0][0]: curr_user = User() curr_user.id = account[0] print(curr_user.id) login_user(curr_user) return redirect(url_for('.homepage')) return '<h>账号、密码错误!</h>'
def new(username, password, email): """create new user.""" from app.main.models import User user = User(username=username, email=email) user.set_password(password) db.session.add(user) db.session.commit()
def api_create_user(): body = request.json if 'first_name' not in body: return bad_request('Missing first_name') if 'last_name' not in body: return bad_request('Missing last_name') if 'email' not in body: return bad_request('Missing email') if 'password' not in body: return bad_request('Missing password') user = User( first_name=body['first_name'], last_name=body['last_name'], email=body['email'], ) user.set_password(body['password']) db.session.add(user) db.session.commit() return jsonify(user._asdict())
def register(): if current_user.is_authenticated: return redirect(url_for('main.index')) form = RegistrationForm() if form.validate_on_submit(): try: if get_user_by_name(form.username.data) is not None: flash('Username already taken') return redirect(url_for('auth.register')) new_user = User( id_generator(), form.username.data, form.email.data, generate_password_hash(form.password.data, method='sha256')) save_user(new_user) login_user(new_user) flash('Congratulations, you are now a registered user!') return redirect(url_for('main.index')) except: new_user = User( id_generator(), form.username.data, form.email.data, generate_password_hash(form.password.data, method='sha256')) save_user(new_user) login_user(new_user) flash('Congratulations, you are now a registered user!') return redirect(url_for('main.index')) return render_template('auth/register.html', title='Register', form=form)
def signup(): if current_user is not None and current_user.is_authenticated(): return redirect(url_for('main.index')) registerForm = RegisterForm(request.form, prefix="registerForm") loginForm = LoginForm(request.form, prefix="loginForm") if request.method == 'POST': if request.form['submit'] == 'register' and registerForm.validate(): user = User(email=registerForm.email.data, passwd=bcrypt.generate_password_hash( registerForm.passwd.data), name=registerForm.name.data) user.url = urllib.quote_plus(registerForm.name.data) #user_url = user_make_url(registerForm.name.data) # Insert the record in our database and commit it db.session.add(user) db.session.commit() login_user(user) return redirect(request.args.get('next') or url_for('main.index')) return render_template('register.html', form="signup", loginForm=loginForm, registerForm=registerForm)
def init_app_data(): db.drop_all() db.create_all() Role.insert_roles() u = User(name='Gloomymoon', password="******", role=Role.query.filter_by(permissions=0xff).first()) db.session.add(u) u2 = User(name='Haoer', password="******", role=Role.query.filter_by(permissions=0xff).first()) db.session.add(u2) u3 = User(name='David', password="******", role=Role.query.filter_by(name="Knight").first()) db.session.add(u3) db.session.commit() s = EnglishSetting(user_id=u3.id, level1=12, level2=8, level3=3, level4=2) db.session.add(s) db.session.commit() add_book("Oxford 3A", "Oxford 3A") import_words("Oxford3A.csv") add_book("WTE 3A", "Longman Welcome to English 3A") import_words("WTE3A.csv") '''
def test_avatar(self): u = User(first_name='Test', last_name='User', email='*****@*****.**') self.assertEqual(u.avatar(128), ('https://www.gravatar.com/avatar/' '980bf7835dbbe25b0bdb9aa7f8c1a773' '?d=identicon&s=128'))
def test_avatar(self): """ 测试头像 """ u = User(user_name='guanwen', password='******', nickname='关文', email='*****@*****.**') avatar = u.avatar(128) expected = 'https://secure.gravatar.com/avatar/5e1f2c7792fea6da8f4e56b899620e5d?d=mm&s=128' assert avatar[0:len(expected)] == expected
def test_check_unique_email(self): u = User(name='john', email='*****@*****.**') db.session.add(u) db.session.commit() result = User.check_unique_email('*****@*****.**') assert result is not False result = User.check_unique_email('*****@*****.**') assert result is False result = User.check_unique_email('*****@*****.**') assert result is True
def create_new_user(name): verify_user_does_not_exist(name) new_user = User() new_user.name = name new_user.questions_total = 0 new_user.questions_won = 0 save(new_user) return get_response({'name': name})
def test_follow(self): """ 测试关注 """ u1 = User(user_name='john', password='******', nickname='john', email='*****@*****.**') u2 = User(user_name='susan', password='******', nickname='susan', email='*****@*****.**') db.session.add(u1) db.session.add(u2) assert u1.unfollow(u2) == None u = u1.follow(u2) db.session.add(u) db.session.commit() assert u1.follow(u2) == None assert u1.is_following(u2) assert u1.followed.count() == 1 assert u1.followed.first().nickname == 'susan' assert u2.followers.count() == 1 assert u2.followers.first().nickname == 'john' u = u1.unfollow(u2) assert u != None db.session.add(u) db.session.commit() assert u1.is_following(u2) == False assert u1.followed.count() == 0 assert u2.followers.count() == 0
def test_follow(self): user1 = User(first_name='Test1', last_name='User', email='*****@*****.**') user2 = User(first_name='Test2', last_name='User', email='*****@*****.**') db.session.add(user1) db.session.add(user2) db.session.commit() self.assertEqual(user1.followed.all(), []) self.assertEqual(user2.followed.all(), []) user1.follow(user2) db.session.commit() self.assertTrue(user1.is_following(user2)) self.assertEqual(user1.followed.count(), 1) self.assertEqual(user1.followed.first().first_name, 'Test2') self.assertEqual(user2.followers.count(), 1) self.assertEqual(user2.followers.first().first_name, 'Test1') user1.unfollow(user2) db.session.commit() self.assertFalse(user1.is_following(user2)) self.assertEqual(user1.followed.count(), 0) self.assertEqual(user2.followers.count(), 0)
def convert_auth0_id_to_api_id(auth0_user_id): user = User.query.filter(User.auth0_id == auth0_user_id).first() if user is None: new_user = User() new_user.auth0_id = auth0_user_id db.session.add(new_user) db.session.commit() user_id = new_user.id else: user_id = user.id return user_id
def load_user(user_id): curr_user = User() curr_user.id = user_id cursor.execute( 'select usertype ' 'from coursefilemanagement.user ' 'where account=%s', user_id) detail = cursor.fetchall() ac_type = detail[0][0] curr_user.utype = ac_type return curr_user
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('main.login')) return render_template('register.html', title='Register', form=form)
def create_all(): db.drop_all() db.create_all() from app.main.models import User from app.ABE.models import Attribute admin = User() admin.username = '******' admin.password = '******' admin.role = 'admin' db.session.add(admin) db.session.commit()
def setUp(self): self.app = create_app() self.client = self.app.test_client() self.user1 = User('*****@*****.**', 'Mohammed Rajab', '0700000000', 'aam123') self.user2 = User('*****@*****.**', 'Abdulaziz Rajab', '0701633016', 'aam786') self.rideOffer = RideOffer(self.user1.email, '07/07/2018', '1030HRS', 'madaraka', 'makina', 250) self.rideRequest = RideRequest(1, self.user2.email) with self.app.app_context(): db.create_all()
def add_user_to_database(self, users): with self.app.app_context(): self.db = SQLAlchemy() self.db.init_app(self.app) for i in range(len(users)): user = User() user.name = users[i]['name'] user.questions_won = users[i]['wins'] user.questions_total = users[i]['answers'] self.db.session.add(user) self.db.session.commit()
def register(): '''View for registering a new user.''' 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() # login ther new user login_user(user, remember='true') return redirect(url_for('main.update')) return render_template('auth/register.html', title='Register', form=form)
def register(): form = RegisterForm() if request.method == "POST": if User.query.filter_by(email=form.email.data).first(): flash(form.email.data + u'已经被注册,请选择其他邮箱!', "warning") return redirect(url_for("auth.register")) if User.query.filter_by(username=form.username.data).first(): flash(form.username.data + u'已经被注册,请选择其他用户名!', "warning") return redirect(url_for("auth.register")) _file = request.files['filename'] _type = _file.filename.split(".")[-1].lower() if not _type or _type not in ['jpeg', 'jpg', 'bmp', "png"]: flash(u"图片格式错误,当前只支持'jpeg', 'jpg', 'bmp', 'png'!", "warning") return redirect(url_for("main.edit_basic")) dirname = current_app.config['UPLOAD_FOLDER'] # 截图存放地点 user = User(email=form.email.data, username=form.username.data, password=form.password.data, about_me=form.about_me.data) db.session.add(user) db.session.commit() user_id = User.query.filter_by(username=user.username).first().id filename = secure_filename(str(user_id) + "." + _type) if not os.path.exists(dirname): try: os.makedirs(dirname) _file.save(os.path.join(dirname, filename)) except Exception as e: print(e) else: _file.save(os.path.join(dirname, filename)) user.image_name = filename db.session.add(user) db.session.commit() token = user.generate_confirmation_token() send_email([user.email], u'验证您的账号', 'auth/email/confirm', user=user, token=token) flash(u"一封验证邮件发送到了你的邮箱,请您验收!", "success") return redirect(url_for("auth.login")) return render_template("auth/register.html", form=form)
def register(): form = RegistrationForm() if form.validate_on_submit(): user = User(email=form.email.data, username=form.username.data, password=form.password.data) db.session.add(user) db.session.commit() token = user.generate_confirmation_token() send_email(user.email, 'Confirm Your Account', 'auth/email/confirm', user=user, token=token) flash('A confirmation email has been sent to you by email.') return redirect(url_for('auth.login')) return render_template('auth/register.html', form=form)
def test_new_user(client, database): """ GIVEN a User Model WHEN a new User is created THEN check the email, hashed_password, """ user = User(email='*****@*****.**', password='******') database.session.add(user) database.session.commit() assert user.email == '*****@*****.**' assert user.id >= 2 assert user.verify_password('password2') with pytest.raises(AttributeError): assert user.password
def get_followed(id): user = User.query.get_or_404(id) page = request.args.get('page', 1, type=int) per_page = min(request.args.get('per_page', 10, type=int), 100) data = User.to_collection_dict(user.followed, page, per_page, 'api.get_users') return jsonify(data)
def index(): """ Presents a "login" form to enter the thunderdome. """ form = LoginForm() if form.validate_on_submit(): form_name = form.name.data # Modify the requested username if already taken if load_user(form_name) is not None: mod = 1 while load_user(form_name + '(' + str(mod) + ')') is not None: mod += 1 form_name = form_name + '(' + str(mod) + ')' # Add username to sqlite db db.session.add(User(username=form_name)) db.session.commit() session['name'] = form_name session['room'] = 'dnd-room' return redirect(url_for('.chat')) return render_template('index.html', form=form)
def create_user(): data = request.get_json() or {} if 'username' not in data or 'email' not in data or 'password' not in data: return bad_request('must include username, email and password fields') if User.query.filter_by(username=data['username']).first(): return bad_request('please use a different username') if User.query.filter_by(email=data['email']).first(): return bad_request('please use a different email address') user = User() user.from_dict(data, new_user=True) db.session.add(user) db.session.commit() response = jsonify(user.to_dict()) response.status_code = 201 response.headers['Location'] = url_for('api.get_user', id=user.id) return response
def create_user(): form = CreateUserForm() if form.validate_on_submit(): name = form.name.data phone_number = form.phone_number.data address = form.address.data email = form.email.data pw = form.password.data user = User(name=name, phone_number=phone_number, address=address, email=email) user.set_password(pw) db.session.add(user) db.session.commit() return redirect(url_for('admin.get_users')) return render_template('admin/create_user.html', form=form)
def get_user_by_id(user_id): with open(path, 'r+') as read_file: data = json.load(read_file) read_file.close() # return user_id, data.get(user_id)['username'], data.get(user_id)['mail'], data.get(user_id)['pass'] return User(user_id, data.get(user_id)['username'], data.get(user_id)['mail'], data.get(user_id)['pass'])
def register(): if current_user.is_authenticated: return redirect(url_for('main.index')) form = RegistrationForm() if form.validate_on_submit(): user = User( first_name=form.first_name.data, last_name=form.last_name.data, email=form.email.data.lower(), ) user.set_password(form.password.data) db.session.add(user) db.session.commit() flash('Your account has been created! Yay!') return redirect(url_for('auth.login')) return render_template('auth/register.html', title='Register', form=form)
def register(): data = request.get_json() name = data["username"] email = data["email"] password = data["password"] user = User.query.filter_by(email=email).first() if user: return jsonify( success=0, results={}, message="Пользователь с такой почтой уже зарегестрирован") user = User(name=name, email=email) user.set_password(password) db.session.add(user) db.session.commit() return jsonify(success=1, results={"user_id": user.id}, message="Пользователь зарегестрирован")
def test_make_unique_nickname(self): """ 测试昵称唯一方法 """ u = User(user_name='guanwen', password='******', nickname='关文', email='*****@*****.**') db.session.add(u) db.session.commit() nickname = User.make_unique_nickname('关文') assert nickname != '关文' u = User(user_name='susan', password='******', nickname=nickname, email='*****@*****.**') db.session.add(u) db.session.commit() nickname2 = User.make_unique_nickname('关文') assert nickname2 != '关文' assert nickname2 != nickname
def database(): db.create_all() user1 = User(email='*****@*****.**', password='******') db.session.add(user1) db.session.commit() yield db db.drop_all()
def reset_password(token): if current_user.is_authenticated: return redirect(url_for('main.index')) user = User.verify_reset_password_token(token) if not user: return redirect(url_for('main.index')) form = ResetPasswordForm() if form.validate_on_submit(): user.set_password(form.password.data) db.session.commit() flash(_('Your password has been reset.')) return redirect(url_for('auth.login')) return render_template('auth/reset_password.html', form=form)
def register(): if current_user.is_authenticated: return redirect(url_for('auth.user_profile', id=current_user.id)) form = RegisterForm() if form.validate_on_submit(): name = form.name.data phone_number = form.phone_number.data address = form.address.data email = form.email.data existed_user = User.query.filter_by(email=email).first() if existed_user: flash('Email đã được sử dụng') return redirect(url_for('auth.register')) pw = form.password.data user = User(name=name, phone_number=phone_number, address=address, email=email) user.set_password(pw) db.session.add(user) db.session.commit() login_user(user) return redirect(url_for('auth.user_profile', id=user.id)) return render_template('auth/register.html', form=form)
def register_user(): json_dict = request.json print(json_dict) user = User() user.update_from_dict(json_dict, ['id']) existing_user = User.get_user_by_email(json_dict['email']) if existing_user: return Responses.OBJECT_EXIST() error = user.update() if len(error) > 0: return Responses.OPERATION_FAILED() return res(user.as_dict())
def test_password_hashing(self): user = User(username='******') user.set_password('cat') self.assertFalse(user.check_password('dog')) self.assertTrue(user.check_password('cat'))