heckyel/mediagoblin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PEP8-ification.

Browse Source
This commit is contained in:
Nathan Yergler 2011-10-01 12:48:43 -07:00
parent 6bfbe02426
commit 5d2abe45b2
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mediagoblin/middleware/csrf.py
View File

@ -37,6 +37,7 @@ class CsrfForm(Form):
csrf_token = HiddenField("",
[validators.Required()])
def render_csrf_form_token(request):
"""Render the CSRF token in a format suitable for inclusion in a
form."""
@ -45,6 +46,7 @@ def render_csrf_form_token(request):
return form.csrf_token
class CsrfMiddleware(object):
"""CSRF Protection Middleware
@ -87,7 +89,8 @@ class CsrfMiddleware(object):
response.set_cookie(
mg_globals.app_config['csrf_cookie_name'],
request.environ['CSRF_TOKEN'],
max_age=60*60*24*7*52, path='/',
max_age=60 * 60 * 24 * 7 * 52,
path='/',
domain=mg_globals.app_config.get('csrf_cookie_domain', None),
secure=(request.scheme.lower() == 'https'),
httponly=True)
@ -100,8 +103,7 @@ class CsrfMiddleware(object):
return hashlib.md5("%s%s" %
(randrange(0, self.MAX_CSRF_KEY),
mg_globals.app_config['secret_key'])
).hexdigest()
mg_globals.app_config['secret_key'])).hexdigest()
def verify_tokens(self, request):
"""Verify that the CSRF Cookie exists and that it matches the
@ -128,4 +130,3 @@ class CsrfMiddleware(object):
# either the tokens didn't match or the form token wasn't
# present; either way, the request is denied
return HTTPForbidden()