Пример #1
0
	def __init__(self, *args, **kwargs):
		RedirectForm.__init__(self, *args, **kwargs)
		self.user = None
Пример #2
0
def thread_mod_action(forum_slug, thread_id, thread_slug, action):
	# This view is becoming spaghetti.
	url_forum = Forum.query.filter_by(slug=forum_slug).first()
	thread = Thread.query.get(thread_id)

	if not thread:
		abort(404)
	if thread.is_private:
		abort(404)

	forum = thread.forum
	requires_moderator = True
	if action == 'sticky':
		url = thread.sticky_url
	elif action == 'lock':
		url = thread.lock_url
	elif action == 'follow':
		if not thread.can_be_followed:
			abort(404)
		url = thread.follow_url
		requires_moderator = False

	if requires_moderator:
		if not forum.can_be_moderated_by(g.user):
			abort(403)
	else:
		if not forum.can_be_viewed_by(g.user):
			abort(403)

	if request.method == 'GET' and (url_forum != thread.forum or thread_slug != thread.slug):
		return redirect(url, code=301)

	ajax = ('ajax' in request.values)
	form = RedirectForm()

	if action == 'sticky':
		old_value = thread.is_stickied
		attr = 'is_stickied'

		if old_value:
			cap_verb = 'Unstick'
			done_word = 'unstickied'
			link_title = 'Stick'
			message = 'Are you sure you want to make this thread a normal thread?'
		else:
			cap_verb = 'Stick'
			done_word = 'stickied'
			link_title = 'Unstick'
			message = 'Are you sure you want to make this thread a sticky?'
	elif action == 'lock':
		old_value = thread.is_locked
		attr = 'is_locked'

		if old_value:
			cap_verb = 'Unlock'
			done_word = 'unlocked'
			link_title = 'Lock'
			message = 'Are you sure you want to unlock this thread?'
		else:
			cap_verb = 'Lock'
			done_word = 'locked'
			link_title = 'Unlock'
			message = 'Are you sure you want to lock this thread?'
	elif action == 'follow':
		old_value = thread.is_followed_by(g.user)
		attr = None

		if old_value:
			cap_verb = 'Follow'
			link_title = 'Follow'
			message = 'Are you sure you want to follow this thread?'
		else:
			cap_verb = 'Unfollow'
			link_title = 'Unfollow'
			message = 'Are you sure you want to stop following this thread?'


	if request.method == 'POST':
		if action == 'follow':
			if old_value:
				thread.followers.remove(g.user)
				thread.follower_count -= 1
				msg = 'You are no longer following this thread.'
			else:
				thread.followers.append(g.user)
				thread.follower_count += 1
				msg = 'You are now following this thread. You will receive notifications about new posts.'
		else:
			setattr(thread, attr, not old_value)
			msg = 'The thread has been %s.' % done_word
		db.session.commit()


		if ajax:
			return jsonify(toast=msg, link_title=link_title)
		else:
			flash(msg)
			return form.redirect(url=thread.url)
	else:
		return render_template('confirm.html',
			form=form,
			crumbs_type='thread',
			forum=forum, thread=thread,
			final_crumb=('%s Thread' % cap_verb),
			message=message,
			url=url)