【Flask开发】嘿马文学web完整flask项目第3篇：2.用户认证,2.用户认证【附代码文档】

{/** jwt token*/"token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJuYW1lIjoiSm9obiBEb2UiLCJhZG1pbiI6dHJ1ZX0.OLvs36KmqB9cmsUrMpUutfhV52_iSz4bQMYJjkI_TLQ",/** 用户信息*/"userInfo": {"uid": "199282123", // 用户id"gender": 1, // 1 男 0 女// 头像"avatarUrl": "},/** 阅读配置*/"config": {"preference": "0", // 0 女 1 男"brightness": 90, // 10~100 亮度"fontSize": "18", // 字号"background": "B1", // B1 ~ B6 内置背景"turn": "T1" // T1 仿真 T2 平滑 T3 无 翻页模式}
}

from flask import Blueprintuser_bp = Blueprint('user', __name__)

@user_bp.route("/login",methods=['POST'])
def login():
    """ 第三方登录 """# code 用户登录凭证（有效期五分钟）。开发者需要在开发者服务器后台调用 auth.code2Session，# 使用 code 换取 openid 和 session_key 等信息code = request.json.get('code', '')# iv 加密算法的初始向量iv = request.json.get('iv', '')# encryptedData 包括敏感数据在内的完整用户信息的加密数据encryptedData = request.json.get('encryptedData', '')if not iv or not encryptedData or not code:return jsonify(msg='参数有误'), 403data = wxauth.get_wxapp_session_key(code)print('data', data)if 'session_key' not in data:return jsonify(msg='获取session_key失败', data=data), 500user_info = wxauth.get_user_info(encryptedData, iv, data['session_key'])if 'openId' not in user_info:return jsonify(msg='获取用户信息失败', user_info=user_info), 403user = User.query.filter_by(openId=user_info.get('openId')).first()if not user:user = User(user_info)db.session.add(user)db.session.flush()# 书架增加随机书籍-----后面完成# _add_book_shelf(user.id, user.gender)else:# 更新用户信息user.update_info(user_info)db.session.commit()# 生成jwt tokentoken = _generate_tokens(user.id)ret_data = {"token": token,"userInfo": {"uid": user.id,"gender": user.gender,"avatarUrl": user.avatarUrl},"config": {"preference": user.preference,"brightness": user.brightness,"fontSize": user.fontSize,"background": user.background,"turn": user.turn}}return jsonify(ret_data)

@user_bp.route('/tmp_add_user', methods=['POST'])
def tmp_add_user():
    """
    添加测试用户
    :return:
    """data = dict(openId='1'*32,nickName='测试用户002',gender=1,city='广州市',province='广东省',country='中国',avatarUrl='default')user = User(data)db.session.add(user)db.session.commit()# 后面完成# _add_book_shelf(user.id, user.gender)ret_data = {'msg': '添加成功','user_id': user.id,}return jsonify(ret_data)

@user_bp.route('/tmp_login')
def tmp_login():
    """
    登录测试接口
    :return:
    """user_id = request.args.get('user_id')user = User.query.get(user_id)# 生成jwt tokentoken = _generate_tokens(user_id)ret_data = {"token": token,"userInfo": {"uid": user.id,"gender": user.gender,"avatarUrl": user.avatarUrl},"config": {"preference": user.preference,"brightness": user.brightness,"fontSize": user.fontSize,"background": user.background,"turn": user.turn}}return jsonify(ret_data)

def before_request():g.user_id = Noneauthorization = request.headers.get('Authorization')print(authorization)payload = verify_jwt(authorization)print(payload)if payload:g.user_id = payload['user_id']

from functools import wraps
from flask import g, jsonifydef login_required(func):# wraps的作用让被装饰器装饰的函数的属性不发生变化。@wraps(func)def wrapper(*args, **kwargs):if not g.user_id:return jsonify({'msg': 'token error'}), 401return func(*args, **kwargs)return wrapper

from flask import Blueprintmybooks_bp = Blueprint('mybooks', __name__)

@mybooks_bp.route('/')
@login_required
def mybooks_list():
    """
    我的书架-列表
    :return:
    """mybooks = BookShelf.query.filter_by(user_id=g.user_id).order_by(BookShelf.created.desc()).all()data = []if not mybooks:books = Book.query.all()bs_books = random.sample(books, 5)for book in bs_books:bs = BookShelf(user_id=g.user_id,book_id=book.book_id,book_name=book.book_name,cover=book.cover)db.session.add(bs)data.append({'id': book.book_id,'imgURL': ' book.cover),'title': book.book_name})db.session.commit()return jsonify(data)else:for bs in mybooks:data.append({'id': bs.book_id,'imgURL': ' bs.cover),'title': bs.book_name})return jsonify(data)

