Fix MPA Rating

Author: colinswinney

mu-plugins/10up-plugin/requirements.md

@@ -39,6 +39,7 @@ Create a WP CLI script that can take a list of IMDB IDs and import movie and per
 ## API Endpoints
 - **Movies**: `https://api.imdbapi.dev/titles/{imdb_id}`
 - **People**: `https://api.imdbapi.dev/names/{imdb_id}`
+- **Movie Certificates**: `https://api.imdbapi.dev/titles/{imdb_id}/certificates`
 
 ## Movie Import Requirements
 
@@ -59,6 +60,7 @@ Based on existing post meta fields in the 10up-plugin, map the following API res
 | `plot` | `tenup_movie_plot` | Movie plot | string |
 | `rating.aggregateRating` | `tenup_movie_viewer_rating` | Viewer rating | string |
 | `rating.voteCount` | `tenup_movie_viewer_rating_count` | Number of votes | string |
+| US Certificate Rating | `tenup_movie_mpa_rating` | MPA rating from US certificate | string |
 
 ### Runtime Conversion
 Convert `runtimeSeconds` to the expected object format:
@@ -69,6 +71,41 @@ Convert `runtimeSeconds` to the expected object format:
 ]
 ```
 
+### MPA Rating Retrieval
+For each movie import, make an additional API call to retrieve MPA rating from certificates:
+
+**API Endpoint**: `https://api.imdbapi.dev/titles/{imdb_id}/certificates`
+
+**Process**:
+1. Make API call to certificates endpoint using the movie's IMDB ID
+2. Parse the `certificates` array in the response
+3. Find the certificate where:
+   - `country.code` equals `"US"`
+   - `attributes` array contains a string that includes `"certificate #"` or `"certificate#"`
+4. Extract the `rating` value from the matching certificate
+5. Store the rating in the `tenup_movie_mpa_rating` meta field
+
+**Example**:
+Based on the API response from [https://api.imdbapi.dev/titles/tt2380307/certificates](https://api.imdbapi.dev/titles/tt2380307/certificates), the script should find:
+```json
+{
+  "rating": "PG",
+  "country": {
+    "code": "US",
+    "name": "United States"
+  },
+  "attributes": [
+    "certificate #51192"
+  ]
+}
+```
+And store `"PG"` as the MPA rating.
+
+**Error Handling**:
+- If no US certificate with "certificate #" or "certificate#" attribute is found, set `tenup_movie_mpa_rating` to "Not Rated"
+- If the certificates API call fails, log the error but continue with the movie import
+- Handle cases where the certificates array is empty or malformed
+
 ### Genre Taxonomy
 - **Taxonomy**: `tenup-genre`
 - **Source**: `genres` array from API response
@@ -228,8 +265,9 @@ wp imdb-import both --file=imdb_ids.txt
 When importing a movie, the script should:
 
 1. **Fetch Movie Data**: Get movie data from IMDB API
-2. **Create Movie Post**: Create the movie post with all meta fields
-3. **Process Stars Array**: For each star in the `stars` array:
+2. **Fetch MPA Rating**: Get MPA rating from certificates API
+3. **Create Movie Post**: Create the movie post with all meta fields including MPA rating
+4. **Process Stars Array**: For each star in the `stars` array:
    - Check if person already exists (by `tenup_person_imdb_id`)
    - If exists:
      - Update person post with latest data from People API
@@ -251,15 +289,16 @@ When importing a movie, the script should:
 - Handle duplicate filenames by appending numbers
 
 ### API Rate Limiting
-- Respect rate limits between movie and people API calls
+- Respect rate limits between movie, certificates, and people API calls
 - Cache people API responses to avoid duplicate calls
+- Cache certificates API responses to avoid duplicate calls
 - Batch process stars to minimize API requests
+- Implement delays between API calls to respect rate limits
 
 ## Additional Notes
 
 ### Fields Not Available in API
 The following existing post meta fields do not have equivalents in the IMDB API responses and will not be populated:
-- `tenup_movie_mpa_rating` (MPA rating not available)
 - `tenup_movie_summary` (summary not available)
 - `tenup_movie_synopsis` (synopsis not available)
 - `tenup_movie_tagline` (tagline not available)

mu-plugins/10up-plugin/src/CLI/Importers/MovieImporter.php

@@ -189,6 +189,9 @@ private function create_movie( $data, $options ) {
 		// Set meta fields.
 		$this->set_movie_meta( $post_id, $data );
 
+		// Fetch and set MPA rating from certificates API.
+		$this->set_mpa_rating( $post_id, $data['id'] );
+
 		// Set featured image.
 		$this->set_featured_image( $post_id, $data );
 
@@ -232,6 +235,9 @@ private function update_movie( $post_id, $data, $options ) {
 		// Update meta fields.
 		$this->set_movie_meta( $post_id, $data );
 
+		// Fetch and update MPA rating from certificates API.
+		$this->set_mpa_rating( $post_id, $data['id'] );
+
 		// Update featured image.
 		$this->set_featured_image( $post_id, $data );
 
@@ -289,6 +295,33 @@ private function set_movie_meta( $post_id, $data ) {
 		}
 	}
 
+	/**
+	 * Set MPA rating from certificates API.
+	 *
+	 * @param int    $post_id Post ID.
+	 * @param string $imdb_id IMDB movie ID.
+	 */
+	private function set_mpa_rating( $post_id, $imdb_id ) {
+		// Fetch certificates data.
+		$certificates_data = $this->api_client->get_movie_certificates( $imdb_id );
+
+		if ( is_wp_error( $certificates_data ) ) {
+			WP_CLI::warning( sprintf( 'Failed to fetch certificates for %s: %s', $imdb_id, $certificates_data->get_error_message() ) );
+			return;
+		}
+
+		// Extract US MPA rating.
+		$mpa_rating = $this->api_client->extract_us_mpa_rating( $certificates_data );
+
+		if ( $mpa_rating ) {
+			update_post_meta( $post_id, 'tenup_movie_mpa_rating', $mpa_rating );
+			WP_CLI::log( sprintf( 'Set MPA rating for %s: %s', $imdb_id, $mpa_rating ) );
+		} else {
+			update_post_meta( $post_id, 'tenup_movie_mpa_rating', 'Not Rated' );
+			WP_CLI::log( sprintf( 'No US MPA rating found for %s, set to: Not Rated', $imdb_id ) );
+		}
+	}
+
 	/**
 	 * Set featured image from API data.
 	 *

mu-plugins/10up-plugin/src/CLI/Utils/IMDBApiClient.php

@@ -28,6 +28,13 @@ class IMDBApiClient {
 	 */
 	private const PERSON_API_URL = 'https://api.imdbapi.dev/names/';
 
+	/**
+	 * Base API URL for movie certificates.
+	 *
+	 * @var string
+	 */
+	private const CERTIFICATES_API_URL = 'https://api.imdbapi.dev/titles/';
+
 	/**
 	 * Rate limiting delay between requests (in seconds).
 	 *
@@ -115,6 +122,66 @@ public function get_person_data( $imdb_id ) {
 		return $data;
 	}
 
+	/**
+	 * Get movie certificates data from IMDB API.
+	 *
+	 * @param string $imdb_id IMDB movie ID.
+	 * @return array|WP_Error Certificates data or error.
+	 */
+	public function get_movie_certificates( $imdb_id ) {
+		$cache_key   = 'imdb_certificates_' . $imdb_id;
+		$cached_data = get_transient( $cache_key );
+
+		if ( false !== $cached_data ) {
+			return $cached_data;
+		}
+
+		$url      = self::CERTIFICATES_API_URL . $imdb_id . '/certificates';
+		$response = $this->make_request( $url );
+
+		if ( is_wp_error( $response ) ) {
+			return $response;
+		}
+
+		$data = json_decode( $response, true );
+
+		if ( ! $data || isset( $data['error'] ) ) {
+			return new WP_Error( 'api_error', 'Failed to fetch certificates data: ' . ( $data['error'] ?? 'Unknown error' ) );
+		}
+
+		// Cache the successful response.
+		set_transient( $cache_key, $data, self::CACHE_DURATION );
+
+		return $data;
+	}
+
+	/**
+	 * Extract US MPA rating from certificates data.
+	 *
+	 * @param array $certificates_data Certificates API response data.
+	 * @return string|null MPA rating or null if not found.
+	 */
+	public function extract_us_mpa_rating( $certificates_data ) {
+		if ( ! isset( $certificates_data['certificates'] ) || ! is_array( $certificates_data['certificates'] ) ) {
+			return null;
+		}
+
+		foreach ( $certificates_data['certificates'] as $certificate ) {
+			// Check if this is a US certificate with "certificate #" or "certificate#" attribute.
+			if ( isset( $certificate['country']['code'] ) && 'US' === $certificate['country']['code'] ) {
+				if ( isset( $certificate['attributes'] ) && is_array( $certificate['attributes'] ) ) {
+					foreach ( $certificate['attributes'] as $attribute ) {
+						if ( is_string( $attribute ) && ( strpos( $attribute, 'certificate #' ) !== false || strpos( $attribute, 'certificate#' ) !== false ) ) {
+							return $certificate['rating'] ?? null;
+						}
+					}
+				}
+			}
+		}
+
+		return null;
+	}
+
 	/**
 	 * Make HTTP request with rate limiting and retry logic.
 	 *

mu-plugins/10up-plugin/src/PostMeta/MovieMPARating.php

@@ -44,20 +44,6 @@ public function get_description(): string {
 	 */
 	protected $has_key_value_options = true;
 
-	/**
-	 * Get allowed values for field schema.
-	 */
-	public function allowed_values(): array {
-		return [
-			'Unrated' => 'Unrated',
-			'G'       => 'G',
-			'PG'      => 'PG',
-			'PG-13'   => 'PG-13',
-			'R'       => 'R',
-			'NC-17'   => 'NC-17',
-		];
-	}
-
 	/**
 	 * Get the post types.
 	 *
@@ -89,7 +75,15 @@ public function add_localized_script() {
 			'tenup_plugin_admin',
 			'TenupMovieMPARating',
 			array(
-				'options' => $this->allowed_values(),
+				'options' => [
+					'Not Rated' => 'Not Rated',
+					'Approved'  => 'Approved',
+					'G'         => 'G',
+					'PG'        => 'PG',
+					'PG-13'     => 'PG-13',
+					'R'         => 'R',
+					'NC-17'     => 'NC-17',
+				],
 			)
 		);
 	}