| 1 | # MediaGoblin -- federated, autonomous media hosting |
| 2 | # Copyright (C) 2011 MediaGoblin contributors. See AUTHORS. |
| 3 | # |
| 4 | # This program is free software: you can redistribute it and/or modify |
| 5 | # it under the terms of the GNU Affero General Public License as published by |
| 6 | # the Free Software Foundation, either version 3 of the License, or |
| 7 | # (at your option) any later version. |
| 8 | # |
| 9 | # This program is distributed in the hope that it will be useful, |
| 10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 | # GNU Affero General Public License for more details. |
| 13 | # |
| 14 | # You should have received a copy of the GNU Affero General Public License |
| 15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 16 | |
| 17 | from webob import exc |
| 18 | |
| 19 | from mediagoblin import messages, mg_globals |
| 20 | from mediagoblin.db.util import DESCENDING, ObjectId |
| 21 | from mediagoblin.tools.text import cleaned_markdown_conversion |
| 22 | from mediagoblin.tools.response import render_to_response, render_404, redirect |
| 23 | from mediagoblin.tools.translate import pass_to_ugettext as _ |
| 24 | from mediagoblin.tools.pagination import Pagination |
| 25 | from mediagoblin.tools.files import delete_media_files |
| 26 | from mediagoblin.user_pages import forms as user_forms |
| 27 | |
| 28 | from mediagoblin.decorators import (uses_pagination, get_user_media_entry, |
| 29 | require_active_login, user_may_delete_media) |
| 30 | |
| 31 | from werkzeug.contrib.atom import AtomFeed |
| 32 | |
| 33 | |
| 34 | @uses_pagination |
| 35 | def user_home(request, page): |
| 36 | """'Homepage' of a User()""" |
| 37 | user = request.db.User.find_one({ |
| 38 | 'username': request.matchdict['user']}) |
| 39 | if not user: |
| 40 | return render_404(request) |
| 41 | elif user['status'] != u'active': |
| 42 | return render_to_response( |
| 43 | request, |
| 44 | 'mediagoblin/user_pages/user.html', |
| 45 | {'user': user}) |
| 46 | |
| 47 | cursor = request.db.MediaEntry.find( |
| 48 | {'uploader': user._id, |
| 49 | 'state': 'processed'}).sort('created', DESCENDING) |
| 50 | |
| 51 | pagination = Pagination(page, cursor) |
| 52 | media_entries = pagination() |
| 53 | |
| 54 | #if no data is available, return NotFound |
| 55 | if media_entries == None: |
| 56 | return render_404(request) |
| 57 | |
| 58 | user_gallery_url = request.urlgen( |
| 59 | 'mediagoblin.user_pages.user_gallery', |
| 60 | user=user['username']) |
| 61 | |
| 62 | return render_to_response( |
| 63 | request, |
| 64 | 'mediagoblin/user_pages/user.html', |
| 65 | {'user': user, |
| 66 | 'user_gallery_url': user_gallery_url, |
| 67 | 'media_entries': media_entries, |
| 68 | 'pagination': pagination}) |
| 69 | |
| 70 | |
| 71 | @uses_pagination |
| 72 | def user_gallery(request, page): |
| 73 | """'Gallery' of a User()""" |
| 74 | user = request.db.User.find_one({ |
| 75 | 'username': request.matchdict['user'], |
| 76 | 'status': 'active'}) |
| 77 | if not user: |
| 78 | return render_404(request) |
| 79 | |
| 80 | cursor = request.db.MediaEntry.find( |
| 81 | {'uploader': user._id, |
| 82 | 'state': 'processed'}).sort('created', DESCENDING) |
| 83 | |
| 84 | pagination = Pagination(page, cursor) |
| 85 | media_entries = pagination() |
| 86 | |
| 87 | #if no data is available, return NotFound |
| 88 | if media_entries == None: |
| 89 | return render_404(request) |
| 90 | |
| 91 | return render_to_response( |
| 92 | request, |
| 93 | 'mediagoblin/user_pages/gallery.html', |
| 94 | {'user': user, |
| 95 | 'media_entries': media_entries, |
| 96 | 'pagination': pagination}) |
| 97 | |
| 98 | MEDIA_COMMENTS_PER_PAGE = 50 |
| 99 | |
| 100 | |
| 101 | @get_user_media_entry |
| 102 | @uses_pagination |
| 103 | def media_home(request, media, page, **kwargs): |
| 104 | """ |
| 105 | 'Homepage' of a MediaEntry() |
| 106 | """ |
| 107 | if ObjectId(request.matchdict.get('comment')): |
| 108 | pagination = Pagination( |
| 109 | page, media.get_comments(True), MEDIA_COMMENTS_PER_PAGE, |
| 110 | ObjectId(request.matchdict.get('comment'))) |
| 111 | else: |
| 112 | pagination = Pagination( |
| 113 | page, media.get_comments(True), MEDIA_COMMENTS_PER_PAGE) |
| 114 | |
| 115 | comments = pagination() |
| 116 | |
| 117 | comment_form = user_forms.MediaCommentForm(request.POST) |
| 118 | |
| 119 | return render_to_response( |
| 120 | request, |
| 121 | 'mediagoblin/user_pages/media.html', |
| 122 | {'media': media, |
| 123 | 'comments': comments, |
| 124 | 'pagination': pagination, |
| 125 | 'comment_form': comment_form, |
| 126 | 'app_config': mg_globals.app_config}) |
| 127 | |
| 128 | |
| 129 | @get_user_media_entry |
| 130 | @require_active_login |
| 131 | def media_post_comment(request, media): |
| 132 | """ |
| 133 | recieves POST from a MediaEntry() comment form, saves the comment. |
| 134 | """ |
| 135 | assert request.method == 'POST' |
| 136 | |
| 137 | comment = request.db.MediaComment() |
| 138 | comment['media_entry'] = media._id |
| 139 | comment['author'] = request.user._id |
| 140 | comment['content'] = unicode(request.POST['comment_content']) |
| 141 | comment['content_html'] = cleaned_markdown_conversion(comment['content']) |
| 142 | |
| 143 | if not comment['content'].strip(): |
| 144 | messages.add_message( |
| 145 | request, |
| 146 | messages.ERROR, |
| 147 | _("Empty comments are not allowed.")) |
| 148 | else: |
| 149 | comment.save() |
| 150 | |
| 151 | messages.add_message( |
| 152 | request, messages.SUCCESS, |
| 153 | _('Comment posted!')) |
| 154 | |
| 155 | return exc.HTTPFound( |
| 156 | location=media.url_for_self(request.urlgen)) |
| 157 | |
| 158 | |
| 159 | @get_user_media_entry |
| 160 | @require_active_login |
| 161 | @user_may_delete_media |
| 162 | def media_confirm_delete(request, media): |
| 163 | |
| 164 | form = user_forms.ConfirmDeleteForm(request.POST) |
| 165 | |
| 166 | if request.method == 'POST' and form.validate(): |
| 167 | if form.confirm.data is True: |
| 168 | username = media.uploader()['username'] |
| 169 | |
| 170 | # Delete all files on the public storage |
| 171 | delete_media_files(media) |
| 172 | |
| 173 | media.delete() |
| 174 | |
| 175 | return redirect(request, "mediagoblin.user_pages.user_home", |
| 176 | user=username) |
| 177 | else: |
| 178 | return exc.HTTPFound( |
| 179 | location=media.url_for_self(request.urlgen)) |
| 180 | |
| 181 | if ((request.user[u'is_admin'] and |
| 182 | request.user._id != media.uploader()._id)): |
| 183 | messages.add_message( |
| 184 | request, messages.WARNING, |
| 185 | _("You are about to delete another user's media. " |
| 186 | "Proceed with caution.")) |
| 187 | |
| 188 | return render_to_response( |
| 189 | request, |
| 190 | 'mediagoblin/user_pages/media_confirm_delete.html', |
| 191 | {'media': media, |
| 192 | 'form': form}) |
| 193 | |
| 194 | |
| 195 | ATOM_DEFAULT_NR_OF_UPDATED_ITEMS = 15 |
| 196 | |
| 197 | |
| 198 | def atom_feed(request): |
| 199 | """ |
| 200 | generates the atom feed with the newest images |
| 201 | """ |
| 202 | |
| 203 | user = request.db.User.find_one({ |
| 204 | 'username': request.matchdict['user'], |
| 205 | 'status': 'active'}) |
| 206 | if not user: |
| 207 | return render_404(request) |
| 208 | |
| 209 | cursor = request.db.MediaEntry.find({ |
| 210 | 'uploader': user._id, |
| 211 | 'state': 'processed'}) \ |
| 212 | .sort('created', DESCENDING) \ |
| 213 | .limit(ATOM_DEFAULT_NR_OF_UPDATED_ITEMS) |
| 214 | |
| 215 | feed = AtomFeed(request.matchdict['user'], |
| 216 | feed_url=request.url, |
| 217 | url=request.host_url) |
| 218 | |
| 219 | for entry in cursor: |
| 220 | feed.add(entry.get('title'), |
| 221 | entry.get('description_html'), |
| 222 | content_type='html', |
| 223 | author=request.matchdict['user'], |
| 224 | updated=entry.get('created'), |
| 225 | url=entry.url_for_self(request.urlgen)) |
| 226 | |
| 227 | return feed.get_response() |
| 228 | |
| 229 | |
| 230 | @require_active_login |
| 231 | def processing_panel(request): |
| 232 | """ |
| 233 | Show to the user what media is still in conversion/processing... |
| 234 | and what failed, and why! |
| 235 | """ |
| 236 | # Get the user |
| 237 | user = request.db.User.find_one( |
| 238 | {'username': request.matchdict['user'], |
| 239 | 'status': 'active'}) |
| 240 | |
| 241 | # Make sure the user exists and is active |
| 242 | if not user: |
| 243 | return render_404(request) |
| 244 | elif user['status'] != u'active': |
| 245 | return render_to_response( |
| 246 | request, |
| 247 | 'mediagoblin/user_pages/user.html', |
| 248 | {'user': user}) |
| 249 | |
| 250 | # XXX: Should this be a decorator? |
| 251 | # |
| 252 | # Make sure we have permission to access this user's panel. Only |
| 253 | # admins and this user herself should be able to do so. |
| 254 | if not (user._id == request.user._id |
| 255 | or request.user.is_admin): |
| 256 | # No? Let's simply redirect to this user's homepage then. |
| 257 | return redirect( |
| 258 | request, 'mediagoblin.user_pages.user_home', |
| 259 | user=request.matchdict['user']) |
| 260 | |
| 261 | # Get media entries which are in-processing |
| 262 | processing_entries = request.db.MediaEntry.find( |
| 263 | {'uploader': user._id, |
| 264 | 'state': 'processing'}).sort('created', DESCENDING) |
| 265 | |
| 266 | # Get media entries which have failed to process |
| 267 | failed_entries = request.db.MediaEntry.find( |
| 268 | {'uploader': user._id, |
| 269 | 'state': 'failed'}).sort('created', DESCENDING) |
| 270 | |
| 271 | # Render to response |
| 272 | return render_to_response( |
| 273 | request, |
| 274 | 'mediagoblin/user_pages/processing_panel.html', |
| 275 | {'user': user, |
| 276 | 'processing_entries': processing_entries, |
| 277 | 'failed_entries': failed_entries}) |