import os

from django.contrib.auth import login
from django.contrib.auth.models import User
from django.core.exceptions import ObjectDoesNotExist
from requests import get

from SprintLib.BaseView import BaseView


class VKAddView(BaseView):
    required_login = True
    endpoint = "vk_add"
    view_file = "vk_auth.html"
    fields_except = ('user_id',)

    def get(self):
        if not self.request.GET:
            return
        access_token = self.request.GET['access_token']
        token = os.getenv("VK_SERVICE_TOKEN")
        resp = get(f'https://api.vk.com/method/secure.checkToken?token={access_token}&access_token={token}').json()
        print("Got response while adding user", resp)
        if 'success' in resp and resp['success'] == 1:
            user_id = resp['user_id']
            self.request.user.userinfo.vk_user_id = user_id
            self.request.user.userinfo.save()
        return '/account'