基於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的頁面，消息表等。。