import sqlite3 as sql

#连接数据库，若不存在则自动创建stu.db
conn = sql.connect("stu.db")
#创建cursor
cur = conn.cursor()
#创建students表单
create_table = ‘‘‘
    CREATE TABLE students(
    username TEXT PRIMARY KEY NOT NULL,
    number INT NOT NULL,
    college TEXT NOT NULL,
    major TEXT NOT NULL,
    password TEXT NOT NULL,
    is_admin NUMERIC DEFAULT 0 NOT NULL)
‘‘‘
#执行数据库语句
cur.execute(create_table)
print("Table created")
#提交更改
conn.commit()
#关闭连接
conn.close()

from flask import *
import sqlite3 as sql
app = Flask(__name__)


@app.route("/")
def index():
    return render_template("index.html")

@app.route("/login")
def login():
    return render_template("login.html")
# @app.route("/login_check")
# def login_check():


@app.route("/register",methods=["GET","POST"])
def register():
    if request.method == "POST":
        try:
            username = request.form["username"]
            number = request.form["number"]
            college = request.form["college"]
            major = request.form["major"]
            password = request.form["password"]
            password2 = request.form["password2"]
            if password != password2:
                msg = "密码不一致"
                return render_template("result.html",msg=msg)
            else:
                with sql.connect("stu.db") as conn:
                    cur = conn.cursor()
                    cur.execute("INSERT INTO students(username,number,college,major,password) VALUES(?,?,?,?,?)",(username,number,college,major,password))
                    conn.commit()
                    conn.close()
                    msg = "注册成功"
        except:
            conn.rollback()
            msg = "注册失败"
        finally:
            return render_template("result.html",msg=msg)
    else:
        return render_template("register.html")






@app.route("/admin_login")
def admin_login():
    return render_template("admin_login.html")

@app.route("/list")
def list():
    return render_template("list.html")

if __name__ == "__main__":
    app.run(debug=True)