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)
