# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
-from webob import exc
+import logging
+import datetime
+import json
from mediagoblin import messages, mg_globals
-from mediagoblin.db.util import DESCENDING, ObjectId
-from mediagoblin.tools.response import render_to_response, render_404, redirect
+from mediagoblin.db.models import (MediaEntry, MediaTag, Collection,
+ CollectionItem, User)
+from mediagoblin.tools.response import render_to_response, render_404, \
+ redirect, redirect_obj
+from mediagoblin.tools.text import cleaned_markdown_conversion
from mediagoblin.tools.translate import pass_to_ugettext as _
from mediagoblin.tools.pagination import Pagination
-from mediagoblin.tools.files import delete_media_files
from mediagoblin.user_pages import forms as user_forms
-
+from mediagoblin.user_pages.lib import add_media_to_collection
+from mediagoblin.notifications import trigger_notification, \
+ add_comment_subscription, mark_comment_notification_seen
from mediagoblin.decorators import (uses_pagination, get_user_media_entry,
- require_active_login, user_may_delete_media)
+ get_media_entry_by_id,
+ require_active_login, user_may_delete_media, user_may_alter_collection,
+ get_user_collection, get_user_collection_item, active_user_from_url)
from werkzeug.contrib.atom import AtomFeed
+from werkzeug.exceptions import MethodNotAllowed
+from werkzeug.wrappers import Response
+
-from mediagoblin.media_types import get_media_manager
+_log = logging.getLogger(__name__)
+_log.setLevel(logging.DEBUG)
@uses_pagination
def user_home(request, page):
"""'Homepage' of a User()"""
- user = request.db.User.find_one({
- 'username': request.matchdict['user']})
+ # TODO: decide if we only want homepages for active users, we can
+ # then use the @get_active_user decorator and also simplify the
+ # template html.
+ user = User.query.filter_by(username=request.matchdict['user']).first()
if not user:
return render_404(request)
elif user.status != u'active':
'mediagoblin/user_pages/user.html',
{'user': user})
- cursor = request.db.MediaEntry.find(
- {'uploader': user._id,
- 'state': 'processed'}).sort('created', DESCENDING)
+ cursor = MediaEntry.query.\
+ filter_by(uploader = user.id,
+ state = u'processed').order_by(MediaEntry.created.desc())
pagination = Pagination(page, cursor)
media_entries = pagination()
'pagination': pagination})
+@active_user_from_url
@uses_pagination
-def user_gallery(request, page):
+def user_gallery(request, page, url_user=None):
"""'Gallery' of a User()"""
- user = request.db.User.find_one({
- 'username': request.matchdict['user'],
- 'status': 'active'})
- if not user:
- return render_404(request)
-
- cursor = request.db.MediaEntry.find(
- {'uploader': user._id,
- 'state': 'processed'}).sort('created', DESCENDING)
-
+ tag = request.matchdict.get('tag', None)
+ cursor = MediaEntry.query.filter_by(
+ uploader=url_user.id,
+ state=u'processed').order_by(MediaEntry.created.desc())
+
+ # Filter potentially by tag too:
+ if tag:
+ cursor = cursor.filter(
+ MediaEntry.tags_helper.any(
+ MediaTag.slug == request.matchdict['tag']))
+
+ # Paginate gallery
pagination = Pagination(page, cursor)
media_entries = pagination()
#if no data is available, return NotFound
+ # TODO: Should we really also return 404 for empty galleries?
if media_entries == None:
return render_404(request)
return render_to_response(
request,
'mediagoblin/user_pages/gallery.html',
- {'user': user,
+ {'user': url_user, 'tag': tag,
'media_entries': media_entries,
'pagination': pagination})
+
MEDIA_COMMENTS_PER_PAGE = 50
"""
'Homepage' of a MediaEntry()
"""
- if ObjectId(request.matchdict.get('comment')):
+ comment_id = request.matchdict.get('comment', None)
+ if comment_id:
+ if request.user:
+ mark_comment_notification_seen(comment_id, request.user)
+
pagination = Pagination(
page, media.get_comments(
mg_globals.app_config['comments_ascending']),
MEDIA_COMMENTS_PER_PAGE,
- ObjectId(request.matchdict.get('comment')))
+ comment_id)
else:
pagination = Pagination(
page, media.get_comments(
comments = pagination()
- comment_form = user_forms.MediaCommentForm(request.POST)
+ comment_form = user_forms.MediaCommentForm(request.form)
- media_template_name = get_media_manager(media.media_type)['display_template']
+ media_template_name = media.media_manager.display_template
return render_to_response(
request,
'app_config': mg_globals.app_config})
-@get_user_media_entry
+@get_media_entry_by_id
@require_active_login
def media_post_comment(request, media):
"""
recieves POST from a MediaEntry() comment form, saves the comment.
"""
- assert request.method == 'POST'
+ if not request.method == 'POST':
+ raise MethodNotAllowed()
comment = request.db.MediaComment()
comment.media_entry = media.id
comment.author = request.user.id
- comment.content = unicode(request.POST['comment_content'])
+ print request.form['comment_content']
+ comment.content = unicode(request.form['comment_content'])
- if not comment.content.strip():
+ # Show error message if commenting is disabled.
+ if not mg_globals.app_config['allow_comments']:
+ messages.add_message(
+ request,
+ messages.ERROR,
+ _("Sorry, comments are disabled."))
+ elif not comment.content.strip():
messages.add_message(
request,
messages.ERROR,
request, messages.SUCCESS,
_('Your comment has been posted!'))
- return exc.HTTPFound(
- location=media.url_for_self(request.urlgen))
+ trigger_notification(comment, media, request)
+ add_comment_subscription(request.user, media)
-@get_user_media_entry
+ return redirect_obj(request, media)
+
+
+
+def media_preview_comment(request):
+ """Runs a comment through markdown so it can be previewed."""
+ # If this isn't an ajax request, render_404
+ if not request.is_xhr:
+ return render_404(request)
+
+ comment = unicode(request.form['comment_content'])
+ cleancomment = { "content":cleaned_markdown_conversion(comment)}
+
+ return Response(json.dumps(cleancomment))
+
+@get_media_entry_by_id
@require_active_login
-@user_may_delete_media
-def media_confirm_delete(request, media):
+def media_collect(request, media):
+ """Add media to collection submission"""
- form = user_forms.ConfirmDeleteForm(request.POST)
+ form = user_forms.MediaCollectForm(request.form)
+ # A user's own collections:
+ form.collection.query = Collection.query.filter_by(
+ creator = request.user.id).order_by(Collection.title)
- if request.method == 'POST' and form.validate():
- if form.confirm.data is True:
- username = media.get_uploader.username
+ if request.method != 'POST' or not form.validate():
+ # No POST submission, or invalid form
+ if not form.validate():
+ messages.add_message(request, messages.ERROR,
+ _('Please check your entries and try again.'))
+
+ return render_to_response(
+ request,
+ 'mediagoblin/user_pages/media_collect.html',
+ {'media': media,
+ 'form': form})
+
+ # If we are here, method=POST and the form is valid, submit things.
+ # If the user is adding a new collection, use that:
+ if form.collection_title.data:
+ # Make sure this user isn't duplicating an existing collection
+ existing_collection = Collection.query.filter_by(
+ creator=request.user.id,
+ title=form.collection_title.data).first()
+ if existing_collection:
+ messages.add_message(request, messages.ERROR,
+ _('You already have a collection called "%s"!')
+ % existing_collection.title)
+ return redirect(request, "mediagoblin.user_pages.media_home",
+ user=media.get_uploader.username,
+ media=media.slug_or_id)
+
+ collection = Collection()
+ collection.title = form.collection_title.data
+ collection.description = form.collection_description.data
+ collection.creator = request.user.id
+ collection.generate_slug()
+ collection.save()
+
+ # Otherwise, use the collection selected from the drop-down
+ else:
+ collection = form.collection.data
+ if collection and collection.creator != request.user.id:
+ collection = None
+
+ # Make sure the user actually selected a collection
+ if not collection:
+ messages.add_message(
+ request, messages.ERROR,
+ _('You have to select or add a collection'))
+ return redirect(request, "mediagoblin.user_pages.media_collect",
+ user=media.get_uploader.username,
+ media_id=media.id)
+
+
+ # Check whether media already exists in collection
+ elif CollectionItem.query.filter_by(
+ media_entry=media.id,
+ collection=collection.id).first():
+ messages.add_message(request, messages.ERROR,
+ _('"%s" already in collection "%s"')
+ % (media.title, collection.title))
+ else: # Add item to collection
+ add_media_to_collection(collection, media, form.note.data)
- # Delete all the associated comments
- for comment in media.get_comments():
- comment.delete()
+ messages.add_message(request, messages.SUCCESS,
+ _('"%s" added to collection "%s"')
+ % (media.title, collection.title))
- # Delete all files on the public storage
- delete_media_files(media)
+ return redirect_obj(request, media)
+
+#TODO: Why does @user_may_delete_media not implicate @require_active_login?
+
+@require_active_login
+def media_confirm_delete(request):
+
+ allowed_state = [u'failed', u'processed']
+ media = None
+ for media_state in allowed_state:
+ media = request.db.MediaEntry.query.filter_by(id=request.matchdict['media_id'], state=media_state).first()
+ if media:
+ break
+
+ if not media:
+ return render_404(request)
+
+ given_username = request.matchdict.get('user')
+ if given_username and (given_username != media.get_uploader.username):
+ return render_404(request)
+
+ uploader_id = media.uploader
+ if not (request.user.is_admin or
+ request.user.id == uploader_id):
+ raise Forbidden()
+
+ form = user_forms.ConfirmDeleteForm(request.form)
+
+ if request.method == 'POST' and form.validate():
+ if form.confirm.data is True:
+ username = media.get_uploader.username
+ # Delete MediaEntry and all related files, comments etc.
media.delete()
messages.add_message(
request, messages.SUCCESS, _('You deleted the media.'))
- return redirect(request, "mediagoblin.user_pages.user_home",
- user=username)
+ location = media.url_to_next(request.urlgen)
+ if not location:
+ location=media.url_to_prev(request.urlgen)
+ if not location:
+ location=request.urlgen("mediagoblin.user_pages.user_home",
+ user=username)
+ return redirect(request, location=location)
else:
messages.add_message(
request, messages.ERROR,
_("The media was not deleted because you didn't check that you were sure."))
- return exc.HTTPFound(
- location=media.url_for_self(request.urlgen))
+ return redirect_obj(request, media)
if ((request.user.is_admin and
- request.user._id != media.uploader)):
+ request.user.id != media.uploader)):
messages.add_message(
request, messages.WARNING,
_("You are about to delete another user's media. "
'form': form})
+@active_user_from_url
+@uses_pagination
+def user_collection(request, page, url_user=None):
+ """A User-defined Collection"""
+ collection = Collection.query.filter_by(
+ get_creator=url_user,
+ slug=request.matchdict['collection']).first()
+
+ if not collection:
+ return render_404(request)
+
+ cursor = collection.get_collection_items()
+
+ pagination = Pagination(page, cursor)
+ collection_items = pagination()
+
+ # if no data is available, return NotFound
+ # TODO: Should an empty collection really also return 404?
+ if collection_items == None:
+ return render_404(request)
+
+ return render_to_response(
+ request,
+ 'mediagoblin/user_pages/collection.html',
+ {'user': url_user,
+ 'collection': collection,
+ 'collection_items': collection_items,
+ 'pagination': pagination})
+
+
+@active_user_from_url
+def collection_list(request, url_user=None):
+ """A User-defined Collection"""
+ collections = Collection.query.filter_by(
+ get_creator=url_user)
+
+ return render_to_response(
+ request,
+ 'mediagoblin/user_pages/collection_list.html',
+ {'user': url_user,
+ 'collections': collections})
+
+
+@get_user_collection_item
+@require_active_login
+@user_may_alter_collection
+def collection_item_confirm_remove(request, collection_item):
+
+ form = user_forms.ConfirmCollectionItemRemoveForm(request.form)
+
+ if request.method == 'POST' and form.validate():
+ username = collection_item.in_collection.get_creator.username
+ collection = collection_item.in_collection
+
+ if form.confirm.data is True:
+ entry = collection_item.get_media_entry
+ entry.collected = entry.collected - 1
+ entry.save()
+
+ collection_item.delete()
+ collection.items = collection.items - 1
+ collection.save()
+
+ messages.add_message(
+ request, messages.SUCCESS, _('You deleted the item from the collection.'))
+ else:
+ messages.add_message(
+ request, messages.ERROR,
+ _("The item was not removed because you didn't check that you were sure."))
+
+ return redirect_obj(request, collection)
+
+ if ((request.user.is_admin and
+ request.user.id != collection_item.in_collection.creator)):
+ messages.add_message(
+ request, messages.WARNING,
+ _("You are about to delete an item from another user's collection. "
+ "Proceed with caution."))
+
+ return render_to_response(
+ request,
+ 'mediagoblin/user_pages/collection_item_confirm_remove.html',
+ {'collection_item': collection_item,
+ 'form': form})
+
+
+@get_user_collection
+@require_active_login
+@user_may_alter_collection
+def collection_confirm_delete(request, collection):
+
+ form = user_forms.ConfirmDeleteForm(request.form)
+
+ if request.method == 'POST' and form.validate():
+
+ username = collection.get_creator.username
+
+ if form.confirm.data is True:
+ collection_title = collection.title
+
+ # Delete all the associated collection items
+ for item in collection.get_collection_items():
+ entry = item.get_media_entry
+ entry.collected = entry.collected - 1
+ entry.save()
+ item.delete()
+
+ collection.delete()
+ messages.add_message(request, messages.SUCCESS,
+ _('You deleted the collection "%s"') % collection_title)
+
+ return redirect(request, "mediagoblin.user_pages.user_home",
+ user=username)
+ else:
+ messages.add_message(
+ request, messages.ERROR,
+ _("The collection was not deleted because you didn't check that you were sure."))
+
+ return redirect_obj(request, collection)
+
+ if ((request.user.is_admin and
+ request.user.id != collection.creator)):
+ messages.add_message(
+ request, messages.WARNING,
+ _("You are about to delete another user's collection. "
+ "Proceed with caution."))
+
+ return render_to_response(
+ request,
+ 'mediagoblin/user_pages/collection_confirm_delete.html',
+ {'collection': collection,
+ 'form': form})
+
+
ATOM_DEFAULT_NR_OF_UPDATED_ITEMS = 15
"""
generates the atom feed with the newest images
"""
-
- user = request.db.User.find_one({
- 'username': request.matchdict['user'],
- 'status': 'active'})
+ user = User.query.filter_by(
+ username = request.matchdict['user'],
+ status = u'active').first()
if not user:
return render_404(request)
- cursor = request.db.MediaEntry.find({
- 'uploader': user._id,
- 'state': 'processed'}) \
- .sort('created', DESCENDING) \
- .limit(ATOM_DEFAULT_NR_OF_UPDATED_ITEMS)
+ cursor = MediaEntry.query.filter_by(
+ uploader = user.id,
+ state = u'processed').\
+ order_by(MediaEntry.created.desc()).\
+ limit(ATOM_DEFAULT_NR_OF_UPDATED_ITEMS)
"""
ATOM feed id is a tag URI (see http://en.wikipedia.org/wiki/Tag_URI)
atomlinks = [{
'href': request.urlgen(
'mediagoblin.user_pages.user_home',
- qualified=True,user=request.matchdict['user']),
+ qualified=True, user=request.matchdict['user']),
'rel': 'alternate',
'type': 'text/html'
- }];
+ }]
+
if mg_globals.app_config["push_urls"]:
for push_url in mg_globals.app_config["push_urls"]:
atomlinks.append({
feed = AtomFeed(
"MediaGoblin: Feed for user '%s'" % request.matchdict['user'],
feed_url=request.url,
- id='tag:'+request.host+',2011:gallery.user-'+request.matchdict['user'],
+ id='tag:{host},{year}:gallery.user-{user}'.format(
+ host=request.host,
+ year=datetime.datetime.today().strftime('%Y'),
+ user=request.matchdict['user']),
links=atomlinks)
-
for entry in cursor:
feed.add(entry.get('title'),
entry.description_html,
- id=entry.url_for_self(request.urlgen,qualified=True),
+ id=entry.url_for_self(request.urlgen, qualified=True),
content_type='html',
author={
'name': entry.get_uploader.username,
return feed.get_response()
+def collection_atom_feed(request):
+ """
+ generates the atom feed with the newest images from a collection
+ """
+ user = User.query.filter_by(
+ username = request.matchdict['user'],
+ status = u'active').first()
+ if not user:
+ return render_404(request)
+
+ collection = Collection.query.filter_by(
+ creator=user.id,
+ slug=request.matchdict['collection']).first()
+ if not collection:
+ return render_404(request)
+
+ cursor = CollectionItem.query.filter_by(
+ collection=collection.id) \
+ .order_by(CollectionItem.added.desc()) \
+ .limit(ATOM_DEFAULT_NR_OF_UPDATED_ITEMS)
+
+ """
+ ATOM feed id is a tag URI (see http://en.wikipedia.org/wiki/Tag_URI)
+ """
+ atomlinks = [{
+ 'href': collection.url_for_self(request.urlgen, qualified=True),
+ 'rel': 'alternate',
+ 'type': 'text/html'
+ }]
+
+ if mg_globals.app_config["push_urls"]:
+ for push_url in mg_globals.app_config["push_urls"]:
+ atomlinks.append({
+ 'rel': 'hub',
+ 'href': push_url})
+
+ feed = AtomFeed(
+ "MediaGoblin: Feed for %s's collection %s" %
+ (request.matchdict['user'], collection.title),
+ feed_url=request.url,
+ id=u'tag:{host},{year}:gnu-mediagoblin.{user}.collection.{slug}'\
+ .format(
+ host=request.host,
+ year=collection.created.strftime('%Y'),
+ user=request.matchdict['user'],
+ slug=collection.slug),
+ links=atomlinks)
+
+ for item in cursor:
+ entry = item.get_media_entry
+ feed.add(entry.get('title'),
+ item.note_html,
+ id=entry.url_for_self(request.urlgen, qualified=True),
+ content_type='html',
+ author={
+ 'name': entry.get_uploader.username,
+ 'uri': request.urlgen(
+ 'mediagoblin.user_pages.user_home',
+ qualified=True, user=entry.get_uploader.username)},
+ updated=item.get('added'),
+ links=[{
+ 'href': entry.url_for_self(
+ request.urlgen,
+ qualified=True),
+ 'rel': 'alternate',
+ 'type': 'text/html'}])
+
+ return feed.get_response()
+
+
@require_active_login
def processing_panel(request):
"""
Show to the user what media is still in conversion/processing...
and what failed, and why!
"""
- # Get the user
- user = request.db.User.find_one(
- {'username': request.matchdict['user'],
- 'status': 'active'})
-
- # Make sure the user exists and is active
- if not user:
- return render_404(request)
- elif user.status != u'active':
- return render_to_response(
- request,
- 'mediagoblin/user_pages/user.html',
- {'user': user})
-
- # XXX: Should this be a decorator?
+ user = User.query.filter_by(username=request.matchdict['user']).first()
+ # TODO: XXX: Should this be a decorator?
#
# Make sure we have permission to access this user's panel. Only
# admins and this user herself should be able to do so.
- if not (user._id == request.user._id
- or request.user.is_admin):
- # No? Let's simply redirect to this user's homepage then.
+ if not (user.id == request.user.id or request.user.is_admin):
+ # No? Simply redirect to this user's homepage.
return redirect(
request, 'mediagoblin.user_pages.user_home',
- user=request.matchdict['user'])
+ user=user.username)
# Get media entries which are in-processing
- processing_entries = request.db.MediaEntry.find(
- {'uploader': user._id,
- 'state': 'unprocessed'}).sort('created', DESCENDING)
+ processing_entries = MediaEntry.query.\
+ filter_by(uploader = user.id,
+ state = u'processing').\
+ order_by(MediaEntry.created.desc())
# Get media entries which have failed to process
- failed_entries = request.db.MediaEntry.find(
- {'uploader': user._id,
- 'state': 'failed'}).sort('created', DESCENDING)
+ failed_entries = MediaEntry.query.\
+ filter_by(uploader = user.id,
+ state = u'failed').\
+ order_by(MediaEntry.created.desc())
+
+ processed_entries = MediaEntry.query.\
+ filter_by(uploader = user.id,
+ state = u'processed').\
+ order_by(MediaEntry.created.desc()).\
+ limit(10)
# Render to response
return render_to_response(
'mediagoblin/user_pages/processing_panel.html',
{'user': user,
'processing_entries': processing_entries,
- 'failed_entries': failed_entries})
+ 'failed_entries': failed_entries,
+ 'processed_entries': processed_entries})
projects / mediagoblin.git / blobdiff