def rename(filename): filename_without_extension = filename extension = "" if os.path.isdir(filename) == False: filename_without_extension, extension = os.path.splitext(filename) query = ImdbQuery.create_query_from_filename(filename_without_extension) movie = Movie.create_movie_from_imdb_query(query) if movie.title == "None": # sys.stdout.write("Query Was: " + query + "\n") return "None" return unicode(movie).encode("utf-8") + extension
def test_create_movie_from_query_dots_and_hyphens(self): LOTR_2 = Movie.create_movie_from_imdb_query('lord.of.rings-two.towers') self.assertEqual(LOTR_2, Movie("The Lord of the Rings: The Two Towers", 2002))
def test_create_movie_from_query_no_spaces(self): the_dark_knight = Movie.create_movie_from_imdb_query('TheDarkKnight') self.assertEqual(the_dark_knight, Movie("The Dark Knight", 2008))
def test_create_movie_from_query_partial_title(self): the_matrix = Movie.create_movie_from_imdb_query('matrix') self.assertEqual(the_matrix, Movie("The Matrix", 1999))
def test_create_movie_from_query_exact_title(self): the_prestige = Movie.create_movie_from_imdb_query('the prestige') self.assertEqual(the_prestige, Movie("The Prestige", 2006))
def rename(filename): filename_without_extension, extension = os.path.splitext(filename) query = ImdbQuery.create_query_from_filename(filename_without_extension) movie = Movie.create_movie_from_imdb_query(query) return str(movie) + extension