Upgrade to Pylons 1.0. #283
[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
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 c.forum = meta.Session.query(forum_model.Forum).get(forum_id)
33 if not c.forum:
34 abort(404)
35
36 c.write_thread_form = WriteThreadForm()
37
38 c.threads = c.forum.threads
39
40 return render('/forum/threads.mako')
41
42 def posts(self, forum_id, thread_id):
43 try:
44 c.thread = meta.Session.query(forum_model.Thread) \
45 .filter_by(id=thread_id, forum_id=forum_id).one()
46 except NoResultFound:
47 abort(404)
48
49 c.write_post_form = WritePostForm()
50
51 return render('/forum/posts.mako')
52
53
54 def write_thread(self, forum_id):
55 """Provides a form for posting a new thread."""
56 if not c.user.can('forum:create-thread'):
57 abort(403)
58
59 try:
60 c.forum = meta.Session.query(forum_model.Forum) \
61 .filter_by(id=forum_id).one()
62 except NoResultFound:
63 abort(404)
64
65 c.write_thread_form = WriteThreadForm(request.params)
66
67 if request.method != 'POST' or not c.write_thread_form.validate():
68 # Failure or initial request; show the form
69 return render('/forum/write_thread.mako')
70
71
72 # Otherwise, add the post.
73 c.forum = meta.Session.query(forum_model.Forum) \
74 .with_lockmode('update') \
75 .get(c.forum.id)
76
77 thread = forum_model.Thread(
78 forum_id = c.forum.id,
79 subject = c.write_thread_form.subject.data,
80 post_count = 1,
81 )
82 post = forum_model.Post(
83 position = 1,
84 author_user_id = c.user.id,
85 content = c.write_thread_form.content.data,
86 )
87
88 thread.posts.append(post)
89 c.forum.threads.append(thread)
90
91 meta.Session.commit()
92
93 # Redirect to the new thread
94 h.flash("Contribution to the collective knowledge of the species successfully recorded.")
95 redirect(
96 url(controller='forum', action='posts',
97 forum_id=forum_id, thread_id=thread.id),
98 code=303,
99 )
100
101 def write(self, forum_id, thread_id):
102 """Provides a form for posting to a thread."""
103 if not c.user.can('forum:create-post'):
104 abort(403)
105
106 try:
107 c.thread = meta.Session.query(forum_model.Thread) \
108 .filter_by(id=thread_id, forum_id=forum_id).one()
109 except NoResultFound:
110 abort(404)
111
112 c.write_post_form = WritePostForm(request.params)
113
114 if request.method != 'POST' or not c.write_post_form.validate():
115 # Failure or initial request; show the form
116 return render('/forum/write.mako')
117
118
119 # Otherwise, add the post.
120 c.thread = meta.Session.query(forum_model.Thread) \
121 .with_lockmode('update') \
122 .get(c.thread.id)
123
124 post = forum_model.Post(
125 position = c.thread.post_count + 1,
126 author_user_id = c.user.id,
127 content = c.write_post_form.content.data,
128 )
129
130 c.thread.posts.append(post)
131 c.thread.post_count += 1
132
133 meta.Session.commit()
134
135 # Redirect to the thread
136 # XXX probably to the post instead; anchor? depends on paging scheme
137 h.flash('Your uniqueness has been added to our own.')
138 redirect(
139 url(controller='forum', action='posts',
140 forum_id=forum_id, thread_id=thread_id),
141 code=303,
142 )