diff --git a/src/main/java/Main.java b/src/main/java/Main.java index 473dd4d..634b97c 100644 --- a/src/main/java/Main.java +++ b/src/main/java/Main.java @@ -1,4 +1,5 @@ import java.io.IOException; +import java.util.List; public class Main { public static void main(String[] args) { @@ -6,15 +7,41 @@ public class Main { String filePath = "src/main/resources/Movies.html"; try { - // Instantiate the parser Parser parser = new Parser(filePath); - - // TODO: You can test your code here before you run the unit tests. - // Example: Call your sorting methods, iterate through the returned list, - // and print out the movies to see if they match the expected results in the Help folder. - + + System.out.println("=================================================="); + System.out.println(" IMDb MOVIE SCRAPER "); + System.out.println("==================================================\n"); + + // sortByTitle + System.out.println("[ MOVIES SORTED BY TITLE ]"); + System.out.println("--------------------------------------------------"); + List sortedByTitle = parser.sortByTitle(); + for (Movie movie : sortedByTitle) { + System.out.println(" • " + movie.toString()); + } + + // sortByRating + System.out.println("\n[ MOVIES SORTED BY RATING (DESCENDING) ]"); + System.out.println("--------------------------------------------------"); + List sortedByRating = parser.sortByRating(); + for (Movie movie : sortedByRating) { + System.out.println(" • " + movie.toString()); + } + + //sortByYear + System.out.println("\n[ MOVIES SORTED BY YEAR (DESCENDING) ]"); + System.out.println("--------------------------------------------------"); + List sortedByYear = parser.sortByYear(); + for (Movie movie : sortedByYear) { + System.out.println(" • " + movie.toString()); + } + + System.out.println("\n=================================================="); + System.out.println(" >> Movies parsed successfully! Total movies: " + sortedByTitle.size()); + System.out.println("=================================================="); System.out.println("Movies parsed successfully! Total movies: " + parser.sortByTitle().size()); - + } catch (IOException e) { System.err.println("Error reading or parsing the HTML file: " + e.getMessage()); } diff --git a/src/main/java/Movie.java b/src/main/java/Movie.java index fc257fe..c2e99cb 100644 --- a/src/main/java/Movie.java +++ b/src/main/java/Movie.java @@ -6,8 +6,9 @@ public class Movie { private int year; public Movie(String title, double rating, int year) { - // TODO: Initialize the instance variables (this.title, this.rating, this.year) - // with the values passed as arguments to this constructor. + this.title = title; + this.rating = rating; + this.year = year; } public String getTitle() { @@ -24,9 +25,7 @@ public class Movie { @Override public String toString() { - // TODO: Return a string representation of the movie matching the exact required format. - // Example format: "Movie: The Godfather, Rating: 9.2, Year: 1972" - return ""; + return "Movie: " + title + ", Rating: " + rating + ", Year: " + year; } @Override diff --git a/src/main/java/Parser.java b/src/main/java/Parser.java index 516cf14..5c0d2b4 100644 --- a/src/main/java/Parser.java +++ b/src/main/java/Parser.java @@ -15,38 +15,33 @@ public class Parser { public List sortByTitle() { List sortedByTitle = new ArrayList<>(movies); - // TODO: Sort the 'sortedByTitle' list alphabetically by the movie's title. - // Hint: You can use Collections.sort() or the List.sort() method along with a custom Comparator. - // Example: Comparator.comparing(Movie::getTitle) + sortedByTitle.sort(Comparator.comparing(Movie::getTitle)); return sortedByTitle; } public List sortByRating() { List sortedByRating = new ArrayList<>(movies); - // TODO: Sort the 'sortedByRating' list by the movie's rating in descending order (highest to lowest). - // Hint: Use Double.compare() in your Comparator or Comparator.comparingDouble().reversed(). + sortedByRating.sort(Comparator.comparingDouble(Movie::getRating).reversed()); return sortedByRating; } public List sortByYear() { List sortedByYear = new ArrayList<>(movies); - // TODO: Sort the 'sortedByYear' list by the movie's release year in descending order (newest to oldest). - // Hint: Compare the year integers. Use Integer.compare() or Comparator.comparingInt().reversed(). + sortedByYear.sort(Comparator.comparingInt(Movie::getYear).reversed()); return sortedByYear; } private void setUp(String filePath) throws IOException { - // TODO: Create a java.io.File object using the given 'filePath'. - // TODO: Parse the HTML file using Jsoup.parse(file, "UTF-8"). This returns a Document object. - - // TODO: Extract the list of movie elements from the Document. - // Hint: Use document.select() or document.getElementsByClass() to find all elements with the class "movie". - - // TODO: Iterate through each movie Element. - // For each movie element: - // 1. Find the title element (e.g., using selector "h3.movie-title") and get its text. - // 2. Find the rating element (e.g., using selector "span.movie-rating"). Extract the text, remove the "/10" part, and parse it to a double. - // 3. Find the year element (e.g., using selector "span.movie-year"). Extract the text and parse it to an int. - // 4. Create a new Movie object with these values and add it to the 'movies' list. + File file = new File(filePath); + Document document = Jsoup.parse(file, "UTF-8"); + Elements movieElements = document.getElementsByClass("movie"); + for (Element element : movieElements) { + String title = element.select("h3.movie-title").text(); + String ratingStr = element.select("span.movie-rating").text().replace("/10", "").trim(); + double rating = Double.parseDouble(ratingStr); + String yearStr = element.select("span.movie-year").text().trim(); + int year = Integer.parseInt(yearStr); + movies.add(new Movie(title, rating, year)); + } } }