You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.7 KiB
114 lines
3.7 KiB
# Create a variable called search_or_ratings, set to "1".
|
|
# You should be able to change this between "1", "2", and "3" to change what your program prints.
|
|
search_or_ratings = 2
|
|
|
|
class Media(object):
|
|
"""Base class for Movie and Book objects"""
|
|
|
|
def __init__(self, publisher, market):
|
|
self.publisher = publisher
|
|
self.market = market
|
|
|
|
def get_media_info(self):
|
|
"""Prints media info to stdout"""
|
|
print(f'The Media object\'s publisher is {self.publisher} \
|
|
and the market is {market}.')
|
|
|
|
class Movie(Media):
|
|
"""
|
|
Movie objects contain all information about a particular movie,
|
|
including the title and rating.
|
|
"""
|
|
|
|
def __init__(self, movie_data, publisher, market):
|
|
"""Initialize Movie class, note that the init of the superclass,
|
|
Media, is called with the super().__init__() call"""
|
|
|
|
# Call the constructor (init) method of Media
|
|
super().__init__(publisher, market)
|
|
# Store the raw data in this object so that we can use the
|
|
# data in the getter functions.
|
|
self.movie_data = movie_data
|
|
|
|
def get_movie_title(self):
|
|
"""
|
|
get_movie_title is a getter function that returns the movie title.
|
|
"""
|
|
|
|
# Return the title from the movie data.
|
|
return self.movie_data["title"]
|
|
|
|
def get_movie_rating(self):
|
|
"""
|
|
get_movie_rating is a getter function that returns the rating.
|
|
"""
|
|
|
|
# Return the rating from the movie data.
|
|
return self.movie_data["rating"]
|
|
|
|
def return_single_movie_object(movie_title, movie_rating, publisher, market):
|
|
"""
|
|
Take in the movie title and rating, and return the movie object.
|
|
"""
|
|
|
|
return Movie({'title': movie_title, 'rating': movie_rating}, publisher, market)
|
|
|
|
def list_search_results(movie_titles):
|
|
"""
|
|
Print list of movies. Later, print a list of title results from a movie search.
|
|
"""
|
|
|
|
# Loop through the list of titles and print them (indented with four spaces).
|
|
for title in movie_titles:
|
|
print(" " + title)
|
|
|
|
def print_single_movie_rating(movie_query):
|
|
"""
|
|
Create a Movie object and print one movie's Rotten Tomatoes rating.
|
|
"""
|
|
|
|
my_movie = return_single_movie_object(movie_query, 7, "Universal Studios", "USA")
|
|
|
|
# Print the whole formatted string.
|
|
print("The rating for", my_movie.get_movie_title(), "is", my_movie.get_movie_rating())
|
|
|
|
def print_all_ratings(movie_list):
|
|
"""
|
|
Take in a list of movies, create a movie object for each, and print the rating.
|
|
"""
|
|
|
|
for movie in movie_list:
|
|
movie_object = return_single_movie_object(movie, 4, "Universal Studios", "USA")
|
|
print("The movie", movie_object.get_movie_title(), "has a rating of", movie_object.get_movie_rating())
|
|
|
|
# Create one main function that will call everything else.
|
|
def main():
|
|
|
|
"""
|
|
Main is the entry point into the program, and it calls into the search or
|
|
ratings functions depending on what the user decides to do.
|
|
"""
|
|
|
|
# A hard-coded movie list with which to test.
|
|
default_movie_list = ["Back to the Future", "Blade", "Spirited Away"]
|
|
|
|
|
|
# Let's test: Call a print_all_ratings() function and pass it the default_movie_list as a parameter.
|
|
print_all_ratings(default_movie_list)
|
|
|
|
if search_or_ratings == 1:
|
|
# If search_or_ratings is 1, call list_search_results().
|
|
list_search_results(default_movie_list)
|
|
elif search_or_ratings == 2:
|
|
# If search_or_ratings is 2, call print_movie_rating().
|
|
print_single_movie_rating("Moana")
|
|
|
|
else:
|
|
# If search_or_ratings is otherwise, give an error.
|
|
print("Error: Your input must be 1 or 2!")
|
|
|
|
|
|
# This line tells Python to run main() when it first opens.
|
|
if __name__ == "__main__":
|
|
main()
|