def my_gifts():
uid = current_user.id
gifts_of_mine = Gift.get_user_gift(uid)
isbn_list = [gift.isbn for gift in gifts_of_mine]
wish_count_list = Gift.get_wish_counts(isbn_list)
view_model = MyGifts(gifts_of_mine, wish_count_list)
return render_template('my_gifts.html', gifts=view_model.gifts)
def my_gifts():
"""
赠送清单视图函数
:return:
"""
uid = current_user.id
# 得到我的礼物列表
gifts_of_mine = Gift.get_user_gifts(uid)
# 每个礼物中的 isbn 编号组成的列表
isbn_list = [gift.isbn for gift in gifts_of_mine]
# 获取每个 isbn 对应的心愿数量
wish_count_list = Gift.get_wish_counts(isbn_list)
view_model = MyTrades(gifts_of_mine, wish_count_list)
return render_template('my_gifts.html', trades=view_model.trades)
def index():
"""
首页视图函数
:return:
"""
recent_gift = Gift.recent()
books = [BookViewModel(gift.book) for gift in recent_gift]
return render_template('index.html', recent=books)
def save_to_gifts(isbn):
"""
保存赠书记录视图函数
:param isbn:
:return:
"""
if current_user.can_save_to_list(isbn):
with db.auto_commit():
gift = Gift()
gift.isbn = isbn
gift.uid = current_user.id
# 每发起一次赠书，系统赠送用户鱼豆
current_user.beans += current_app.config['BEANS_UPLOAD_ONE_BOOK']
db.session.add(gift)
else:
flash('这本书已添加至你的赠送清单或已存在于你的心愿清单，请不要重复添加')
return redirect(url_for('web.book_detail', isbn=isbn))
def save_to_gifts(isbn):
if current_user.can_save_to_list(isbn):
# 事务
# 回滚
# try:
with db.auto_commit():
gift = Gift()
gift.isbn = isbn
gift.uid = current_user.id
current_user.beans += current_app.config['BEANS_UPLOAD_ONE_BOOK']
db.session.add(gift)
# db.session.commit()
# except Exception as e:
# db.session.rollback()
# raise e
else:
flash('已经在赠送清单或者心愿清单，请不要重复添加')
return redirect(url_for('web.book_detail', isbn=isbn))
def index():
recent_gifts = Gift.recent()
books = [BookViewModel(gift.book) for gift in recent_gifts]
return render_template('index.html', recent=books)