Changeset 21995

Timestamp:

09/25/2012 05:01:59 AM (14 years ago)

Author:

nacin

Message:

Add get() and set() methods to WP_User_Query for modifying query variables the way you might with WP_Query. props wonderboymusic. fixes #21426.

File:

• trunk/wp-includes/user.php (modified) (6 diffs)

trunk/wp-includes/user.php

@@ -313 +313 @@
  */
 class WP_User_Query {
+
+        /**
+         * Query vars, after parsing
+         *
+         * @since 3.5.0
+         * @access public
+         * @var array
+         */
+        var $query_vars = array();
 
         /**
@@ -382 +391 @@
                 global $wpdb;
 
-                $qv = &$this->query_vars;
+                $qv =& $this->query_vars;
 
                 if ( is_array( $qv['fields'] ) ) {
@@ -397 +406 @@
                 }
 
-                if ( $this->query_vars['count_total'] )
+                if ( $qv['count_total'] )
                         $this->query_fields = 'SQL_CALC_FOUND_ROWS ' . $this->query_fields;
 
@@ -529 +538 @@
                 global $wpdb;
 
-                if ( is_array( $this->query_vars['fields'] ) || 'all' == $this->query_vars['fields'] ) {
+                $qv =& $this->query_vars;
+
+                if ( is_array( $qv['fields'] ) || 'all' == $qv['fields'] ) {
                         $this->results = $wpdb->get_results("SELECT $this->query_fields $this->query_from $this->query_where $this->query_orderby $this->query_limit");
                 } else {
@@ -535 +546 @@
                 }
 
-                if ( $this->query_vars['count_total'] )
+                if ( $qv['count_total'] )
                         $this->total_users = $wpdb->get_var( apply_filters( 'found_users_query', 'SELECT FOUND_ROWS()' ) );
 
@@ -541 +552 @@
                         return;
 
-                if ( 'all_with_meta' == $this->query_vars['fields'] ) {
+                if ( 'all_with_meta' == $qv['fields'] ) {
                         cache_users( $this->results );
 
                         $r = array();
                         foreach ( $this->results as $userid )
-                                $r[ $userid ] = new WP_User( $userid, '', $this->query_vars['blog_id'] );
+                                $r[ $userid ] = new WP_User( $userid, '', $qv['blog_id'] );
 
                         $this->results = $r;
                 }
+        }
+
+        /**
+         * Retrieve query variable.
+         *
+         * @since 3.5.0
+         * @access public
+         *
+         * @param string $query_var Query variable key.
+         * @return mixed
+         */
+        function get( $query_var ) {
+                if ( isset( $this->query_vars[$query_var] ) )
+                        return $this->query_vars[$query_var];
+
+                return null;
+        }
+
+        /**
+         * Set query variable.
+         *
+         * @since 3.5.0
+         * @access public
+         *
+         * @param string $query_var Query variable key.
+         * @param mixed $value Query variable value.
+         */
+        function set( $query_var, $value ) {
+                $this->query_vars[$query_var] = $value;
         }