From 600ffe25e1d332852f31756a38f6052d876b0c90 Mon Sep 17 00:00:00 2001 From: Theron Luhn Date: Mon, 15 Apr 2019 19:15:36 -0700 Subject: Use SessionAuthenticationHelper in SessionAuthenticationPolicy. --- src/pyramid/authentication.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/src/pyramid/authentication.py b/src/pyramid/authentication.py index 4f8077309..c1e010516 100644 --- a/src/pyramid/authentication.py +++ b/src/pyramid/authentication.py @@ -1099,23 +1099,19 @@ class SessionAuthenticationPolicy(CallbackAuthenticationPolicy): def __init__(self, prefix='auth.', callback=None, debug=False): self.callback = callback - self.prefix = prefix or '' - self.userid_key = prefix + 'userid' self.debug = debug + self.helper = SessionAuthenticationHelper(prefix) def remember(self, request, userid, **kw): """ Store a userid in the session.""" - request.session[self.userid_key] = userid - return [] + return self.helper.remember(request, userid, **kw) def forget(self, request): """ Remove the stored userid from the session.""" - if self.userid_key in request.session: - del request.session[self.userid_key] - return [] + return self.helper.forget(request) def unauthenticated_userid(self, request): - return request.session.get(self.userid_key) + return self.helper.identify(request) class SessionAuthenticationHelper: -- cgit v1.2.3