Scissors - Stella

main-sub-menu.py

@@ -0,0 +1,263 @@
+import requests
+from video_store import VideoStore
+
+URL = "http://127.0.0.1:5000"
+BACKUP_URL = "https://retro-video-store-api.herokuapp.com"
+
+def print_stars():
+    print("\n**************************\n")
+
+
+def main_menu():
+    menu = {
+        "1": "Video Management", 
+        "2": "Customer Management",
+        "3": "Rental Management (video check-ins/outs)", 
+        "END" : "Quit"        
+    }
+
+    print_stars()
+    print("WELCOME TO STELLA'S RETRO VIDEO STOR.")
+    print("Please select from the following main menu.")
+    print_stars()
+
+    for item_num in menu:
+        print(f"Option {item_num}: {menu[item_num]}")
+
+    print_stars()
+    return menu
+
+def video_management(): 
+    options = {
+        "1": "Add a video", 
+        "2": "Edit a video",
+        "3": "Delete a video", 
+        "4": "Get information of all videos", 
+        "5": "Get information of one video", 
+        "*": "List all options",
+        "MAIN": "Return to main menu",
+        "END": "Quit"
+        }
+
+    for choice_num in options:
+        print(f"Option {choice_num}. {options[choice_num]}")
+
+    print_stars()
+    return options
+
+
+def customer_management(): 
+    options = {
+        "*": "List all options",
+        "1": "Add a customer",
+        "2": "Edit a customer",
+        "3": "Delete a customer",
+        "4": "Get information of all customers",
+        "5": "Get information of one customers",
+        "MAIN": "Return to main menu",
+        "END" : "Quit"
+        }
+
+    for choice_num in options:
+        print(f"Option {choice_num}. {options[choice_num]}")
+
+    print_stars()
+    return options
+
+
+def rental_management(): 
+    options = {
+        "*": "List all options",
+        "1": "Check out a video to a customer",
+        "2": "Check in a video from a customer",
+        "MAIN": "Return to main menu",
+        "END" : "Quit"
+        }
+
+    for choice_num in options:
+        print(f"Option {choice_num}. {options[choice_num]}")
+
+    print_stars()
+    return options
+
+
+def make_choice(options, video_store):
+    valid_choices = options.keys()
+    choice = None
+    while choice not in valid_choices:
+        print("What would you like to do?")
+        choice = input("Make your selection using the option number: ")
+    return choice
+
+
+
+#CLI
+def run_cli(play=True):
+
+    #initialize video store
+    video_store = VideoStore(url=BACKUP_URL)
+    #print main menu
+    menu = main_menu()
+    main_menu_choice = make_choice(menu, video_store)
+
+    while play==True: 
+
+        #QUIT
+        if main_menu_choice=="END":
+            play=False
+            print("\nThanks for using the Video Store CLI!")
+
+        #VIDEO MANAGEMENT
+        elif main_menu_choice == '1':
+            print("---> VIDEO MANAGEMENT <---")
+            management = video_management()
+            subchoice = make_choice (management,video_store)
+
+            #Option 1: add a video
+            if subchoice=='1':
+                print("Adding a new video. Please enter the following information.")
+                new_title = input("Title: ")
+                new_release_date = input("Release Date: ")
+                new_total_inventory = input("Total Inventory: ")
+                response = video_store.add_video(title=new_title, release_date=new_release_date, total_inventory=new_total_inventory)
+                print("New video created, id# ", response["id"])
+
+            #Option 2: edit a video 
+            elif subchoice=='2':
+                select_video = input("Enter the id of the video to update: ")
+                if select_video.isnumeric():
+                    title=input("Enter the updated title of your video: ")
+                    release_date=input("Enter the updated release date of your video: ")
+                    total_inventory=input("Enter the updated total inventory of your video: ")
+                    response = video_store.edit_video(select_video,title=title, release_date=release_date, total_inventory=total_inventory)      
+                    print(f"Video #{response['id']} has been updated.") 
+
+            #Option 3: delete a video 
+            elif subchoice=="3":
+                delete_id = input("Enter the video ID to delete: ")
+                if delete_id.isnumeric():
+                    video_store.delete_video(id=delete_id)
+                    print(f"The video id #{delete_id} has been deleted.")
+
+            #Option 4: get information of all videos
+            elif subchoice=="4":
+                for video in video_store.list_all_videos():
+                    print(video)
+
+            #Option 5: get information of one video 
+            elif subchoice=="5":
+                video_id = input("Select the video ID: ")
+                if video_id.isnumeric():
+                    video = video_store.get_video(id=video_id)
+                    print(video)
+
+            #List all options
+            elif subchoice == "*":
+                video_management()    
+
+            #Return to main menu 
+            elif subchoice == "MAIN":
+                main_menu()
+
+            #Quit
+            elif subchoice == "END": 
+                play=False
+                print("\nThanks for using the Video Store CLI!")
+
+
+
+        #CUSTOMER MANAGEMENT
+        elif main_menu_choice == '2':
+            print("---> CUSTOMER MANAGEMENT <---")
+            management = customer_management()
+            subchoice = make_choice (management,video_store)
+
+            #Option 1: add a customer
+            if subchoice=='1':
+                print("Adding a new customer. Please enter the following information.")
+                name = input("Customer Name: ")
+                postal_code = input("Postal Code: ")
+                phone = input("Phone: ")
+                response = video_store.add_customer(name=name, postal_code=postal_code, phone=phone)
+                print("New customer created, id# ", response["id"])
+
+            #Option 2: edit a customer
+            elif subchoice=='2':
+                select_customer = input("Enter the id of the video to update: ")
+                if select_customer.isnumeric():
+                    name=input("Enter the updated name of the customer: ")
+                    postal_code=input("Enter the updated postal code of the customer: ")
+                    phone=input("Enter the updated phone number of the customer: ")
+                    response = video_store.update_customer(select_customer, name=name,postal_code=postal_code, phone=phone)      
+                    print(f"Customer #{response['id']} has been updated.") 
+
+            #Option 3: delete a customer 
+            elif subchoice=="3":
+                delete_id = input("Enter the customer ID to delete: ")
+                if delete_id.isnumeric():
+                    video_store.delete_customer(id=delete_id)
+                    print(f"Customer id #{delete_id} has been deleted.")
+
+            #Option 4: get information of all customers
+            elif subchoice=="4":
+                for customer in video_store.list_all_customers():
+                    print(customer)
+
+            #Option 5: get information of one customer
+            elif subchoice=="5":
+                customer_id = input("Enter the customer ID to view information: ")
+                if customer_id.isnumeric():
+                    customer = video_store.get_customer(id=customer_id)
+                    print(customer)
+
+            #Return to main menu 
+            elif subchoice == "MAIN":
+                main_menu()
+
+            #Quit
+            elif subchoice == "END": 
+                play=False
+                print("\nThanks for using the Video Store CLI!")
+
+
+
+        #RENTAL MANAGEMENT
+        elif main_menu_choice == '3':
+            print("---> RENTAL MANAGEMENT <---")
+            management = rental_management()
+            subchoice = make_choice (management,video_store)
+
+            # Option 1: check out a video to a customer
+            if subchoice == "1":
+                print("Checking out a video.")
+                video_id = input("Enter the video id: ")
+                customer_id = input("Enter the customer id: ")
+
+                if customer_id == None or video_id == None:
+                    print("No record found. Please enter valid information.")
+                else:
+                    rental = video_store.checkout_video(customer_id=customer_id, video_id=video_id)
+                    print(f"Video #{video_id} has been checked out to customer #{customer_id}.")
+
+            # Option 2: check in a video from a customer
+            elif subchoice == "2":
+                print("Checking in a video")
+                video_id = input("Enter the video id: ")
+                customer_id = input("Enter the customer id: ")
+
+                if video_id == None or customer_id == None:
+                    print("No record found. Please enter valid information.")
+                else:
+                    return_rental = video_store.checkin_video(customer_id=customer_id, video_id=video_id)
+                    print(f"Video #{video_id} has been returned from customer #{customer_id}.")
+
+            #Return to main menu 
+            elif subchoice == "MAIN":
+                main_menu()
+
+            #Quit
+            elif subchoice == "END": 
+                play=False
+                print("\nThanks for using the Video Store CLI!")
+
+run_cli()