diff --git a/rowers/idoklad.py b/rowers/idoklad.py index 4156e12e..cb6f2593 100644 --- a/rowers/idoklad.py +++ b/rowers/idoklad.py @@ -59,10 +59,15 @@ def idoklad_token(): response = requests.post('https://identity.idoklad.cz/server/connect/token', headers=headers, data=data) if response.status_code == 200: - token = response.json() - token['updated_at'] = timezone.now() - token = iDokladToken.objects.filter(id=1).update(**token) - return iDokladToken.objects.get(id=1) + token_json = response.json() + token = iDokladToken.objects.all().order_by('-updated_at')[0] + token.access_token = token_json['access_token'] + token.id_token = token_json['id_token'] + token.refresh_token = token_json['refresh_token'] + token.expires_in = token_json['expires_in'] + token.updated_at = timezone.now() + token.save() + return token else: return None