import os

from flask.ext.login import make_secure_token

from common.security import pwd_context, secret_key


class User:
    def __init__(self, **kwargs):
        for (key, value) in kwargs.items():
            setattr(self, key, value)

    def get_id(self):
        if self.id:
            return str(self.id)
        else:
            raise ValueError("No user")

    def is_authenticated(self):
        if self.authenticated > 0:
            return True
        else:
            return False

    def is_active(self):
        if self.active > 0:
            return True
        else:
            return False

    def is_anonymous(self):
        if self.anonymous > 0:
            return True
        else:
            return False

    def verify(self, password):
        if self.id and pwd_context.verify(password, self.password_hash):
            self.authenticated = 1
            return True
        else:
            return False

    def new_password(self, password, category=None):
        if self.id:
            hash = None

            if category:
                hash = pwd_context.encrypt(password, category=category)
            else:
                hash = pwd_context.encrypt(password)

            api_key = make_secure_token(hash, os.urandom(64), key=secret_key)

            return hash, api_key

        else:
            raise ValueError("No user")