欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python之Flask实现简单登录功能的示例代码

程序员文章站 2022-07-10 22:54:23
网站少不了要和数据库打交道,归根到底都是一些增删改查操作,这里做一个简单的用户登录功能来学习一下flask如何操作mysql。 用到的一些知识点:flask-sqlalc...

网站少不了要和数据库打交道,归根到底都是一些增删改查操作,这里做一个简单的用户登录功能来学习一下flask如何操作mysql。

用到的一些知识点:flask-sqlalchemy、flask-login、flask-wtf、pymysql

这里通过一个完整的登录实例来介绍,程序已经成功运行,在未登录时拦截了success.html页面跳转到登录页面,登录成功后才能访问success。

以下是项目的整体结构图:

python之Flask实现简单登录功能的示例代码

首先是配置信息,配置了数据库连接等基本的信息,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做登录的表单验证,这里简单做了账号密码不为空如,当我们不填写密码时,点击登录:

python之Flask实现简单登录功能的示例代码

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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。