userfactory.py

""" ProfileFacotry"""
from user import User
from staff import Staff


class UserFactory:
    # FACTORY class
    # Group of Four: FACTORY METHOD

    @staticmethod
    def factory(role):
        if role is not None:
            if role.lower() == "staff":
                return Staff()
            elif role.lower() == "user":
                return User()
            else:
                return None