python之Flask實現登陸功能

 

網站少不了要和數據庫打交道，歸根到底都是一些增刪改查操做，這裏作一個簡單的用戶登陸功能來學習一下Flask如何操做MySQL。

用到的一些知識點：Flask-SQLAlchemy、Flask-Login、Flask-WTF、PyMySQL

這裏經過一個完整的登陸實例來介紹，程序已經成功運行，在未登陸時攔截了success.html頁面跳轉到登陸頁面，登陸成功後才能訪問success。

如下是項目的總體結構圖：

 

首先是配置信息，配置了數據庫鏈接等基本的信息，config.py

DEBUG = True  
SQLALCHEMY_ECHO = False  
SQLALCHEMY_DATABASE_URI = 'mysql+pymysql://root:1011@localhost/rl_project?charset=utf8'  
SECRET_KEY = '*\xff\x93\xc8w\x13\x0e@3\xd6\x82\x0f\x84\x18\xe7\xd9\\|\x04e\xb9(\xfd\xc3' 

　　

common/_init_.py

# config=utf-8  
from flask_sqlalchemy import SQLAlchemy  
  
__all__ = ['db']  
db = SQLAlchemy()  

　　

數據庫配置類，common/data.py

# config=utf-8  
from sqlalchemy import create_engine  
from sqlalchemy.sql import text  
from config import SQLALCHEMY_DATABASE_URI, SQLALCHEMY_ECHO  
  
  
def db_query(sql, settings=None, echo=None):  
  
    if settings is None:  
        settings = SQLALCHEMY_DATABASE_URI  
  
    if echo is None:  
        echo = SQLALCHEMY_ECHO  
  
    return create_engine(settings, echo=echo).connect().execute(text(sql)).fetchall()  
  
  
def db_execute(sql, settings=None, echo=None):  
  
    if settings is None:  
        settings = SQLALCHEMY_DATABASE_URI  
  
    if echo is None:  
        echo = SQLALCHEMY_ECHO  
  
    return create_engine(settings, echo=echo).connect().execute(text(sql)).rowcount  

　　

SQLALCHEMY_DATABASE_URI用於鏈接數據的數據庫。
SQLALCHEMY_ECHO若是設置成 True，SQLAlchemy 將會記錄全部 發到標準輸出(stderr)的語句，這對調試頗有幫助。
固然，咱們在setting中設置了基本的鏈接數據庫信息，啓動時加載app = create_app('../config.py')，因此這個類刪掉也不會報錯。

form/login_form.py

# config=utf-8  
from flask_wtf import FlaskForm as Form  
from wtforms import StringField, PasswordField  
from wtforms.validators import DataRequired  
  
  
class LoginForm(Form):  
    accountNumber = StringField('accountNumber', validators=[DataRequired('accountNumber is null')])  
    password = PasswordField('password', validators=[DataRequired('password is null')])  

　　

使用Flask-WTF作登陸的表單驗證，這裏簡單作了帳號密碼不爲空

 

如，當咱們不填寫密碼時，點擊登陸：

model/_init_.py

# config=utf-8  
from flask import Flask  
from flask_login import LoginManager  
from common import db  
  
login_manager = LoginManager()  
login_manager.login_view = "user.login"  
  
  
def create_app(config_filename=None):  
    app = Flask(__name__)  
    login_manager.init_app(app)  
  
    if config_filename is not None:  
        app.config.from_pyfile(config_filename)  
        configure_database(app)  
  
    return app  
  
  
def configure_database(app):  
    db.init_app(app)  

　　

其中，login_manager.login_view = "user.login" 指定了未登陸時跳轉的頁面，即被攔截後統一跳到user/login這個路由下

 

model/user_model.py

# config=utf-8  
from flask_login import UserMixin  
from common import db  
  
  
class User(db.Model, UserMixin):  
  
    user_id = db.Column('id', db.Integer, primary_key=True)  
    accountNumber = db.Column(db.String(200), unique=True)  
    password = db.Column(db.String(50), unique=True)  
    name = db.Column(db.String(20), unique=True)  
  
    __tablename__ = 'tb_user'  
  
    def __init__(self, user_id=None, account_number=None, password=None, name="anonymous"):  
  
        self.user_id = user_id  
        self.accountNumber = account_number  
        self.password = password  
        self.name = name  
  
    def is_authenticated(self):  
        return True  
  
    def is_active(self):  
        return True  
  
    def is_anonymous(self):  
        return False  
  
    def get_id(self):  
        return unicode(self.user_id)  
  
    def __repr__(self):  
        return '<User %r>' % (self.accountNumber)  

　　

