Python Assignment Solution on Movie Recommender

• 23rd Dec, 2021
• 01:01 AM

#Netflix type system demo - FakeFlix
import csv
import sys
import pandas as pd
def main():
   menu()
def menu():
    print("************Welcome to Movie Recomender System**************")
    print()

    choice = input("""
                      A: Add a user
                      B: List all user
                      C: Suggest a movie
                      D: List all movie
                      Q: Logout

                      Please enter your choice: """)

    if choice == "A" or choice =="a":
        register()
    elif choice == "B" or choice =="b":
        displayUser()
    elif choice == "C" or choice =="c":
        suggestMovie()
    elif choice == "D" or choice =="d":
        displayMovie()
    elif choice=="Q" or choice=="q":
        sys.exit()
    else:
        print("You must only select either A or B")
        print("Please try again")
        menu()

def register():
  
    #user is prompted to input all the required fields
    print("Enter first name")
    firstname=input()
    print("Enter lastname")
    lastname=input()
    print("Enter rating")
    rating=input()
    
    print("Your unique username is", firstname+lastname)
    global username
    username=firstname+lastname
       
    
    with open('users.txt','a') as users:
        usersWriter=csv.writer(users)
        usersWriter.writerow([username,firstname,lastname,rating])
        print("User has been added succesfully")
        users.close()
        menu()
        
    
def displayUser():
    df = pd.read_csv("users.txt",delimiter=',')
    df.to_csv('users.csv')
    print(df.head())
    menu()
    
def  suggestMovie():
    df = pd.read_csv("movie.csv",delimiter=',')
    #df.to_csv('movie.csv')
    print(df.head())
    menu()
    
global username
def displayMovie():
    #open the file
       with open("movie.csv","r") as f:
      #prompt the user to enter the desired title that they are searching for
          rating=input("Enter the RATING you are after *and we'll show you films that match*:")
      #call up on the csv reader (this will allow us to work with the file and do clever stuff!)
          fReader=csv.reader(f)
      #for reach row that is read by the reader
          for row in fReader:
         #and for each field in that row (this feature is automated by the reader)
             for field in row:
            #if the field is equal to the rating that you are looking for
                if rating in field: #this looks for title or any part of title in the field (not necessarily a perfect solution)
                       print("Searching file ....please wait")
                       print("Found:", row)
                       print("This is film no:", row[0])
    
                choice=input("""Would you like to view any of these films?
                             Y: Yes
                             N: No, thanks

                             Please enter your choice:""")
                if choice=="Y" or choice=="y":
                    print("We are taking you to the WATCH FILMS menu - ratings are displayed next to films")
                    suggestMovie()
                elif choice=="N" or choice=="n":
                                     sys.exit()
    #print(df.head())
#the program is initiated, so to speak, here
main()