Пример #1
0
    def test_UserDict():
        """ Unit test for UserToDictionary function """

        newUser = User("beaf;o", "test", ["koidw"])

        dict = UserManager.UserToDictionary(newUser)

        assert dict['lastName'] == "test"
Пример #2
0
    def main():
        #MAIN FUNCTION OF THE PROGRAM
        
        movieList = [] #Main movie list of the program
        userList = [] #Main user list of the programs

        shouldExit = False #Set to true to exit loop

        initializationText = """ Hello! What would you like to do?
        User menu:
            1. Add a user
            2. Change a user's name
            3. Remove a user
        Movie menu:
            4. Add a movie
            5. Update a movie's price
            6. List all movies
        Miscellaneous menu:
            7. Order a movie
            8. List user's movie orders
            9. Search movies by rating
            10. Filter movies by actors

            0. Exit program
        """
        
        #Run Unit tests
        TestClass.RunTests()

        #The main lists of the program
        movieList = []
        userList = []

        #Populate the user list from the database
        try:
            userInfile = open("userDatabase", "rb")
        except:

            print("No user database found!")

        else:
            while True:
            
                try:
                    userList.append(UserManager.DictionaryToUser(pickle.load(userInfile)))
                except:
                    break
        
        #Populate the movie list from the database
        try:
            movieInfile = open("movieDatabase", "rb")
        except:

            print("No movie database found!")

        else:
            while True:
            
                try:
                    movieList.append(MovieManager.DictionaryToMovie(pickle.load(movieInfile)))
                except:
                    break
                

        while shouldExit == False:
            #MAIN LOOP OF THE PROGRAM
            
            #Print the user menu
            print(initializationText)
            
            #Get user input
            try:
                userChoice = int( input("Your choice: ") )
            except ValueError:
                print("Choice is not an integer!")
                continue

            if userChoice == 0:
                #Exits program

                #Save the users to the database
                userOutfile = open("userDatabase", "wb")

                for user in userList:

                    dict = UserManager.UserToDictionary(user)
                    pickle.dump(dict, userOutfile)

                userOutfile.close()

                #Save the movies to the database
                movieOutfile = open("movieDatabase", "wb")

                for movie in movieList:

                    dict = MovieManager.MovieToDictionary(movie)
                    pickle.dump(dict, movieOutfile)

                movieOutfile.close()

                print("Goodbye!")
                shouldExit = True
        
            elif userChoice == 1:
                #Add user option
                newUser = UserManager.getUserInput()
                userList = UserManager.addUser(userList, newUser)         
            
            elif userChoice == 2:
                #Change a user's name          
                userList = UserManager.updateLastname(userList)

            elif userChoice == 3:
                #Remove a user           
                #Check the list isn't empty
                if len(userList) == 0:
        
                    print("The user list is empty!")
                    continue
                
                UserManager.printUsers(userList)

                try:
                    removeUserChoice = int(input("Which user would you like to remove? Index: "))
                except ValueError:
                    print("Choice is not an integer!")
                    continue

                #Clamp the value of the index between the min and the max indexes
                if removeUserChoice < 0:
        
                    removeUserChoice = 0
    
                elif removeUserChoice > len(userList) - 1:

                    removeUserChoice = len(userList) - 1
            
                #Remove the selected user
                del userList[removeUserChoice]

            elif userChoice == 4:
                #Add a movie

                newMovie = MovieManager.getMovieInput()
                movieList = MovieManager.addMovie(movieList, newMovie)
            
            elif userChoice == 5:
                #Update a movie's price

                movieList = MovieManager.updatePrice(movieList)

            elif userChoice == 6:
                #List all movies
            
                MovieManager.printMovies(movieList)

            
            elif userChoice == 7:
                #Order a movie
            
                userList = MiscManager.orderMovies(userList, movieList)

            elif userChoice == 8:
                #List all user's movie orders
                        
                UserManager.printUsers(userList)

            elif userChoice == 9:
                #Search all movies by rating
            
                MiscManager.searchMoviesByRating(movieList)

            elif userChoice == 10:
                #Filter movies by actors
            
                MiscManager.searchMoviesByActor(movieList)