Changeset 21995

Timestamp:

09/25/2012 05:01:59 AM (12 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;
     }