from .models import User


class StorageUser:
    def __init__(self, id, login, email, password, password2):
        self.id = id
        self.login = login
        self.email = email
        self.password = password
        self.password2 = password2

    @classmethod
    def create_user_from_dict(cls, user_object, save_to_db=True):
        # TODO DODAJ WALIDACJE CZY JEST INSTANCJA Z DANYMI
        user = StorageUser(
            id=user_object['id'],
            login=user_object['login'],
            email=user_object['email'],
            password=user_object['password'],
        )

        if save_to_db:
            return cls.create_user_db(user)

    def create_user_db(self, user):
        #TODO POMYSL TUTAJ NAD WALIDACJA
        user = User(id=user['id'],
                    login=user['login'],
                    email=user['email'],
                    password=user['password'],
                    )
        return user

    def get_user_by_id(self, id):
        user = User.query.filter(User.id == id)
        return user

    def get_user_by_login(self, login):
        user = User.query.filter(User.login == login)
        return user