WooCommerce 我的账户注册表单添加下拉选择字段

<?php

/**

* @snippet       Add Select Field to "My Account" Register Form | WooCommerce

* @how-to        Get CustomizeWoo.com FREE

* @sourcecode    https://businessbloomer.com/?p=72508

* @author        Rodolfo Melogli

* @testedwith    WooCommerce 3.5.7+

* @donate $9     https://businessbloomer.com/bloomer-armada/

*/

// -------------------

// 1. 在我的账户页面的注册表单添加选择字段

add_action( 'woocommerce_register_form', 'bbloomer_extra_register_select_field' );

function bbloomer_extra_register_select_field() {

?>

	<p class="form-row form-row-wide">

		<label for="find_where"><?php _e( 'Where did you find us?', 'woocommerce' ); ?>  <span class="required">*</span></label>

		<select name="find_where" id="find_where" />

			<option value="goo">Google</option>

			<option value="fcb">Facebook</option>

			<option value="twt">Twitter</option>

		</select>

	</p>

<?php

}

// -------------------

// 2. 在注册时保存字段值

add_action( 'woocommerce_created_customer', 'bbloomer_save_extra_register_select_field' );

function bbloomer_save_extra_register_select_field( $customer_id ) {

	if ( isset( $_POST['find_where'] ) ) {

		update_user_meta( $customer_id, 'find_where', $_POST['find_where'] );

	}

}

// -------------------

// 3. 在后台“我的个人资料”和前台“我的账户-编辑账户”界面显示新增的字段

add_action( 'show_user_profile', 'bbloomer_show_extra_register_select_field', 30 );

add_action( 'edit_user_profile', 'bbloomer_show_extra_register_select_field', 30 ); 

add_action( 'woocommerce_edit_account_form', 'bbloomer_show_extra_register_select_field', 30 );

function bbloomer_show_extra_register_select_field($user){ 

	if (empty ($user) ) {

		$user_id = get_current_user_id();

		$user = get_userdata( $user_id );

	}

	?>    

	<p class="form-row form-row-wide">

		<label for=""><?php _e( 'Where did you find us?', 'woocommerce' ); ?>  <span class="required">*</span></label>

		<select name="find_where" id="find_where" />

			<option disabled value> -- select an option -- </option>

			<option value="goo" <?php if (get_the_author_meta( 'find_where', $user->ID ) == "goo") echo 'selected="selected" '; ?>>Google</option>

			<option value="fcb" <?php if (get_the_author_meta( 'find_where', $user->ID ) == "fcb") echo 'selected="selected" '; ?>>Facebook</option>

			<option value="twt" <?php if (get_the_author_meta( 'find_where', $user->ID ) == "twt") echo 'selected="selected" '; ?>>Twitter</option>

		</select>

	</p>

<?php

}

// -------------------

// 4. 在后台“我的个人资料”和前台“我的账户-编辑账户”保存新增字段的值

add_action( 'personal_options_update', 'bbloomer_save_extra_register_select_field_admin' );    

add_action( 'edit_user_profile_update', 'bbloomer_save_extra_register_select_field_admin' );   

add_action( 'woocommerce_save_account_details', 'bbloomer_save_extra_register_select_field_admin' );

function bbloomer_save_extra_register_select_field_admin( $customer_id ){

	if ( isset( $_POST['find_where'] ) ) {

	   update_user_meta( $customer_id, 'find_where', $_POST['find_where'] );

	}

}