基于flask的网页聊天室(二)
基于flask的网页聊天室(二)
前言
接上一次的内容继续完善,今天完成的内容不是很多,只是简单的用户注册登录,内容具体如下
具体内容
这次要加入与数据哭交互的操作,所以首先要建立相关表结构,这里使用flask-sqlalchemy来辅助创建
首先修改之前的init文件为:
from flask import flask from flask_sqlalchemy import sqlalchemy db = sqlalchemy() from web_chatroom.models import * from web_chatroom.auth import auth from web_chatroom.chat import chat def create_app(): app = flask(__name__) app.register_blueprint(auth) app.register_blueprint(chat) # 添加配置 app.config.from_object('web_chatroom.settings.debugconfig') # 读取配置 db.init_app(app) return app
这是为了引入sqlalchemy()对象db,并读取相关的配置,首先要注意的是db = sqlalchemy()必须要在from web_chatroom.auth import auth之前,因为我在auth.py中使用了db,所以必须要在db创建之后,同时,models是创建的数据库表结构,也要引入内存当中。
接下来看一下配置文件settings,只是一个很简单的配置:
class debugconfig(): debug = true sqlalchemy_database_uri = 'mysql+pymysql://root:******@127.0.0.1:3306/webchatroom' secret_key = 'webchatroom'
主要是在这里先把数据库url配置好
然后是表结构models目前只有user表
from web_chatroom import db class user(db.model): id = db.column(db.integer,primary_key=true,autoincrement=true) email = db.column(db.string(256),unique=true,nullable=false) username = db.column(db.string(32),unique=true,nullable=false) password_hash = db.column(db.string(128))
为了能够像django一样能够执行数据库迁移的操作,我们使用flask-migrate来维护数据库迁移的版本,于是将之前的app.py修改:
from web_chatroom import create_app,db from flask_script import manager from flask_migrate import migrate,migratecommand app = create_app() manager = manager(app) migrate = migrate(app,db) manager.add_command('db',migratecommand) if __name__ == '__main__': manager.run()
这样就可以通过执行
-
python app.py db init
-
python app.py db migrate
-
python app.py db upgrate
完成数据库的迁移
那么相关数据库的建立已完成,接下来可以完善用户的注册登录了
为了辅助表单验证,先建立了utils文件夹,并在其中建立forms.py保存了wtforms的相关form类:
from wtforms import form from wtforms.fields import simple,html5 from wtforms import validators from wtforms import widgets class loginform(form): username = simple.stringfield( validators=[ validators.datarequired(message='用户名不能为空.') ], widget=widgets.textinput(), ) password = simple.passwordfield( validators=[ validators.datarequired(message='密码不能为空.'), validators.length(min=8, message='密码长度必须大于%(min)d'), ], widget=widgets.passwordinput(), ) class registerform(form): username = simple.stringfield( validators=[ validators.datarequired(message='用户名不能为空.') ], widget=widgets.textinput(), ) password = simple.passwordfield( validators=[ validators.datarequired(message='密码不能为空.'), validators.length(min=8, message='密码长度必须大于%(min)d'), ], widget=widgets.passwordinput(), ) email = html5.emailfield( label='邮箱', validators=[ validators.datarequired(message='邮箱不能为空.'), validators.email(message='邮箱格式错误') ], widget=widgets.textinput(input_type='email'), )
接下来修改一下登录与注册的html页面,具体代码就不放了,然后在auth.py中的视图函数中处理验证:
from flask import render_template,flash,redirect,url_for from flask import request from web_chatroom import models from web_chatroom import db import hashlib from utils import forms auth = blueprint('auth', __name__) @auth.route('/login', methods=['get', "post"],endpoint='login') def login(): if request.method == 'get': return render_template('login.html') elif request.method == 'post': form = forms.loginform(formdata=request.form) if form.validate(): password_hash = hashlib.md5(form.data['password'].encode('utf-8')).hexdigest() count = db.session.query(models.user).filter(db.and_(models.user.username == form.data['username'], models.user.password_hash == password_hash)).count() if count: return redirect(url_for('chat.chat')) else: flash('用户名或密码错误') return redirect(url_for('auth.login')) pass else: for error in form.errors: flash(form.errors[error][0]) return redirect(url_for('auth.login')) @auth.route('/register', methods=['get', "post"],endpoint='register') def register(): if request.method == 'get': return render_template('register.html') elif request.method == 'post': form = forms.registerform(formdata=request.form) if form.validate(): count = db.session.query(models.user).filter(db.or_(models.user.username == form.data['username'], models.user.email == form.data['email'])).count() if count: flash('用户名或邮箱已存在') return redirect(url_for('auth.register')) else: password_hash = hashlib.md5(form.data['password'].encode('utf-8')).hexdigest() db.session.add(models.user(username=form.data['username'], email=form.data['email'], password_hash=password_hash)) db.session.commit() db.session.close() flash('注册成功') return redirect(url_for('auth.login')) else: for error in form.errors: flash(form.errors[error][0]) return redirect(url_for('auth.register'))
验证的错误信息通过flash在页面显示,像这样:
注册同理,最后登录成功会跳转到chat.html页面,chat.py中的视图函数就只先写了个显示:
from flask import blueprint from flask import render_template,flash,redirect,url_for from flask import request chat = blueprint('chat', __name__) @chat.route('/chat', methods=['get', "post"],endpoint='chat') def chatroom(): if request.method == 'get': return render_template('chatroom.html')
当然页面里目前什么也没有。。。
最后,当前项目的目录结构是这样的:
总结
今天主要完成了简单的数据库交互,明天继续加入cookie,session,以及csrf的保护,以及chat的页面,消息表等。。
上一篇: ai文件太大打不开怎么缩小?
下一篇: day 8 - 1 文件操作练习