From ebf58264e84899f90a74743b09fb55d046d828ce Mon Sep 17 00:00:00 2001 From: Jason McBrayer Date: Tue, 23 Apr 2019 18:18:11 -0400 Subject: [PATCH] Add accounts_dict to session, update it on logging in. --- brutaldon/views.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/brutaldon/views.py b/brutaldon/views.py index bdd2f53..9c09f7a 100644 --- a/brutaldon/views.py +++ b/brutaldon/views.py @@ -300,8 +300,16 @@ def oauth_callback(request): client = Client.objects.get(api_base_id=request.session['active_instance']), preferences = preferences) account.save() + request.session['active_user'] = user request.session['active_username'] = user.username + '@' + request.session['active_instance_hostname'] + + accounts_dict = request.session.get('accounts_dict') + if not accounts_dict: + accounts_dict = {} + accounts_dict[account.username] = { account_id: account.id, user: user } + request.session['accounts_dict'] = accounts_dict + return redirect(home) @@ -365,6 +373,13 @@ def old_login(request): account.username = request.session['active_username'] request.session['timezone'] = account.preferences.timezone; account.save() + + accounts_dict = request.session.get('accounts_dict') + if not accounts_dict: + accounts_dict = {} + accounts_dict[account.username] = { account_id: account.id, user: user } + request.session['accounts_dict'] = accounts_dict + return redirect(home) except Exception as ex: