support exporting attachments when specific post ids are requested (assuming the post whose ids were specified were not attachment themselves)

Raphaël Droz · Raphaël Droz · commit 68ee7f766ea8 · 2017-09-01T16:39:46.000-03:00
diff --git a/src/WP_Export_Query.php b/src/WP_Export_Query.php
@@ -140,6 +140,10 @@ public function posts() {
 	private function calculate_post_ids() {
 		global $wpdb;
 		if ( is_array( $this->filters['post_ids'] ) ) {
+			if ( getenv( 'WP_WITH_ATTACHMENT' ) ) {
+				$attachment_post_ids = $this->attachments_for_specific_post_types( $this->filters['post_ids'] );
+				$this->filters['post_ids'] = array_merge( $this->filters['post_ids'], $attachment_post_ids );
+			}
 			return $this->filters['post_ids'];
 		}
 		$this->post_type_where();
@@ -262,7 +266,7 @@ private function category_where() {
 
 	private function attachments_for_specific_post_types( $post_ids ) {
 		global $wpdb;
-		if ( !$this->filters['post_type'] ) {
+		if ( ! $post_ids ) {
 			return array();
 		}
 		$attachment_ids = array();
