WooCommerce自定义注册上传字段 后台获取附件URL实现下载

需求说明 #

• 搭建仅会员可访问的站点，已在WooCommerce注册页面添加文件上传输入框
• 需要将用户上传的文件以下载链接/按钮形式展示在WordPress后台用户管理菜单中，供管理员校验用户提交的材料
• 当前问题：get_the_author_meta( 'image_files', $user->ID )仅返回上传文件的附件ID，无法直接拼接下载链接，需要获取对应文件的访问URL

现有代码 #

1. 注册表单添加文件上传输入框 #

// Add file input to register form
function add_woo_account_registration_fields() {
    
   ?>
    
   <p class="form-row" id="image_files">

      <label for="image_files" class="">Image (JPG, PNG, PDF)</label>

      <span class="woocommerce-input-wrapper">
         <input type='file' name='image_files' accept='image/*,.pdf'>
      </span>

   </p>
    
   <?php
       
}

add_action( 'woocommerce_register_form', 'add_woo_account_registration_fields' );

2. 自定义字段校验逻辑 #

// 2. Validate new field
function validate_woo_account_registration_fields( $errors, $username, $email ) {
   if ( isset( $_POST['image_files'] ) && empty( $_POST['image_files'] ) ) {
      $errors->add( 'image_files_error', __( 'Please provide a valid image', 'woocommerce' ) );
   }
   return $errors;
}

add_filter( 'woocommerce_registration_errors', 'validate_woo_account_registration_fields', 10, 3 );

3. 注册时上传文件存储逻辑 #

// Save new field
function save_woo_account_registration_fields( $customer_id ) {
   if ( isset( $_FILES['image_files'] ) ) {
      require_once( ABSPATH . 'wp-admin/includes/image.php' );
      require_once( ABSPATH . 'wp-admin/includes/file.php' );
      require_once( ABSPATH . 'wp-admin/includes/media.php' );
      $attachment_id = media_handle_upload( 'image_files', 0 );
      if ( is_wp_error( $attachment_id ) ) {
         update_user_meta( $customer_id, 'image_files', $_FILES['image_files'] . ": " . $attachment_id->get_error_message() );
      } else {
         update_user_meta( $customer_id, 'image_files', $attachment_id );
      }
   }
}

add_action( 'user_register', 'save_woo_account_registration_fields', 1 );

4. 注册表单添加文件上传支持 #

// Add enctype to form to allow image upload
function enctype_custom_registration_forms() {
   echo 'enctype="multipart/form-data"';
}

add_action( 'woocommerce_register_form_tag', 'enctype_custom_registration_forms' );

5. 后台用户页字段展示（原有错误版本） #

// Add field - admin
function add_user_files_field( $user ) {
   ?>
       <h3><?php _e('File','woocommerce' ); ?></h3>
       <table class="form-table">
           <tr>
               <th>
                  <label for="image_files"><?php _e( 'Files', 'woocommerce' ); ?></label>
               </th>
               <td>
                  
                  <!-- download file -->
                  <a href="<?php echo esc_attr( get_the_author_meta( 'image_files', $user->ID ) ); ?>" download>Download</a>
                  
               </td>
           </tr>
       </table>
       <br/>
   <?php
}
add_action( 'show_user_profile', 'add_user_files_field', 10, 1 );
add_action( 'edit_user_profile', 'add_user_files_field', 10, 1 );

6. 后台字段保存逻辑 #

// Save field - admin
function save_user_files_field( $user_id ) {
   if( ! empty($_POST['image_files']) ) {
       update_user_meta( $user_id, 'image_files', sanitize_text_field( $_POST['image_files'] ) );
   }
}
add_action( 'personal_options_update', 'save_user_files_field', 10, 1 );
add_action( 'edit_user_profile_update', 'save_user_files_field', 10, 1 );

解决方法 #

你存储的image_files字段值是WordPress媒体库的附件ID，直接使用WordPress原生函数wp_get_attachment_url()传入附件ID即可获取对应文件的完整访问URL。
需要额外做容错判断：如果上传失败，字段存储的是错误字符串而非数字ID，直接调用函数会报错，所以要先判断值是否为有效附件ID。
替换后台用户页字段展示的代码即可：

// Add field - admin 修正版
function add_user_files_field( $user ) {
   $file_value = get_user_meta( $user->ID, 'image_files', true );
   ?>
       <h3><?php _e('File','woocommerce' ); ?></h3>
       <table class="form-table">
           <tr>
               <th>
                  <label for="image_files"><?php _e( 'Files', 'woocommerce' ); ?></label>
               </th>
               <td>
                  <?php if ( is_numeric( $file_value ) && wp_get_attachment_url( $file_value ) ) : 
                    $file_url = wp_get_attachment_url( $file_value );
                  ?>
                  <!-- 有效文件下载链接 -->
                  <a href="<?php echo esc_url( $file_url ); ?>" class="button button-primary" download>下载用户提交文件</a>
                  <?php else : ?>
                  <!-- 上传失败显示错误信息 -->
                  <span class="description" style="color:#d63638;"><?php echo esc_html( $file_value ?: '用户未上传文件' ); ?></span>
                  <?php endif; ?>
               </td>
           </tr>
       </table>
       <br/>
   <?php
}
add_action( 'show_user_profile', 'add_user_files_field', 10, 1 );
add_action( 'edit_user_profile', 'add_user_files_field', 10, 1 );

额外注意事项 #

• 原有后台字段保存逻辑存在bug：如果管理员编辑用户资料时没有提交image_files字段，会把原有存储的附件ID清空，如果不需要在后台编辑该上传字段，建议直接删除save_user_files_field相关的代码。
• 输出URL时使用esc_url()做安全转义，不要用esc_attr()。

内容的提问来源于stack exchange，提问作者IamLearningPHP