Ticket #48193: trac-48193-wp-loop.diff

src/wp-includes/query.php

@@ -1161 +1161 @@
 
         return false;
 }
+
+/**
+ * Simplifies the WordPress loop.
+ *
+ * @since 5.x
+ *
+ * @global WP_Query $wp_query WordPress Query object.
+ *
+ * @param WP_Query|WP_Post[] $iterable WordPress query object or an array of posts.
+ *
+ * @return Generator
+ */
+function wp_loop( $iterable = null ) {
+
+        if ( null === $iterable ) {
+                $iterable = $GLOBALS['wp_query'];
+        }
+
+        $posts = $iterable;
+
+        if ( is_object( $iterable ) && property_exists( $iterable, 'posts' ) ) {
+                $posts = $iterable->posts;
+        }
+
+        if ( ! is_array( $posts ) ) {
+                /* translators: Data type */
+                _doing_it_wrong( __FUNCTION__, sprintf( __( 'Expected an array, received %s instead.' ), gettype( $posts ) ), '3.1.0' );
+
+                return false;
+        }
+
+        global $post;
+
+        // Save the global post object so we can restore it later
+        $save_post = $post;
+
+        try {
+                foreach ( $posts as $post ) {
+
+                        setup_postdata( $post );
+                        yield $post;
+
+                }
+        } finally {
+
+                wp_reset_postdata();
+                // Restore the global post object
+                $post = $save_post;
+
+        }
+}