這裏須要注意：

def get_id(self):  
        return unicode(self.user_id)  

該方法不可缺乏，不然會報：NotImplementedError: No `id` attribute - override `get_id`錯誤。

 

get_id()

返回一個能惟一識別用戶的，並能用於從 user_loader 回調中 加載用戶的 unicode 。注意着 必須 是一個 unicode ——若是 ID 本來是 一個 int 或其它類型，你須要把它轉換爲 unicode 。

 

is_authenticated()

當用戶經過驗證時，也即提供有效證實時返回 True

 

is_active()

若是這是一個經過驗證、已激活、未被停用的帳戶返回 True 。

 

is_anonymous()

若是是一個匿名用戶，返回 True 。

 

login.py

#encoding:utf-8  
#!/usr/bin/env python  
from flask import render_template, request, redirect, Flask, Blueprint  
from flask_login import login_user, login_required  
from model.user_model import User  
from model import login_manager  
from form.login_form import LoginForm  
  
  
userRoute = Blueprint('user', __name__, url_prefix='/user', template_folder='templates', static_folder='static')  
 
@login_manager.user_loader  
def load_user(user_id):  
    return User.query.get(int(user_id))  
 
 
@userRoute.before_request  
def before_request():  
    pass  
 
@userRoute.route('/success')  
@login_required  
def index():  
    return render_template('success.html')  
 
@userRoute.route('/login', methods=['GET', 'POST'])  
def login():  
    form = LoginForm()  
    if request.method == 'POST':  
        if not form.validate_on_submit():  
            print form.errors  
            return render_template('login.html', form=form)  
  
        user = User.query.filter(User.accountNumber == form.accountNumber.data,  
                                 User.password == form.password.data).first()  
        if user:  
            login_user(user)  
            return render_template('success.html')  
  
    return render_template('login.html', form=form) 

　　

其中，要實現一個load_user()回調方法，這個回調用於從會話中存儲的用戶 ID 從新加載用戶對象，id存在則返回對應的用戶對象，不存在則返回none。

 

有些操做是須要用戶登陸的，有些操做則無需用戶登陸，這裏使用到了@login_required，在每個須要登陸才能訪問的路由方法上加@login_required便可。

 

啓動類，runserver.py

# config=utf-8  
from login import userRoute  
from model import create_app  
  
DEFAULT_MODULES = [userRoute]  
  
app = create_app('../config.py')  
  
for module in DEFAULT_MODULES:  
    app.register_blueprint(module)  
 
 
@app.before_request  
def before_request():  
    pass  
  
  
if __name__ == '__main__':  
    app.run(debug=True)  

　　

DEFAULT_MODULES = [userRoute]是將userRoute藍圖註冊入app，才能啓動login中的userRoute路由，咱們在login.py中使用了藍圖：userRoute = Blueprint('user', __name__, url_prefix='/user', template_folder='templates', static_folder='static')

@app.before_request  
def before_request():  
    pass  

這是一個全局的方法，在請求開始以前被調用，在某些場景下作一些提示或者特殊處理，固然這裏沒用上，直接pass，這個方法去掉對項目沒有影響。

 

基本樣式模版，base.html

<!DOCTYPE html>  
<html lang="cn">  
<head>  
    <meta charset="UTF-8">  
    <title>{% block title %}{% endblock %}</title>  
    <script src="{{ url_for('static', filename='jquery1.42.min.js') }}"></script>  
    {% block head %}{% endblock %}  
</head>  
<body>  
{% block content %}{% endblock %}  
</body>  
</html>  

　　

登陸前臺頁面，login.html

{% extends "base.html" %}  
{% block title %}python flask user page{% endblock %}  
{% block head %}  
    <style type="text/css"></style>  
{% endblock %}  
{% block content %}  
    <form action="{{ url_for('user.login') }}" method="post">  
        {% if form.errors %}  
            <ul>  
                {% for name, errors in form.errors.items() %}  
                    {% for error in errors %}  
                        <li>{{ error }}</li>  
                    {% endfor %}  
                {% endfor %}  
            </ul>  
        {% endif %}  
        帳號：{{ form.accountNumber(size=20) }}<label>{{ form.accountNumber.errors[0] }}</label><br/>  
        密碼：<input name="password" type="password"/><br/>  
        {{ form.hidden_tag() }}  
        <button type="submit">登陸</button>  
    </form>  
{% endblock %}  

　　

到此，一個Flask實現簡單登陸功能就作完了。