Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move identify_user function to views module.
In preparation for reuse by pylons.
- Loading branch information
Showing
2 changed files
with
118 additions
and
111 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,115 @@ | ||
| from flask import request, redirect | ||
|
|
||
| import ckan.model as model | ||
| from ckan.views.api import APIKEY_HEADER_NAME_DEFAULT | ||
| from ckan.common import c | ||
| import ckan.plugins as p | ||
|
|
||
| import logging | ||
| log = logging.getLogger(__name__) | ||
|
|
||
|
|
||
| def identify_user(): | ||
| '''Try to identify the user | ||
| If the user is identified then: | ||
| c.user = user name (unicode) | ||
| c.userobj = user object | ||
| c.author = user name | ||
| otherwise: | ||
| c.user = None | ||
| c.userobj = None | ||
| c.author = user's IP address (unicode)''' | ||
| # see if it was proxied first | ||
| c.remote_addr = request.environ.get('HTTP_X_FORWARDED_FOR', '') | ||
| if not c.remote_addr: | ||
| c.remote_addr = request.environ.get('REMOTE_ADDR', | ||
| 'Unknown IP Address') | ||
|
|
||
| # Authentication plugins get a chance to run here break as soon as a user | ||
| # is identified. | ||
| authenticators = p.PluginImplementations(p.IAuthenticator) | ||
| if authenticators: | ||
| for item in authenticators: | ||
| item.identify() | ||
| if c.user: | ||
| break | ||
|
|
||
| # We haven't identified the user so try the default methods | ||
| if not getattr(c, 'user', None): | ||
| _identify_user_default() | ||
|
|
||
| # If we have a user but not the userobj let's get the userobj. This means | ||
| # that IAuthenticator extensions do not need to access the user model | ||
| # directly. | ||
| if c.user and not getattr(c, 'userobj', None): | ||
| c.userobj = model.User.by_name(c.user) | ||
|
|
||
| # general settings | ||
| if c.user: | ||
| c.author = c.user | ||
| else: | ||
| c.author = c.remote_addr | ||
| c.author = unicode(c.author) | ||
|
|
||
|
|
||
| def _identify_user_default(): | ||
| ''' | ||
| Identifies the user using two methods: | ||
| a) If they logged into the web interface then repoze.who will | ||
| set REMOTE_USER. | ||
| b) For API calls they may set a header with an API key. | ||
| ''' | ||
|
|
||
| # environ['REMOTE_USER'] is set by repoze.who if it authenticates a | ||
| # user's cookie. But repoze.who doesn't check the user (still) exists | ||
| # in our database - we need to do that here. (Another way would be | ||
| # with an userid_checker, but that would mean another db access. | ||
| # See: http://docs.repoze.org/who/1.0/narr.html#module-repoze.who\ | ||
| # .plugins.sql ) | ||
| c.user = request.environ.get('REMOTE_USER', '') | ||
| if c.user: | ||
| c.user = c.user.decode('utf8') | ||
| c.userobj = model.User.by_name(c.user) | ||
| if c.userobj is None or not c.userobj.is_active(): | ||
|
|
||
| # This occurs when a user that was still logged in is deleted, or | ||
| # when you are logged in, clean db and then restart (or when you | ||
| # change your username). There is no user object, so even though | ||
| # repoze thinks you are logged in and your cookie has | ||
| # ckan_display_name, we need to force user to logout and login | ||
| # again to get the User object. | ||
|
|
||
| ev = request.environ | ||
| if 'repoze.who.plugins' in ev: | ||
| pth = getattr(ev['repoze.who.plugins']['friendlyform'], | ||
| 'logout_handler_path') | ||
| redirect(pth) | ||
| else: | ||
| c.userobj = _get_user_for_apikey() | ||
| if c.userobj is not None: | ||
| c.user = c.userobj.name | ||
|
|
||
|
|
||
| def _get_user_for_apikey(): | ||
| # TODO: use config | ||
| # apikey_header_name = config.get(APIKEY_HEADER_NAME_KEY, | ||
| # APIKEY_HEADER_NAME_DEFAULT) | ||
| apikey_header_name = APIKEY_HEADER_NAME_DEFAULT | ||
| apikey = request.headers.get(apikey_header_name, '') | ||
| if not apikey: | ||
| apikey = request.environ.get(apikey_header_name, '') | ||
| if not apikey: | ||
| # For misunderstanding old documentation (now fixed). | ||
| apikey = request.environ.get('HTTP_AUTHORIZATION', '') | ||
| if not apikey: | ||
| apikey = request.environ.get('Authorization', '') | ||
| # Forget HTTP Auth credentials (they have spaces). | ||
| if ' ' in apikey: | ||
| apikey = '' | ||
| if not apikey: | ||
| return None | ||
| log.debug("Received API Key: %s" % apikey) | ||
| apikey = unicode(apikey) | ||
| query = model.Session.query(model.User) | ||
| user = query.filter_by(apikey=apikey).first() | ||
| return user |