python使用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实现简单登录功能就做完了。

总结

以上所述是小编给大家介绍的python使用flask操作mysql实现登录功能，希望对大家有所帮助