Make sure newly created accounts have preferences attached

This commit is contained in:
Jason McBrayer 2018-06-19 06:26:35 -04:00
parent 6cbb2d10ed
commit cbb5b544ed
1 changed files with 14 additions and 1 deletions

View File

@ -216,6 +216,17 @@ def oauth_callback(request):
redirect_uri=redirect_uri)
request.session['access_token'] = access_token
user = mastodon.account_verify_credentials()
try:
account = Account.objects.get(username=username, client_id=client.id)
account.access_token = access_token
account.save()
except (Account.DoesNotExist, Account.MultipleObjectsReturned):
preferences = Preferences(theme = Theme.objects.get(1))
account = Account(username=user.acct,
access_token = access_token,
client = Client.objects.get(api_base_id=request.session['instance']),
preferences = preferences)
account.save()
request.session['user'] = user
return redirect(home)
@ -259,10 +270,12 @@ def old_login(request):
try:
account = Account.objects.get(username=username, client_id=client.id)
except (Account.DoesNotExist, Account.MultipleObjectsReturned):
preferences = Preferences(theme = Theme.objects.get(1))
account = Account(
username = username,
access_token = "",
client = client)
client = client,
preferences = preferences)
try:
access_token = mastodon.log_in(username,
password)