Blog media type initial migration.

[mediagoblin.git] / mediagoblin / media_types / blog / lib.py

diff --git a/mediagoblin/media_types/blog/lib.py b/mediagoblin/media_types/blog/lib.py
index 73ed6060766c85074e7edff1c0dce7762f644f9f..b6e3dc06bd1d59ec47dc23b2b700857a95513314 100644 (file)
--- a/mediagoblin/media_types/blog/lib.py
+++ b/mediagoblin/media_types/blog/lib.py
@@ -24,7 +24,7 @@ def check_blog_slug_used(author_id, slug, ignore_b_id=None):
     return does_exist
     
 def may_edit_blogpost(request, blog):
-    if request.user.is_admin or request.user.id == blog.author:
+    if request.user.has_privilege(u'admin') or request.user.id == blog.author:
         return True
     return False
 
@@ -45,5 +45,12 @@ def get_all_blogposts_of_blog(request, blog, state=None):
             blog_posts_list.append(blog_post)
     blog_posts_list.reverse()
     return blog_posts_list
-    
+
+def get_blog_by_slug(request, slug, **kwargs):
+    if slug.startswith('blog_'):
+        blog_id = int(slug[5:])
+        blog = request.db.Blog.query.filter_by(id=blog_id, **kwargs).first()
+    else:
+        blog = request.db.Blog.query.filter_by(slug=slug, **kwargs).first()
+    return blog