Users: Allow any DB field to be returned by `WP_User_Query`.

Restore behaviour of `fields` parameter in `WP_User_Query` to allow developers to specify any database field to be returned either individually or as part of a subset. Add these fields to the documentation.

When a subset of `fields` includes the `id` paramater, include it in the results in both upper and lowercase to maintain backward compatibility.

Follow up to [53327].

Props dd32, pbearne, kraftbj, peterwilsoncc.
Fixes #53177.


Built from https://develop.svn.wordpress.org/trunk@53362


git-svn-id: http://core.svn.wordpress.org/trunk@52951 1a063a9b-81f0-0310-95a4-ce76da25c4cd
This commit is contained in:
Peter Wilson 2022-05-07 03:32:11 +00:00
parent b27a924101
commit f6ef37b050
2 changed files with 27 additions and 8 deletions

View File

@ -230,6 +230,11 @@ class WP_User_Query {
* - 'user_email' * - 'user_email'
* - 'user_url' * - 'user_url'
* - 'user_registered' * - 'user_registered'
* - 'user_pass'
* - 'user_activation_key'
* - 'user_status'
* - 'spam' (only available on multisite installs)
* - 'deleted' (only available on multisite installs)
* - 'all' for all fields * - 'all' for all fields
* - 'all_with_meta' to include meta fields. * - 'all_with_meta' to include meta fields.
* Default 'all'. * Default 'all'.
@ -275,25 +280,33 @@ class WP_User_Query {
$qv = $this->fill_query_vars( $qv ); $qv = $this->fill_query_vars( $qv );
$allowed_fields = array( $allowed_fields = array(
'ID', 'id',
'display_name',
'user_login', 'user_login',
'user_pass',
'user_nicename', 'user_nicename',
'user_email', 'user_email',
'user_url', 'user_url',
'user_registered', 'user_registered',
'user_activation_key',
'user_status',
'display_name',
); );
if ( is_multisite() ) {
$allowed_fields[] = 'spam';
$allowed_fields[] = 'deleted';
}
if ( is_array( $qv['fields'] ) ) { if ( is_array( $qv['fields'] ) ) {
$qv['fields'] = array_map( 'strtolower', $qv['fields'] );
$qv['fields'] = array_intersect( array_unique( $qv['fields'] ), $allowed_fields ); $qv['fields'] = array_intersect( array_unique( $qv['fields'] ), $allowed_fields );
if ( empty( $qv['fields'] ) ) { if ( empty( $qv['fields'] ) ) {
$qv['fields'] = array( 'ID' ); $qv['fields'] = array( 'id' );
} }
$this->query_fields = array(); $this->query_fields = array();
foreach ( $qv['fields'] as $field ) { foreach ( $qv['fields'] as $field ) {
$field = 'ID' === $field ? 'ID' : sanitize_key( $field ); $field = 'id' === $field ? 'ID' : sanitize_key( $field );
$this->query_fields[] = "$wpdb->users.$field"; $this->query_fields[] = "$wpdb->users.$field";
} }
$this->query_fields = implode( ',', $this->query_fields ); $this->query_fields = implode( ',', $this->query_fields );
@ -302,7 +315,7 @@ class WP_User_Query {
} elseif ( ! in_array( $qv['fields'], $allowed_fields, true ) ) { } elseif ( ! in_array( $qv['fields'], $allowed_fields, true ) ) {
$this->query_fields = "$wpdb->users.ID"; $this->query_fields = "$wpdb->users.ID";
} else { } else {
$field = 'ID' === $qv['fields'] ? 'ID' : sanitize_key( $qv['fields'] ); $field = 'id' === strtolower( $qv['fields'] ) ? 'ID' : sanitize_key( $qv['fields'] );
$this->query_fields = "$wpdb->users.$field"; $this->query_fields = "$wpdb->users.$field";
} }
@ -822,8 +835,14 @@ class WP_User_Query {
if ( ! $this->results ) { if ( ! $this->results ) {
return; return;
} }
if (
if ( 'all_with_meta' === $qv['fields'] ) { is_array( $qv['fields'] ) &&
isset( $this->results[0]->ID )
) {
foreach ( $this->results as $result ) {
$result->id = $result->ID;
}
} elseif ( 'all_with_meta' === $qv['fields'] ) {
cache_users( $this->results ); cache_users( $this->results );
$r = array(); $r = array();

View File

@ -16,7 +16,7 @@
* *
* @global string $wp_version * @global string $wp_version
*/ */
$wp_version = '6.1-alpha-53360'; $wp_version = '6.1-alpha-53362';
/** /**
* Holds the WordPress DB revision, increments when changes are made to the WordPress DB schema. * Holds the WordPress DB revision, increments when changes are made to the WordPress DB schema.