console.py

#!/usr/bin/python3
''' Module class console.py '''
import cmd
import ast
import shlex
import models
from models.base_model import BaseModel
from models.user import User
from models.state import State
from models.city import City
from models.amenity import Amenity
from models.place import Place
from models.review import Review


class_ls = {'BaseModel': BaseModel, 'User': User,
            'State': State, 'City': City, 'Amenity': Amenity,
            'Place': Place, 'Review': Review}
store_all = models.storage.all()


class HBNBCommand(cmd.Cmd):
    ''' Class command interpreter HBNB '''

    def do_EOF(self, enter):
        ''' Close the command interpreter when you use the \'EOF\' command '''
        return True

    def do_quit(self, enter):
        ''' Close the command interpreter when you use the \'quit\' command '''
        return True

    def emptyline(self):
        ''' Clean the last nonempty command entered '''
        return False

    def do_create(self, args):
        ''' Create a new class including his id number '''
        if not args:
            print('** class name missing **')
        else:
            arg = args.split()
            if arg[0] in class_ls:
                exc = class_ls[arg[0]]()
                print(exc.id)
                exc.save()
            else:
                print('** class doesn\'t exist **')

    def do_show(self, args):
        ''' Print the object with id specified and his dictionary '''
        if not args:
            print('** class name missing **')
        else:
            arg = args.split()
            if not arg[0] in class_ls:
                print('** class doesn\'t exist **')
            else:
                if len(arg) > 1:
                    class_key = ''
                    class_key = arg[0] + '.' + arg[1]
                    if class_key in store_all:
                        print(store_all[class_key])
                    else:
                        print('** no instance found **')
                else:
                    print('** instance id missing **')

    def do_destroy(self, args):
        ''' Removes an object with id specified and his dictionary '''
        if not args:
            print('** class name missing **')
        else:
            arg = args.split()
            if not arg[0] in class_ls:
                print('** class doesn\'t exist **')
            else:
                if len(arg) > 1:
                    class_key = ''
                    class_key = arg[0] + '.' + arg[1]
                    if class_key in store_all:
                        store_all.pop(class_key)
                        models.storage.save()
                    else:
                        print('** no instance found **')
                else:
                    print('** instance id missing **')

    def do_all(self, args):
        '''
        Prints all string representation of all instances
        based or not on the class name
        ex: \'(hbnb ) all\' or \'(hbnb )\' all BaseModel
        '''
        ls_val = []
        arg = args.split()
        if len(arg) == 0:
            for val in store_all.values():
                ls_val.append(str(val))
            print(ls_val)
        elif arg[0] in class_ls:
            for class_key in store_all:
                if arg[0] in class_key:
                    ls_val.append(str(store_all[class_key]))
            print(ls_val)
        else:
            print('** class doesn\'t exist **')

    def do_update(self, args):
        '''
        Updates an instance based on the class name and
        id by adding or updating attribute.
        ex: \'(hbnb )\' update BaseModel 123456789 attr_name attr_value
        '''
        arg = shlex.split(args)
        length = len(arg)
        if length == 0:
            print('** class name missing **')
            return
        if not arg[0] in class_ls:
            print('** class doesn\'t exist **')
            return
        else:
            if length == 1:
                print('** instance id missing **')
                return
            else:
                class_key = ''
                class_key = arg[0] + '.' + arg[1]
            if length == 2:
                print('** attribute name missing **')
                return
            if length == 3:
                print('** value missing **')
                return
            if length >= 4:
                if class_key in store_all:
                    setattr(store_all[class_key], arg[2], arg[3])
                    models.storage.save()
                else:
                    print('** no instance found **')
                    return

    def count(self, args):
        '''
        count all string representation of all instances
        based or not on the class name
        ex: \'(hbnb ) all\' or \'(hbnb )\' all BaseModel
        '''
        count = 0
        ls_val = []
        arg = args.split()
        if arg[0] in class_ls:
            for class_key in store_all:
                if arg[0] in class_key:
                    ls_val.append(str(store_all[class_key]))
                    count += 1
            print(count)
        else:
            print('** class doesn\'t exist **')

    @staticmethod
    def do_BaseModel(args):
        '''
        do_BaseModel, use the class.command of the console as input
        while execute the command inserted
        '''
        if args:
            fnd = args[args.find('("') + 2:args.find('")')]
            command = args[0:args.find('(')]
            arg = command.split(".")
            if arg[1] == 'all':
                HBNBCommand.do_all(HBNBCommand, 'BaseModel')
            if arg[1] == 'count':
                HBNBCommand.count(HBNBCommand, 'BaseModel')
            if arg[1] == 'show':
                HBNBCommand.do_show(HBNBCommand, 'BaseModel {}'.format(fnd))
            if arg[1] == 'destroy':
                HBNBCommand.do_destroy(HBNBCommand, 'BaseModel {}'.format(fnd))
            if arg[1] == 'update':
                ls_fd = fnd.split("\", \"")
                if len(ls_fd) >= 3:
                    st = 'BaseModel {} {} {}'.format(ls_fd[0],
                                                     ls_fd[1], ls_fd[2])
                    HBNBCommand.do_update(HBNBCommand, st)
                else:
                    fnd3 = fnd[0:fnd.find('"')]
                    fnd4 = fnd[fnd.find('", ') + 3:len(args) - 1]
                    dc = ast.literal_eval(fnd4)
                    for key, val in dc.items():
                        st = 'BaseModel {} {} {}'.format(fnd3, key, val)
                        HBNBCommand.do_update(HBNBCommand, st)

    @staticmethod
    def do_User(args):
        '''
        do_User, use the class.command of the console as input
        while execute the command inserted
        '''
        if args:
            fnd = args[args.find('("') + 2:args.find('")')]
            command = args[0:args.find('(')]
            arg = command.split(".")
            if arg[1] == 'all':
                HBNBCommand.do_all(HBNBCommand, 'User')
            if arg[1] == 'count':
                HBNBCommand.count(HBNBCommand, 'User')
            if arg[1] == 'show':
                HBNBCommand.do_show(HBNBCommand, 'User {}'.format(fnd))
            if arg[1] == 'destroy':
                HBNBCommand.do_destroy(HBNBCommand, 'User {}'.format(fnd))
            if arg[1] == 'update':
                ls_fd = fnd.split("\", \"")
                if len(ls_fd) >= 3:
                    st = 'User {} {} {}'.format(ls_fd[0], ls_fd[1], ls_fd[2])
                    HBNBCommand.do_update(HBNBCommand, st)
                else:
                    fnd3 = fnd[0:fnd.find('"')]
                    fnd4 = fnd[fnd.find('", ') + 3:len(args) - 1]
                    dc = ast.literal_eval(fnd4)
                    for key, val in dc.items():
                        st = 'User {} {} {}'.format(fnd3, key, val)
                        HBNBCommand.do_update(HBNBCommand, st)

    @staticmethod
    def do_State(args):
        '''
        do_State, use the class.command of the console as input
        while execute the command inserted
        '''
        if args:
            fnd = args[args.find('("') + 2:args.find('")')]
            command = args[0:args.find('(')]
            arg = command.split(".")
            if arg[1] == 'all':
                HBNBCommand.do_all(HBNBCommand, 'State')
            if arg[1] == 'count':
                HBNBCommand.count(HBNBCommand, 'State')
            if arg[1] == 'show':
                HBNBCommand.do_show(HBNBCommand, 'State {}'.format(fnd))
            if arg[1] == 'destroy':
                HBNBCommand.do_destroy(HBNBCommand, 'State {}'.format(fnd))
            if arg[1] == 'update':
                ls_fd = fnd.split("\", \"")
                if len(ls_fd) >= 3:
                    st = 'State {} {} {}'.format(ls_fd[0], ls_fd[1], ls_fd[2])
                    HBNBCommand.do_update(HBNBCommand, st)
                else:
                    fnd3 = fnd[0:fnd.find('"')]
                    fnd4 = fnd[fnd.find('", ') + 3:len(args) - 1]
                    dc = ast.literal_eval(fnd4)
                    for key, val in dc.items():
                        st = 'State {} {} {}'.format(fnd3, key, val)
                        HBNBCommand.do_update(HBNBCommand, st)

    @staticmethod
    def do_City(args):
        '''
        do_City, use the class.command of the console as input
        while execute the command inserted
        '''
        if args:
            fnd = args[args.find('("') + 2:args.find('")')]
            command = args[0:args.find('(')]
            arg = command.split(".")
            if arg[1] == 'all':
                HBNBCommand.do_all(HBNBCommand, 'City')
            if arg[1] == 'count':
                HBNBCommand.count(HBNBCommand, 'City')
            if arg[1] == 'show':
                HBNBCommand.do_show(HBNBCommand, 'City {}'.format(fnd))
            if arg[1] == 'destroy':
                HBNBCommand.do_destroy(HBNBCommand, 'City {}'.format(fnd))
            if arg[1] == 'update':
                ls_fd = fnd.split("\", \"")
                if len(ls_fd) >= 3:
                    st = 'City {} {} {}'.format(ls_fd[0], ls_fd[1], ls_fd[2])
                    HBNBCommand.do_update(HBNBCommand, st)
                else:
                    fnd3 = fnd[0:fnd.find('"')]
                    fnd4 = fnd[fnd.find('", ') + 3:len(args) - 1]
                    dc = ast.literal_eval(fnd4)
                    for key, val in dc.items():
                        st = 'City {} {} {}'.format(fnd3, key, val)
                        HBNBCommand.do_update(HBNBCommand, st)

    @staticmethod
    def do_Amenity(args):
        '''
        do_Amenity, use the class.command of the console as input
        while execute the command inserted
        '''
        if args:
            fnd = args[args.find('("') + 2:args.find('")')]
            command = args[0:args.find('(')]
            arg = command.split(".")
            if arg[1] == 'all':
                HBNBCommand.do_all(HBNBCommand, 'Amenity')
            if arg[1] == 'count':
                HBNBCommand.count(HBNBCommand, 'Amenity')
            if arg[1] == 'show':
                HBNBCommand.do_show(HBNBCommand, 'Amenity {}'.format(fnd))
            if arg[1] == 'destroy':
                HBNBCommand.do_destroy(HBNBCommand, 'Amenity {}'.format(fnd))
            if arg[1] == 'update':
                ls_fd = fnd.split("\", \"")
                if len(ls_fd) >= 3:
                    st = 'Amenity {} {} {}'.format(ls_fd[0],
                                                   ls_fd[1], ls_fd[2])
                    HBNBCommand.do_update(HBNBCommand, st)
                else:
                    fnd3 = fnd[0:fnd.find('"')]
                    fnd4 = fnd[fnd.find('", ') + 3:len(args) - 1]
                    dc = ast.literal_eval(fnd4)
                    for key, val in dc.items():
                        st = 'Amenity {} {} {}'.format(fnd3, key, val)
                        HBNBCommand.do_update(HBNBCommand, st)

    @staticmethod
    def do_Place(args):
        '''
        do_Place, use the class.command of the console as input
        while execute the command inserted
        '''
        if args:
            fnd = args[args.find('("') + 2:args.find('")')]
            command = args[0:args.find('(')]
            arg = command.split(".")
            if arg[1] == 'all':
                HBNBCommand.do_all(HBNBCommand, 'Place')
            if arg[1] == 'count':
                HBNBCommand.count(HBNBCommand, 'Place')
            if arg[1] == 'show':
                HBNBCommand.do_show(HBNBCommand, 'Place {}'.format(fnd))
            if arg[1] == 'destroy':
                HBNBCommand.do_destroy(HBNBCommand, 'Place {}'.format(fnd))
            if arg[1] == 'update':
                ls_fd = fnd.split("\", \"")
                if len(ls_fd) >= 3:
                    st = 'Place {} {} {}'.format(ls_fd[0], ls_fd[1], ls_fd[2])
                    HBNBCommand.do_update(HBNBCommand, st)
                else:
                    fnd3 = fnd[0:fnd.find('"')]
                    fnd4 = fnd[fnd.find('", ') + 3:len(args) - 1]
                    dc = ast.literal_eval(fnd4)
                    for key, val in dc.items():
                        st = 'Place {} {} {}'.format(fnd3, key, val)
                        HBNBCommand.do_update(HBNBCommand, st)

    @staticmethod
    def do_Review(args):
        '''
        do_Review, use the class.command of the console as input
        while execute the command inserted
        '''
        if args:
            fnd = args[args.find('("') + 2:args.find('")')]
            command = args[0:args.find('(')]
            arg = command.split(".")
            if arg[1] == 'all':
                HBNBCommand.do_all(HBNBCommand, 'Review')
            if arg[1] == 'count':
                HBNBCommand.count(HBNBCommand, 'Review')
            if arg[1] == 'show':
                HBNBCommand.do_show(HBNBCommand, 'Review {}'.format(fnd))
            if arg[1] == 'destroy':
                HBNBCommand.do_destroy(HBNBCommand, 'Review {}'.format(fnd))
            if arg[1] == 'update':
                ls_fd = fnd.split("\", \"")
                if len(ls_fd) >= 3:
                    st = 'Review {} {} {}'.format(ls_fd[0], ls_fd[1], ls_fd[2])
                    HBNBCommand.do_update(HBNBCommand, st)
                else:
                    fnd3 = fnd[0:fnd.find('"')]
                    fnd4 = fnd[fnd.find('", ') + 3:len(args) - 1]
                    dc = ast.literal_eval(fnd4)
                    for key, val in dc.items():
                        st = 'Review {} {} {}'.format(fnd3, key, val)
                        HBNBCommand.do_update(HBNBCommand, st)


if __name__ == '__main__':
    HBNBCommand.prompt = '(hbnb) '
    HBNBCommand().cmdloop()