neelxie/ireport-api

""" This is the Users' models file."""
import datetime
from ..utility.validation import Valid


class Base:
    """ This is the base class for a person and
        holds the person's names and phone number.
    """

    def __init__(self, first_name, last_name, other_name, phone_number):
        """ Constructor for the base class.
        """
        self.first_name = first_name
        self.last_name = last_name
        self.other_name = other_name
        self.phone_number = phone_number


class Credential:
    """ Class to hold the necessary log in credentials.
    """

    def __init__(self, email, user_name, password):
        """ Initializing the class will require the above attributes.
        """
        self.email = email
        self.user_name = user_name
        self.password = password


class User:
    """ Class for comprehensive Users.
    """

    def __init__(self, base, credential, is_admin, user_id):
        """ Using composition, integrate base class, credential and
            other remaining attributes to make a complete User Class.
        """
        self.base = base
        self.credential = credential
        self.is_admin = is_admin
        self.user_id = user_id
        self.registered = str(datetime.datetime.now())

    def to_dict(self):
        """ Method to change the user class to a JSON object for retrieval.
        """
        return {
            "first_name": self.base.first_name,
            "last_name": self.base.last_name,
            "other_name": self.base.other_name,
            "phone_number": self.base.phone_number,
            "email": self.credential.email,
            "user_name": self.credential.user_name,
            "password": self.credential.password,
            "is_admin": self.is_admin,
            "user_id": self.user_id,
            "registered": self.registered
        }


class UserDB:
    """ App users will be stored in this class.
    """
    valid = Valid()

    def __init__(self):
        """ app users will be held in a list called all_users.
        """
        self.all_users = []

    def create_user(self, user):
        """ Method for adding a user.
        """
        return self.all_users.append(user)

    def single_user(self, user_id):
        """ Method to get an app user by ID.
        """
        for me_as_user in self.all_users:
            if me_as_user.user_id == user_id:
                return me_as_user
        return None

    def checking_user(self, user_name, email):
        """ Check whether username or email already exist in list.
        """
        temp = [
            user for user in self.all_users if user.credential.user_name == user_name]
        mail = [user for user in self.all_users if user.credential.email == email]

        if len(temp) > 0:
            return "User name is already taken."
        elif len(mail) > 0:
            return "Email already has an account."
        else:
            return None

    def validate_login(self, user_name, password):
        error = self.valid.validate_login(user_name, password)

        if error:
            return error

        temp = [
            user for user in self.all_users if user.credential.user_name == user_name]

        if len(temp) != 1:
            return "Username not found. Please sign up."

        if temp[0].credential.password != password:
            return "Wrong Password"

        return None