def __call__(self, form, field): userEmail = form['userEmail'].data password = field.data # myuser = Myuser.query.filter_by(userEmail = userEmail).first() if type(MyUser.login(userEmail)) == str: raise ValueError('잘못된 이메일입니다.') elif MyUser.login(userEmail)[0] != password: print("Wrong Password") raise ValueError('잘못된 패스워드입니다.')
def hello(): userEmail = session.get('userEmail', None) if userEmail: userName = MyUser.login(userEmail)[1] return render_template('index.html', userName=userName) return render_template('index.html')
def login(w): if w == 'login': form = LoginForm() mode = 'login' if form.validate_on_submit(): session['userEmail'] = form.data.get('userEmail') userEmail = session['userEmail'] return redirect('/') elif w == 'register': form = RegisterForm() mode = 'register' if form.validate_on_submit(): userEmail = form.data.get('userEmail') userName = form.data.get('userName') password = form.data.get('password') userInfo = { 'userEmail': userEmail, 'userName': userName, 'password': password } if MyUser.login(userEmail) == 'Wrong Email': db.userInfo.insert_one(userInfo) return redirect('/user/login') else: return redirect('/user/register') return render_template('login.html', form=form, mode=mode)