Commit | Line | Data |
---|---|---|
03ae172a AW |
1 | # GNU 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 mediagoblin.db.util import ObjectId | |
18 | ||
19 | def setup_user_in_request(request): | |
20 | """ | |
21 | Examine a request and tack on a request.user parameter if that's | |
22 | appropriate. | |
23 | """ | |
24 | if not request.session.has_key('user_id'): | |
25 | request.user = None | |
26 | return | |
27 | ||
28 | user = None | |
29 | user = request.app.db.User.one( | |
30 | {'_id': ObjectId(request.session['user_id'])}) | |
31 | ||
32 | if not user: | |
33 | # Something's wrong... this user doesn't exist? Invalidate | |
34 | # this session. | |
35 | request.session.invalidate() | |
36 | ||
37 | request.user = user |