中国保密在线网站培训,晋中做网站,免费网站空间 推荐,网站建设合同 技术合同范本1. 在commons.py文件中新增一个装饰器类: 注#xff1a;一定要加入wraps进行装饰否则#xff0c;装饰器在给多个函数进行装饰时会报错
from functools import wraps
from flask import session, current_app, g# 定义登陆装饰器#xff0c;封装用户的登陆数据
def user_log…1. 在commons.py文件中新增一个装饰器类: 注一定要加入wraps进行装饰否则装饰器在给多个函数进行装饰时会报错
from functools import wraps
from flask import session, current_app, g# 定义登陆装饰器封装用户的登陆数据
def user_login_data(view_func):wraps(view_func)def wrapper(*args, **kwargs):# 1,从sess1on中取出用户的user1duser_id session.get(user_id)# 2通过user1d取出用户对象user Noneif user_id:try:from info.models import Useruser User.query.get(user_id)except Exception as e:current_app.logger.error(e)# 3.将u5er数据封装到g对象g.user userreturn view_func(*args, **kwargs)return wrapper
2.装饰器的使用
from flask import current_app, jsonify, render_template, abort, session, g
from . import news_blue
from info import models
from info.utils.response_code import RET
from ...utils.commons import user_login_datanews_blue.route(/int:news_id)
##########自定义装饰器#######################
user_login_data
def news_detail(news_id):try:############################原本的这里的功能被装饰器取代了############## # 0,从session中获取到user_id并根据user_id获取到用户信息# user_id session.get(user_id)# user models.User.query.get(user_id)# 1,根据新闻编号查询新闻对象try:news models.News.query.get(news_id)except Exception as e:current_app.logger.error(e)return jsonify(errnoRET.DBERR, errmsg获取新闻失败)# 2. 如果新闻对象不存在直接抛出异常if not news:abort(404)# 获取前6条热点新闻click_news models.News.query.order_by(models.News.clicks.desc()).limit(6)click_news_list []for item in click_news:click_news_list.append(item.to_dict())# 2.携带数据渲染页面data {news_info: news.to_dict(),#########这里的user直接从g中取出如果使用装饰包裹时g中是没有user的#########user_info: g.user.to_dict() if g.user else ,news: click_news_list}except Exception as e:current_app.logger.error(e)return jsonify(errnoRET.UNKOWNERR, errmsg未知错误)return render_template(news/detail.html, datadata)
