python实战flask简易登陆注册

# coding:utf-8

import sys
reload(sys)
sys.setdefaultencoding('utf-8')#解决中文乱码问题

from flask import Flask,request,render_template
from flask_sqlalchemy import SQLAlchemy
# 引入wtf表单
from wtforms import PasswordField,StringField,Form
#引入表单验证器
from wtforms.validators import DataRequired
from models import *

class LoginForm(Form):
    Username = StringField("Username",validators=[DataRequired()])
    Password = PasswordField("Password",validators=[DataRequired])
import os
app =Flask(__name__)
app.config.from_object('config')
#指定数据库URI
app.config['SQLALCHEMY_DATABASE_URI']=os.getenv('DATABASE_URL','mysql+pymysql://root:000000@localhost:3306/test')
db = SQLAlchemy(app)


# 登录模块
@app.route('/',methods=['GET','POST'])
def login():
    form = LoginForm(request.form)#表单赋值
    message=""#初始化message
    if request.method=="POST":
        flag = Existed(form.Username.data,form.Password.data)
        if flag:
            message = "登陆成功！"
        else:
            message = "账号或密码错误！"
    return render_template("login.html",form=form,message=message)

#注册模块
@app.route('/register',methods=['GET','POST'])
def register():
    form = LoginForm(request.form)  # 表单赋值

    message = ""#初始化message
    if request.method == 'POST':
        flag = Add_user(form.Username.data, form.Password.data)
        if flag == False:
            message = "用户已存在！"
        else:
            message = "注册成功！"
    return render_template("register.html", form=form, message=message )

if __name__ == '__main__':
    app.run(debug=app.config['DEBUG'])
    #调试模式
# coding:utf-8

impor