Thread creation and some tiny posting polish.
[zzz-spline-forum.git] / splinext / forum / controllers / forum.py
1 import logging
2
3 from pylons import config, request, response, session, tmpl_context as c, url
4 from pylons.controllers.util import abort, redirect_to
5 from routes import request_config
6 from sqlalchemy.orm.exc import NoResultFound
7 import wtforms
8 from wtforms import fields
9
10 from spline.model import meta
11 from spline.lib import helpers as h
12 from spline.lib.base import BaseController, render
13 from splinext.forum import model as forum_model
14
15 log = logging.getLogger(__name__)
16
17
18 class WritePostForm(wtforms.Form):
19 content = fields.TextAreaField('Content')
20
21 class WriteThreadForm(WritePostForm):
22 subject = fields.TextField('Subject')
23
24 class ForumController(BaseController):
25
26 def forums(self):
27 c.forums = meta.Session.query(forum_model.Forum) \
28 .order_by(forum_model.Forum.id.asc())
29 return render('/forum/forums.mako')
30
31 def threads(self, forum_id):
32 try:
33 c.forum = meta.Session.query(forum_model.Forum).get(forum_id)
34 except NoResultFound:
35 abort(404)
36
37 c.write_thread_form = WriteThreadForm()
38
39 c.threads = c.forum.threads
40
41 return render('/forum/threads.mako')
42
43 def posts(self, forum_id, thread_id):
44 try:
45 c.thread = meta.Session.query(forum_model.Thread) \
46 .filter_by(id=thread_id, forum_id=forum_id).one()
47 except NoResultFound:
48 abort(404)
49
50 c.write_post_form = WritePostForm()
51
52 return render('/forum/posts.mako')
53
54
55 def write_thread(self, forum_id):
56 """Provides a form for posting a new thread."""
57 if not c.user.can('create_forum_thread'):
58 abort(403)
59
60 try:
61 c.forum = meta.Session.query(forum_model.Forum) \
62 .filter_by(id=forum_id).one()
63 except NoResultFound:
64 abort(404)
65
66 c.write_thread_form = WriteThreadForm(request.params)
67
68 if request.method != 'POST' or not c.write_thread_form.validate():
69 # Failure or initial request; show the form
70 return render('/forum/write_thread.mako')
71
72
73 # Otherwise, add the post.
74 c.forum = meta.Session.query(forum_model.Forum) \
75 .with_lockmode('update') \
76 .get(c.forum.id)
77
78 thread = forum_model.Thread(
79 forum_id = c.forum.id,
80 subject = c.write_thread_form.subject.data,
81 post_count = 1,
82 )
83 post = forum_model.Post(
84 position = 1,
85 author_user_id = c.user.id,
86 content = c.write_thread_form.content.data,
87 )
88
89 thread.posts.append(post)
90 c.forum.threads.append(thread)
91
92 meta.Session.commit()
93
94 # Redirect to the new thread
95 h.flash("Contribution to the collective knowledge of the species successfully recorded.")
96 redirect_to(controller='forum', action='posts',
97 forum_id=forum_id, thread_id=thread.id,
98 _code=303)
99
100 def write(self, forum_id, thread_id):
101 """Provides a form for posting to a thread."""
102 if not c.user.can('create_forum_post'):
103 abort(403)
104
105 try:
106 c.thread = meta.Session.query(forum_model.Thread) \
107 .filter_by(id=thread_id, forum_id=forum_id).one()
108 except NoResultFound:
109 abort(404)
110
111 c.write_post_form = WritePostForm(request.params)
112
113 if request.method != 'POST' or not c.write_post_form.validate():
114 # Failure or initial request; show the form
115 return render('/forum/write.mako')
116
117
118 # Otherwise, add the post.
119 c.thread = meta.Session.query(forum_model.Thread) \
120 .with_lockmode('update') \
121 .get(c.thread.id)
122
123 post = forum_model.Post(
124 position = c.thread.post_count + 1,
125 author_user_id = c.user.id,
126 content = c.write_post_form.content.data,
127 )
128
129 c.thread.posts.append(post)
130 c.thread.post_count += 1
131
132 meta.Session.commit()
133
134 # Redirect to the thread
135 # XXX probably to the post instead; anchor? depends on paging scheme
136 h.flash('Your uniqueness has been added to our own.')
137 redirect_to(controller='forum', action='posts',
138 forum_id=forum_id, thread_id=thread_id,
139 _code=303)