console.py

#!/usr/bin/python3
"""contains the entry point of the command interpreter
"""
import cmd
import re

from models import storage
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 HBNBCommand(cmd.Cmd):
    """entry point of the command interpreter
    """
    prompt = "(hbnb) "
    airbnb_engine_classes = {
        "User": User,
        "BaseModel": BaseModel,
        "Place": Place,
        "State": State,
        "City": City,
        "Amenity": Amenity,
        "Review": Review
    }

    def emptyline(self):
        """Handles empty line
        """
        pass

    def do_EOF(self, arg):
        """exit the program
        """
        return True

    def do_quit(self, arg):
        """exit the program
        """
        return True

    def do_create(self, arg):
        """Creates a new instance of BaseModel,
            saves it (to the JSON file)
            and prints the id
            Ex: $ create BaseModel
        """
        if not arg:
            print("** class name missing **")
            return

        try:
            args = arg.split()
            base_model = self.airbnb_engine_classes[args[0]]()
            base_model.save()
            print(base_model.id)
        except KeyError as ex:
            print("** class doesn't exist **")
            return

    def do_show(self, arg):
        """
        Prints the string representation
        of an instance based on the class name and id
        $ show BaseModel 1234-1234-1234
        """
        if not arg:
            print("** class name missing **")
            return

        try:
            class_name, id = arg.split()
            if class_name in self.airbnb_engine_classes:
                key = class_name + "." + id
                objects = storage.all()
                if key not in objects:
                    print("** no instance found **")
                    return
                else:
                    print(objects[key])
                    return

            else:
                print("** class doesn't exist ***")
                return

        except ValueError as ex:
            print("** instance id missing **")
            return

    def do_destroy(self, arg):
        """
        Deletes an instance based on the class name
        and id (save the change into the JSON file)
        Ex: $ destroy BaseModel 1234-1234-1234
        """
        if not arg:
            print("** class name missing **")
            return

        try:
            class_name, id = arg.split()
            if class_name in self.airbnb_engine_classes:
                key = class_name + "." + id
                objects = storage.all()
                if key not in objects:
                    print("** no instance found **")
                    return
                else:
                    del objects[key]
                    storage.save()
                    return

            else:
                print("** class doesn't exist **")
                return

        except ValueError as ex:
            print("** instance id missing **")
            return

    def do_all(self, arg):
        """
        Prints all string representation of all
        instances based or not on the class name
        $ all BaseModel or $ all.
        """
        value_list = []
        objects = storage.all()

        if not arg:
            for value in objects.values():
                value_list.append(value.__str__())
            print(value_list)
            return

        if arg in self.airbnb_engine_classes:
            for value in objects.values():
                if arg == type(value).__name__:
                    value_list.append(value.__str__())
            print(value_list)
            return
        else:
            print("** class doesn't exist **")
            return

    def do_update(self, arg):
        """
        Updates an instance based on the class
        name and id by adding or updating attribute
        (save the change into the JSON file)
        Ex: $ update BaseModel 1234-1234-1234
            email "aibnb@mail.com".
        """
        arg_list = arg.split()
        length = len(arg_list)
        if length == 0:
            print("** class name missing **")
            return

        base_model = arg_list[0]

        if base_model not in self.airbnb_engine_classes:
            print("** class doesn't exist **")
            return

        if length == 1:
            print("** instance id missing **")
            return

        objects = storage.all()

        if length > 1:
            obj_key = arg_list[0] + '.' + arg_list[1]

        if obj_key not in objects:
            print("** no instance found **")
            return

        if length == 2:
            print("** attribute name missing **")
            return

        if length == 3:
            print("** value missing **")
            return

        class_instance = objects[obj_key]
        attribute, value = arg_list[2], arg_list[3]

        setattr(class_instance, attribute, value)
        class_instance.save()

    def default(self, arg):
        """Handles defaults arguments not created
        """
        count = 0
        args = arg.split(".")

        if args[0] in self.airbnb_engine_classes and args[1] == "all()":
            self.do_all(args[0])
        elif args[0] in self.airbnb_engine_classes and args[1] == "count()":
            if (args[0] not in self.airbnb_engine_classes):
                print("** class doesn't exist **")
                return (False)
            else:
                for key in storage.all():
                    if key.startswith(args[0]):
                        count += 1
                print(count)
        elif args[0] in self.airbnb_engine_classes and \
                args[1].startswith('show'):
            arg = args[1].split('"')
            if len(arg) == 3:
                arg1 = args[0] + " " + arg[1]
                self.do_show(arg1)
        elif args[0] in self.airbnb_engine_classes and \
                args[1].startswith('destroy'):
            arg = args[1].split('"')
            if len(arg) == 3:
                arg1 = args[0] + " " + arg[1]
                self.do_destroy(arg1)
        elif args[0] in self.airbnb_engine_classes and \
                args[1].startswith('update'):
            start = 'update('
            end = ')'
            arg = re.findall(re.escape(start)+"(.*)" +
                             re.escape(end), args[1])[0]
            arg = arg.replace('(', '').replace(')', '').replace(',', '')
            arg = arg.replace('"', '')
            arg1 = args[0] + " " + arg
            self.do_update(arg1)
        else:
            print("*** Unknown syntax: {}".format(arg))


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