1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011 Free Software Foundation, Inc
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.
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.
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/>.
20 from mediagoblin
.util
import redirect
21 from mediagoblin
.db
.util
import ObjectId
, InvalidId
24 def _make_safe(decorator
, original
):
26 Copy the function data from the old function to the decorator.
28 decorator
.__name
__ = original
.__name
__
29 decorator
.__dict
__ = original
.__dict
__
30 decorator
.__doc
__ = original
.__doc
__
34 def require_active_login(controller
):
36 Require an active login from the user.
38 def new_controller_func(request
, *args
, **kwargs
):
40 request
.user
.get('status') == u
'needs_email_verification':
41 return redirect(request
,
42 'mediagoblin.auth.verify_email_notice')
43 elif not request
.user
or request
.user
.get('status') != u
'active':
45 location
="%s?next=%s" % (
46 request
.urlgen("mediagoblin.auth.login"),
49 return controller(request
, *args
, **kwargs
)
51 return _make_safe(new_controller_func
, controller
)
54 def uses_pagination(controller
):
56 Check request GET 'page' key for wrong values
58 def wrapper(request
, *args
, **kwargs
):
60 page
= int(request
.GET
.get('page', 1))
62 return exc
.HTTPNotFound()
64 return exc
.HTTPNotFound()
66 return controller(request
, page
=page
, *args
, **kwargs
)
68 return _make_safe(wrapper
, controller
)
71 def get_user_media_entry(controller
):
73 Pass in a MediaEntry based off of a url component
75 def wrapper(request
, *args
, **kwargs
):
76 user
= request
.db
.User
.find_one(
77 {'username': request
.matchdict
['user']})
80 return exc
.HTTPNotFound()
82 media
= request
.db
.MediaEntry
.find_one(
83 {'slug': request
.matchdict
['media'],
85 'uploader': user
['_id']})
87 # no media via slug? Grab it via ObjectId
90 media
= request
.db
.MediaEntry
.find_one(
91 {'_id': ObjectId(request
.matchdict
['media']),
93 'uploader': user
['_id']})
95 return exc
.HTTPNotFound()
97 # Still no media? Okay, 404.
99 return exc
.HTTPNotFound()
101 return controller(request
, media
=media
, *args
, **kwargs
)
103 return _make_safe(wrapper
, controller
)
105 def get_media_entry_by_id(controller
):
107 Pass in a MediaEntry based off of a url component
109 def wrapper(request
, *args
, **kwargs
):
111 media
= request
.db
.MediaEntry
.find_one(
112 {'_id': ObjectId(request
.matchdict
['media']),
113 'state': 'processed'})
115 return exc
.HTTPNotFound()
117 # Still no media? Okay, 404.
119 return exc
.HTTPNotFound()
121 return controller(request
, media
=media
, *args
, **kwargs
)
123 return _make_safe(wrapper
, controller
)