def delete(request): user = current_user(request) # 删除评论 comment_id = int(request.query.get('id', -1)) c = Comment.find(comment_id) t = Tweet.find(c.tweet_id) uid = t.user_id if user.id == t.user_id: Comment.delete(comment_id) return redirect('/tweet/index?user_id={}'.format(uid))
def add(request): user = current_user(request) # 创建评论 form = request.form() c = Comment(form) c.user_id = user.id c.save() log('comment add', c, c.tweet_id) t = Tweet.find(c.tweet_id) u = User.find(t.user_id) return redirect('/tweet/index?user_id={}'.format(u.id))
def delete(request): headers = { 'Content-Type': 'text/html', } uid = current_user(request) header = response_with_headers(headers) user = User.find(uid) # 删除微博 weibo_id = request.query.get('id', None) weibo_id = int(weibo_id) w = Tweet.find(weibo_id) w.delete() return redirect('/tweet?user_id={}'.format(user.id))
def update(request): username = current_user(request) user = User.find_by(username=username) form = request.form() content = form.get('content', '') weibo_id = int(form.get('id', -1)) w = Tweet.find(weibo_id) if user.id != w.user_id: return error(request) w.content = content w.save() # 重定向到用户的主页 return redirect('/tweet?user_id={}'.format(user.id))
def edit(request): headers = { 'Content-Type': 'text/html', } header = response_with_headers(headers) weibo_id = request.query.get('id', -1) weibo_id = int(weibo_id) w = Tweet.find(weibo_id) if w is None: return error(request) # 生成一个 edit 页面 body = template('weibo_edit.html', weibo_id=w.id, weibo_content=w.content) r = header + '\r\n' + body return r.encode(encoding='utf-8')