mediagoblin/media_types/blog/views.py

   1 # GNU MediaGoblin -- federated, autonomous media hosting
   2 # Copyright (C) 2011, 2012 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 import logging
  18
  19 _log = logging.getLogger(__name__)
  20
  21 from datetime import datetime
  22
  23 from werkzeug.exceptions import Forbidden
  24
  25 from mediagoblin import mg_globals
  26
  27 from mediagoblin.media_types.blog import forms as blog_forms
  28 from mediagoblin.media_types.blog.models import Blog, BlogPostData
  29 from mediagoblin.media_types.blog.lib import may_edit_blogpost
  30
  31 from mediagoblin.messages import add_message, SUCCESS, ERROR
  32 from mediagoblin.decorators import (require_active_login, active_user_from_url,
  33                             get_media_entry_by_id, user_may_alter_collection,
  34                             get_user_collection)
  35 from mediagoblin.tools.response import (render_to_response,
  36                                         redirect, redirect_obj, render_404)
  37 from mediagoblin.tools.translate import pass_to_ugettext as _
  38 from mediagoblin.tools.template import render_template
  39 from mediagoblin.tools.text import (
  40     convert_to_tag_list_of_dicts, media_tags_as_string)
  41 from mediagoblin.tools.url import slugify
  42
  43 from mediagoblin.db.util import check_media_slug_used, check_collection_slug_used
  44 from mediagoblin.db.models import User, Collection, MediaEntry
  45
  46 from mediagoblin.notifications import add_comment_subscription
  47
  48
  49 @require_active_login
  50 def blog_edit(request):
  51     """
  52     View for editing the existing blog or automatically
  53     creating a new blog if user does not have any yet.
  54     """
  55     url_user = request.matchdict.get('user', None)
  56     blog_slug = request.matchdict.get('blog_slug', None)
  57
  58     max_blog_count = 1
  59     form = blog_forms.BlogEditForm(request.form)
  60     # the blog doesn't exists yet
  61     if not blog_slug:
  62         if Blog.query.filter_by(author=request.user.id).count()<max_blog_count:
  63             if request.method=='GET':
  64                 return render_to_response(
  65                     request,
  66                     'mediagoblin/blog/blog_edit_create.html',
  67                     {'form': form,
  68                     'user' : request.user,
  69                     'app_config': mg_globals.app_config})
  70
  71             if request.method=='POST' and form.validate():
  72                 _log.info("Here")
  73                 blog = Blog()
  74                 blog.title = unicode(form.title.data)
  75                 blog.description = unicode(form.description.data)  #remember clean html data.
  76                 blog.author = request.user.id
  77                 blog.generate_slug()
  78
  79                 blog.save()
  80                 return redirect(request, "mediagoblin.user_pages.user_home",
  81                         user=request.user.username)
  82         else:
  83             add_message(request, ERROR, "You can not create any more blogs")
  84             return redirect(request, "mediagoblin.user_pages.user_home",
  85                         user=request.user.username)
  86
  87
  88
  89     else:
  90         if request.method == 'GET':
  91             blog = Blog.query.filter_by(slug=blog_slug).first()
  92             defaults = dict(
  93                 title = blog.title,
  94                 description = blog.description,
  95                 author = request.user.id)
  96
  97             form = blog_forms.BlogEditForm(**defaults)
  98
  99             return render_to_response(
 100                     request,
 101                     'mediagoblin/blog/blog_edit_create.html',
 102                     {'form': form,
 103                      'user': request.user,
 104                      'app_config': mg_globals.app_config})
 105         else:
 106             if request.method == 'POST' and form.validate():
 107                 blog.title = unicode(form.title.data)
 108                 blog.description = unicode(form.description.data)
 109                 blog.author = request.user.id
 110                 blog.generate_slug()
 111
 112                 blog.save()
 113                 add_message(request, SUCCESS, "Your blog is updated.")
 114                 return redirect(request, "mediagoblin.user_pages.user_home",
 115                             user=request.user.username)
 116
 117 @require_active_login
 118 def blogpost_create(request):
 119
 120
 121     form = blog_forms.BlogPostEditForm(request.form, license=request.user.license_preference)
 122
 123     if request.method == 'POST' and form.validate():
 124         blog_slug = request.matchdict.get('blog_slug')
 125         blog = request.db.Blog.query.filter_by(slug=blog_slug,
 126             author=request.user.id).first()
 127
 128
 129         blogpost = request.db.MediaEntry()
 130         blogpost.media_type = 'mediagoblin.media_types.blogpost'
 131         blogpost.title = unicode(form.title.data)
 132         blogpost.description = unicode(form.description.data)
 133         blogpost.tags =  convert_to_tag_list_of_dicts(form.tags.data)
 134         blogpost.license = unicode(form.license.data) or None
 135         blogpost.uploader = request.user.id
 136         blogpost.state = 'processed'
 137
 138         blogpost.generate_slug()
 139
 140         blogpost.save()
 141
 142         # connect this blogpost to its blog
 143         blog_post_data = request.db.BlogPostData()
 144         blog_post_data.blog = blog.id
 145         blog_post_data.media_entry = blogpost.id
 146         blog_post_data.save()
 147
 148         add_message(request, SUCCESS, _('Woohoo! Submitted!'))
 149         add_comment_subscription(request.user, blogpost)
 150         return redirect(request, "mediagoblin.user_pages.user_home",
 151             user=request.user.username)
 152
 153     return render_to_response(
 154         request,
 155         'mediagoblin/blog/blog_post_edit_create.html',
 156         {'form': form,
 157         'app_config': mg_globals.app_config,
 158         'user': request.user.username})
 159
 160
 161 @require_active_login
 162 def blogpost_edit(request):
 163     blog_slug = request.matchdict.get('blog_slug')
 164     blog_post_slug = request.matchdict.get('blog_post_slug')
 165
 166     blog = request.db.Blog.query.filter_by(slug=blog_slug, author=request.user.id).first()
 167     blog_post_data = request.db.BlogPostData.query.filter_by(blog=blog.id).first()
 168
 169     blogpost = blog_post_data.get_media_entry
 170
 171     defaults = dict(
 172                 title = blogpost.title,
 173                 description = blogpost.description,
 174                 tags=media_tags_as_string(blogpost.tags),
 175                 license=blogpost.license)
 176
 177     form = blog_forms.BlogPostEditForm(request.form, **defaults)
 178     if request.method == 'POST' and form.validate():
 179         blogpost.title = unicode(form.title.data)
 180         blogpost.description = unicode(form.description.data)
 181         blogpost.tags =  convert_to_tag_list_of_dicts(form.tags.data)
 182         blogpost.license = unicode(form.license.data)
 183
 184         blogpost.generate_slug()
 185         blogpost.save()
 186
 187         add_message(request, SUCCESS, _('Woohoo! edited blogpost is submitted'))
 188         return redirect(request, "mediagoblin.user_pages.user_home",
 189             user=request.user.username)
 190
 191     return render_to_response(
 192         request,
 193         'mediagoblin/blog/blog_post_edit_create.html',
 194         {'form': form,
 195         'app_config': mg_globals.app_config,
 196         'user': request.user.username,
 197         'blog_post_slug': blog_post_slug
 198         })
 199
 200 @require_active_login
 201 def blog_dashboard(request):
 202     blog_posts_list = []
 203     blog_slug = request.matchdict.get('blog_slug')
 204     _log.info(blog_slug)
 205
 206     blog = request.db.Blog.query.filter_by(slug=blog_slug).first()
 207     blog_post_data = request.db.BlogPostData.query.filter_by(blog=blog.id).all()
 208
 209     for each_blog_post_data in blog_post_data:
 210         blog_post = each_blog_post_data.get_media_entry
 211         if blog_post:
 212             blog_posts_list.append(blog_post)
 213     blog_post_count = len(blog_posts_list)
 214
 215     if may_edit_blogpost(request, blog):
 216         return render_to_response(
 217         request,
 218         'mediagoblin/blog/blog_admin_dashboard.html',
 219         {'blog_posts_list': blog_posts_list,
 220         'blog_slug':blog_slug,
 221         'blog':blog,
 222         'blog_post_count':blog_post_count
 223         })
 224
 225
 226
 227
 228
 229