Changeset 27423 for trunk/src/wp-includes/post.php

Timestamp:

03/05/2014 10:25:31 PM (11 years ago)

Author:

johnbillion

Message:

Allow get_page_by_path() and get_page_by_title() to accept an array of post types. Fixes #24763. Props zbtirrell.

File:

• trunk/src/wp-includes/post.php (modified) (3 diffs)

trunk/src/wp-includes/post.php

@@ -3498 +3498 @@
  * @param string $page_path Page path
  * @param string $output Optional. Output type. OBJECT, ARRAY_N, or ARRAY_A. Default OBJECT.
- * @param string $post_type Optional. Post type. Default page.
+ * @param string|array $post_type Optional. Post type or array of post types. Default page.
  * @return WP_Post|null WP_Post on success or null on failure
  */
-function get_page_by_path($page_path, $output = OBJECT, $post_type = 'page') {
+function get_page_by_path( $page_path, $output = OBJECT, $post_type = 'page' ) {
     global $wpdb;
 
@@ -3511 +3511 @@
     $parts = array_map( 'sanitize_title_for_query', $parts );
 
-    $in_string = "'". implode( "','", $parts ) . "'";
-    $post_type_sql = esc_sql( $post_type );
-    $pages = $wpdb->get_results( "SELECT ID, post_name, post_parent, post_type FROM $wpdb->posts WHERE post_name IN ($in_string) AND (post_type = '$post_type_sql' OR post_type = 'attachment')", OBJECT_K );
+    $in_string = "'" . implode( "','", $parts ) . "'";
+
+    if ( is_array( $post_type ) ) {
+        $post_types = $post_type;
+    } else {
+        $post_types = array( $post_type, 'attachment' );
+    }
+
+    $post_types = esc_sql( $post_types );
+    $post_type_in_string = "'" . implode( "','", $post_types ) . "'";
+    $sql = "
+        SELECT ID, post_name, post_parent, post_type
+        FROM $wpdb->posts
+        WHERE post_name IN ($in_string)
+        AND post_type IN ($post_type_in_string)
+    ";
+
+    $pages = $wpdb->get_results( $sql, OBJECT_K );
 
     $revparts = array_reverse( $parts );
@@ -3552 +3567 @@
  * @param string $page_title Page title
  * @param string $output Optional. Output type. OBJECT, ARRAY_N, or ARRAY_A. Default OBJECT.
- * @param string $post_type Optional. Post type. Default page.
+ * @param string|array $post_type Optional. Post type or array of post types. Default page.
  * @return WP_Post|null WP_Post on success or null on failure
  */
-function get_page_by_title($page_title, $output = OBJECT, $post_type = 'page' ) {
+function get_page_by_title( $page_title, $output = OBJECT, $post_type = 'page' ) {
     global $wpdb;
-    $page = $wpdb->get_var( $wpdb->prepare( "SELECT ID FROM $wpdb->posts WHERE post_title = %s AND post_type= %s", $page_title, $post_type ) );
+
+    if ( is_array( $post_type ) ) {
+        $post_type = esc_sql( $post_type );
+        $post_type_in_string = "'" . implode( "','", $post_type ) . "'";
+        $sql = $wpdb->prepare( "
+            SELECT ID
+            FROM $wpdb->posts
+            WHERE post_title = %s
+            AND post_type IN ($post_type_in_string)
+        ", $page_title );
+    } else {
+        $sql = $wpdb->prepare( "
+            SELECT ID
+            FROM $wpdb->posts
+            WHERE post_title = %s
+            AND post_type = %s
+        ", $page_title, $post_type );
+    }
+
+    $page = $wpdb->get_var( $sql );
+
     if ( $page )
         return get_post( $page, $output );