目录结构(mvc模式)
样式文件/static/style.css
div{
margin: 0 auto;
width: 500px;
height: 400px;
border: 1px solid red;
text-align: center;
}
配置文件设计/templates/config.py
#数据库连接配置
import pymysql
conn = pymysql.connect(
host='192.XXX.XXX.XX',
port=320xx,
user='root',
password='123456',
database='test_XX'
)
首页/templates/index.html
DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<link rel="stylesheet" type="text/css" href="/static/style.css">
<title>林家小猪测试小站title>
head>
<body>
<div>
<h1>您好,{{ username }},欢迎来到我的小站h1>
<a href="{{ url_for('user_login') }}">退出a>
<br/>
div>
body>
html>
登录页面/templates/login.html
DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<link rel="stylesheet" type="text/css" href="/static/style.css">
<title>登录title>
head>
<body>
<div>
<h1>用户登录h1>
<form method="POST">
<input type="text" name="username" placeholder="请输入用户名" />
<br/>
<input type="password" name="password" placeholder="请输入密码(小于12位)" />
<br/>
{% if message %} {{message}} {% endif %}
<br/>
<input type="submit" value="登录" />
<input type="reset" value="重置" />
<a href="{{ url_for('register') }}">注册a>
form>
div>
body>
html>
注册页面/templates/register.html
DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<link rel="stylesheet" type="text/css" href="/static/style.css">
<title>注册title>
head>
<body>
<div>
<h1>用户注册h1>
<form method="POST">
<input type="text" name="username" placeholder="请输入用户名" />
<br/>
<input type="password" name="password" placeholder="请输入密码(小于12位)" />
<br/>
{% if message %} {{message}} {% endif %}
<br/>
<input type="submit" value="注册" />
<input type="reset" value="重置" />
<a href="{{ url_for('user_login') }}">登录a>
form>
div>
body>
html>
登录校验 /model/check_login.py
from templates.config import conn
cur = conn.cursor()
def is_null(username,password):
if(username==''or password==''):
return True
else:
return False
def is_existed(username,password):
sql="SELECT * FROM user WHERE username ='%s' and password ='%s'" %(username,password)
cur.execute(sql)
result = cur.fetchall()
if (len(result) == 0):
return False
else:
return True
def exist_user(username):
sql = "SELECT * FROM user WHERE username ='%s'" % (username)
cur.execute(sql)
result = cur.fetchall()
if (len(result) == 0):
return False
else:
return True
注册校验 /model/regist_login.py
from templates.config import conn
cur = conn.cursor()
def add_user(username, password):
# sql commands
sql = "INSERT INTO user(username, password) VALUES ('%s','%s')" %(username, password)
# execute(sql)
cur.execute(sql)
# commit
conn.commit() # 对数据库内容有改变,需要commit()
conn.close()
最后编辑运行文件
app.py
from flask import Flask,render_template
from flask import redirect
from flask import url_for
from flask import request
from model.check_login import is_existed,exist_user,is_null
from model.check_regist import add_user
app = Flask(__name__)
@app.route('/')
def index():
return redirect(url_for('user_login'))
@app.route('/user_login',methods=['GET','POST'])
def user_login():
if request.method=='POST': # 注册发送的请求为POST请求
username = request.form['username']
password = request.form['password']
if is_null(username,password):
login_massage = "温馨提示:账号和密码是必填"
return render_template('login.html', message=login_massage)
elif is_existed(username, password):
return render_template('index.html', username=username)
elif exist_user(username):
login_massage = "温馨提示:密码错误,请输入正确密码"
return render_template('login.html', message=login_massage)
else:
login_massage = "温馨提示:不存在该用户,请先注册"
return render_template('login.html', message=login_massage)
return render_template('login.html')
@app.route("/regiser",methods=["GET", 'POST'])
def register():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
if is_null(username,password):
login_massage = "温馨提示:账号和密码是必填"
return render_template('register.html', message=login_massage)
elif exist_user(username):
login_massage = "温馨提示:用户已存在,请直接登录"
# return redirect(url_for('user_login'))
return render_template('register.html', message=login_massage)
else:
add_user(request.form['username'], request.form['password'] )
return render_template('index.html', username=username)
return render_template('register.html')
if __name__=="__main__":
app.run()
2022-03-17更新:
最近试了下,注册登录多次之后会报错,报错信息如下
经过查找资料,发现是数据库断开了
那么就在每条查询的sql后面补充下这个,就不会报错了
conn.ping(reconnect=True)
cur.execute(sql)
conn.commit()