from flask import Flask, render_template, Response, request
from apps.service.LoginService import LoginService
from apps.pojo.Entity import db
from apps.controller.UserController import user_bp

app = Flask(__name__)
# 注册蓝图对象
app.register_blueprint(user_bp)

# 设置数据库连接的参数
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:123456@localhost:3306/flask'

# 此处初始化数据表
with app.app_context():
    db.init_app(app)
    # db.drop_all()
    # db.create_all()



@app.route('/')
def tologin():
    return render_template('login.html')


@app.route('/login', methods=['POST'])
def dologin():
    param = request.form
    username = param.get('username')
    password = param.get('password')
    # 调用登录方法
    user = LoginService.login(username, password)
    if user:
        return render_template('index.html', user=user)
    else:
        return render_template('login.html', msg='用户名或密码错误')


@app.route('/index_v1')
def index_v1():
    return render_template('index_v1.html')  # 渲染模板文件


@app.route('/index_v3')
def index_v3():
    return render_template('index_v3.html')

if __name__ == '__main__':
    app.run()