from flask import Blueprintmybooks_bp = Blueprint('mybooks', __name__)

# 定义路由，书架管理--添加书籍@login_required
@mybooks_bp.route("/<book_id>",methods=['POST'])
def add_book(book_id):
    """
    book_id：url固定参数，必须作为视图参数直接传入，Flask中使用转换器进行处理，默认的数据类型是str；
    :return:
    """# 1.添加登录验证装饰器user_id = g.user_id# 2.接收参数，书籍id# 3.根据书籍id，查询书籍表，确认数据的存在book = Book.query.filter(Book.book_id==book_id).first()# 确认查询结果if not book:return jsonify(msg='书籍不存在'),404# 4.查询书架表，确认该书在书架中是否存在book_shelf = BookShelf.query.filter(BookShelf.user_id==user_id,BookShelf.book_id==book_id).first()# 判断书架的查询结果if not book_shelf:# 5.如果书架中不存在，添加书籍bk_shelf = BookShelf(user_id=user_id,book_id=book.book_id,book_name=book.book_name,cover=book.cover)db.session.add(bk_shelf)db.session.commit()# 返回添加成功的信息return jsonify(msg='添加成功')else:# 否则，书架中该书籍已经存在return jsonify(msg='书架中该书籍已经存在'),400

@login_required
@mybooks_bp.route("/<book_id>",methods=['DELETE'])
def del_book(book_id):# - 1.添加登录验证装饰器user_id = g.user_id# - 2.接收参数，书籍idbk_shelf = BookShelf.query.filter_by(user_id=user_id,book_id=book_id).first()# - 3.根据书籍id，查询书籍表，确认数据的存在if not bk_shelf:return jsonify(msg='该书籍在书架中不存在'),400# - 4.删除书籍db.session.delete(bk_shelf)db.session.commit()return jsonify(msg='删除成功')pass

from flask import Blueprintmybooks_bp = Blueprint('mybooks', __name__)

# 定义路由，书架管理--最后阅读@login_required
@mybooks_bp.route("/last")
def book_last_reading():# 1.使用登录验证装饰器，获取用户信息user_id = g.user_iduser = User.query.get(user_id)read_rate = None# 2.判断用户没有阅读书籍if not user.last_read:# 3.如果用户没有阅读，默认查询第一本书籍，当做用户的阅读书籍# -----也可以查询书架的第一本书book = Book.query.first()# 保存用户的阅读书籍的iduser.last_read = book.book_id# 4.查询该书籍的章节信息，默认升序排序，bk_chapter = BookChapters.query.filter_by(book_id=book.book_id).order_by(BookChapters.chapter_id.asc()).first()# 保存用户的阅读书籍的章节信息user.last_read_chapter_id = bk_chapter.chapter_id# - 把查询结果，存入阅读进度表read_rate = ReadRate(user_id=user.id,book_id=book.book_id,chapter_id=bk_chapter.chapter_id,chapter_name=bk_chapter.chapter_name)# 保存数据db.session.add(read_rate)db.session.add(user)# 保存两个对象，参数必须列表# db.session.add_all([read_rate,user])db.session.commit()# 5.如果用户阅读书籍，查询用户阅读的书籍else:book = Book.query.get(user.last_read)# 6.判断是否有阅读进度，如果没有，查询阅读进度表if not read_rate:read_rate = ReadRate.query.filter_by(user_id=user.id,book_id=book.book_id,chapter_id=user.last_read_chapter_id).first()# 7.返回查询结果data = {'id':book.book_id,'title':book.book_name,'chapter':read_rate.chapter_name,'progress':read_rate.rate,'imgURL':'}# 转成json格式返回数据return jsonify(data)

from flask import Blueprintcategory_bp = Blueprint('category',__name__,url_prefix='/categories')

@category_bp.route('/')
def category_list():# 1.用户参数，性别参数,查询字符串# 如果未传入参数，默认值为1表示男，int表示类型转换# gender = int(gender)gender = request.args.get("gender",1,int)# 2.根据性别，查询大分类数据bk_big_category = BookBigCategory.query.filter(BookBigCategory.channel==gender).all()data = []# 3.遍历大分类列表数据，保存for book_category in bk_big_category:big_temp = {'id':book_category.cate_id,'title':book_category.cate_name,'imgURL':''subCategory':[]}# 4.可以通过关系引用，获取大分类下面的二级分类数据for category in book_category.second_cates:temp = {'id':category.cate_id,'title':category.cate_name}big_temp['subCategory'].append(temp)data.append(big_temp)# 5.返回分类数据return jsonify(data)