from opnfv_testapi.common import constants
from opnfv_testapi.common import raises
from opnfv_testapi.resources import handlers
from opnfv_testapi.resources import models


class User(models.ModelBase):
    def __init__(self, user=None, email=None, fullname=None, groups=None):
        self.user = user
        self.email = email
        self.fullname = fullname
        self.groups = groups


class UserHandler(handlers.GenericApiHandler):
    def __init__(self, application, request, **kwargs):
        super(UserHandler, self).__init__(application, request, **kwargs)
        self.table = 'users'
        self.table_cls = User

    def get(self):
        username = self.get_secure_cookie(constants.TESTAPI_ID)
        if username:
            self._get_one(query={'user': username})
        else:
            raises.Unauthorized('Unauthorized')