import re import pprint from media import Movie, TvShow from fresh_tomatoes import open_movies_page ROOT_DIR = os.path.dirname(__file__) RESOURCES_DIR = os.path.join(ROOT_DIR, 'resources') if not os.path.exists(os.path.join(RESOURCES_DIR, 'adn_trailers.db')): db_handler = open(os.path.join(RESOURCES_DIR, 'adn_trailers.db'), 'w') movies = [] for i in range(5): movie_obj = Movie() movie_obj.title = 'Kung Fu Panda' movie_obj.trailer_youtube_url = 'https://www.youtube.com/watch?v=PXi3Mv6KMzY' movie_obj.poster_image_url = 'http://ia.media-imdb.com/images/M/MV5BMTIxOTY1NjUyN15BMl5BanBnXkFtZTcwMjMxMDk1MQ@@._V1_SX214_AL_.jpg' movie_obj.storyline = 'The Dragon Warrior has to clash against the savage Tai Lung as China\'s fate hangs in the balance: However, the Dragon Warrior mantle is supposedly mistaken to be bestowed upon an obese panda who is a tyro in martial arts.' movie_obj.release_date = '2008' movie_obj.actors = [] movies.append(movie_obj) pickle.dump(movies, db_handler) db_handler = open(os.path.join(RESOURCES_DIR, 'adn_trailers.db'), 'rb') movies = pickle.load(db_handler) open_movies_page(movies)