如何在不接触核心文件的情况下为WooCommerce订阅插件中的订阅添加自定义字段？

Question

我正在使用WooCommerce订阅来管理我的WordPress网站上的经常性付款，该网站出售婴儿尿布。

我希望向订阅中添加其他详细信息，以便我的客户可以通过订阅id以外的其他方式区分他们的订阅。

在顾客之旅开始时，我以一种形式捕捉顾客的线索，包括他们的名字。

为了将额外的详细信息字段添加到订阅数据中，我修改了位于下面的wcs_create_subscription函数：

wp-content/plugins/woocommerce-subscriptions/vendor/woocommerce/subscriptions-core/wcs-functions.php

首先在$default_args数组中添加一个‘$default_args’参数，设置为用get_current_user_id()存储当前用户id，只为了测试功能：

$default_args = array(
    'status'             => '',
    'baby_name'          => get_current_user_id(),
    'order_id'           => 0,
    'customer_note'      => null,
    'customer_id'        => ( ! empty( $order ) ) ? $order->get_user_id() : null,
    'start_date'         => $default_start_date,
    'date_created'       => $now,
    'created_via'        => ( ! empty( $order ) ) ? wcs_get_objects_property( $order, 'created_via' ) : '',
    'order_version'      => ( ! empty( $order ) ) ? wcs_get_objects_property( $order, 'version' ) : WC_VERSION,
    'currency'           => ( ! empty( $order ) ) ? wcs_get_objects_property( $order, 'currency' ) : get_woocommerce_currency(),
    'prices_include_tax' => ( ! empty( $order ) ) ? ( ( wcs_get_objects_property( $order, 'prices_include_tax' ) ) ? 'yes' : 'no' ) : get_option( 'woocommerce_prices_include_tax' ), // we don't use wc_prices_include_tax() here because WC doesn't use it in wc_create_order(), not 100% sure why it doesn't also check the taxes are enabled, but there could forseeably be a reason
);

然后运行一个update_post_meta()函数将参数保存在订阅数据库中：

update_post_meta( $subscription_id, '_baby_name', $args['baby_name'] );

以下是我的完整wcs_create_subscription函数：

/**
 * Create a new subscription
 *
 * Returns a new WC_Subscription object on success which can then be used to add additional data.
 *
 * @return WC_Subscription | WP_Error A WC_Subscription on success or WP_Error object on failure
 * @since  2.0
 */
function wcs_create_subscription( $args = array() ) {

    $now   = gmdate( 'Y-m-d H:i:s' );
    $order = ( isset( $args['order_id'] ) ) ? wc_get_order( $args['order_id'] ) : null;

    if ( ! empty( $order ) ) {
        $default_start_date = wcs_get_datetime_utc_string( wcs_get_objects_property( $order, 'date_created' ) );
    } else {
        $default_start_date = ( isset( $args['date_created'] ) ) ? $args['date_created'] : $now;
    }

    $default_args = array(
        'status'             => '',
        'baby_name'          => get_current_user_id(),
        'order_id'           => 0,
        'customer_note'      => null,
        'customer_id'        => ( ! empty( $order ) ) ? $order->get_user_id() : null,
        'start_date'         => $default_start_date,
        'date_created'       => $now,
        'created_via'        => ( ! empty( $order ) ) ? wcs_get_objects_property( $order, 'created_via' ) : '',
        'order_version'      => ( ! empty( $order ) ) ? wcs_get_objects_property( $order, 'version' ) : WC_VERSION,
        'currency'           => ( ! empty( $order ) ) ? wcs_get_objects_property( $order, 'currency' ) : get_woocommerce_currency(),
        'prices_include_tax' => ( ! empty( $order ) ) ? ( ( wcs_get_objects_property( $order, 'prices_include_tax' ) ) ? 'yes' : 'no' ) : get_option( 'woocommerce_prices_include_tax' ), // we don't use wc_prices_include_tax() here because WC doesn't use it in wc_create_order(), not 100% sure why it doesn't also check the taxes are enabled, but there could forseeably be a reason
    );

    $args              = wp_parse_args( $args, $default_args );
    $subscription_data = array();

    // Validate the date_created arg.
    if ( ! is_string( $args['date_created'] ) || false === wcs_is_datetime_mysql_format( $args['date_created'] ) ) {
        return new WP_Error( 'woocommerce_subscription_invalid_date_created_format', _x( 'Invalid created date. The date must be a string and of the format: "Y-m-d H:i:s".', 'Error message while creating a subscription', 'woocommerce-subscriptions' ) );
    } elseif ( wcs_date_to_time( $args['date_created'] ) > current_time( 'timestamp', true ) ) {
        return new WP_Error( 'woocommerce_subscription_invalid_date_created', _x( 'Subscription created date must be before current day.', 'Error message while creating a subscription', 'woocommerce-subscriptions' ) );
    }

    // Validate the start_date arg.
    if ( ! is_string( $args['start_date'] ) || false === wcs_is_datetime_mysql_format( $args['start_date'] ) ) {
        return new WP_Error( 'woocommerce_subscription_invalid_start_date_format', _x( 'Invalid date. The date must be a string and of the format: "Y-m-d H:i:s".', 'Error message while creating a subscription', 'woocommerce-subscriptions' ) );
    }

    // check customer id is set
    if ( empty( $args['customer_id'] ) || ! is_numeric( $args['customer_id'] ) || $args['customer_id'] <= 0 ) {
        return new WP_Error( 'woocommerce_subscription_invalid_customer_id', _x( 'Invalid subscription customer_id.', 'Error message while creating a subscription', 'woocommerce-subscriptions' ) );
    }

    // check the billing period
    if ( empty( $args['billing_period'] ) || ! in_array( strtolower( $args['billing_period'] ), array_keys( wcs_get_subscription_period_strings() ) ) ) {
        return new WP_Error( 'woocommerce_subscription_invalid_billing_period', __( 'Invalid subscription billing period given.', 'woocommerce-subscriptions' ) );
    }

    // check the billing interval
    if ( empty( $args['billing_interval'] ) || ! is_numeric( $args['billing_interval'] ) || absint( $args['billing_interval'] ) <= 0 ) {
        return new WP_Error( 'woocommerce_subscription_invalid_billing_interval', __( 'Invalid subscription billing interval given. Must be an integer greater than 0.', 'woocommerce-subscriptions' ) );
    }

    $subscription_data['post_type']     = 'shop_subscription';
    $subscription_data['post_status']   = 'wc-' . apply_filters( 'woocommerce_default_subscription_status', 'pending' );
    $subscription_data['ping_status']   = 'closed';
    $subscription_data['post_author']   = 1;
    $subscription_data['post_password'] = uniqid( 'order_' );
    // translators: Order date parsed by strftime
    $post_title_date = strftime( _x( '%b %d, %Y @ %I:%M %p', 'Used in subscription post title. "Subscription renewal order - <this>"', 'woocommerce-subscriptions' ) ); // phpcs:ignore WordPress.WP.I18n.UnorderedPlaceholdersText
    // translators: placeholder is order date parsed by strftime
    $subscription_data['post_title']    = sprintf( _x( 'Subscription &ndash; %s', 'The post title for the new subscription', 'woocommerce-subscriptions' ), $post_title_date );
    $subscription_data['post_date_gmt'] = $args['date_created'];
    $subscription_data['post_date']     = get_date_from_gmt( $args['date_created'] );

    if ( $args['order_id'] > 0 ) {
        $subscription_data['post_parent'] = absint( $args['order_id'] );
    }

    if ( ! is_null( $args['customer_note'] ) && ! empty( $args['customer_note'] ) ) {
        $subscription_data['post_excerpt'] = $args['customer_note'];
    }

    // Only set the status if creating a new subscription, use wcs_update_subscription to update the status
    if ( $args['status'] ) {
        if ( ! in_array( 'wc-' . $args['status'], array_keys( wcs_get_subscription_statuses() ) ) ) {
            return new WP_Error( 'woocommerce_invalid_subscription_status', __( 'Invalid subscription status given.', 'woocommerce-subscriptions' ) );
        }
        $subscription_data['post_status']  = 'wc-' . $args['status'];
    }

    $subscription_id = wp_insert_post( apply_filters( 'woocommerce_new_subscription_data', $subscription_data, $args ), true );

    if ( is_wp_error( $subscription_id ) ) {
        return $subscription_id;
    }

    // Default order meta data.
    update_post_meta( $subscription_id, '_order_key', wcs_generate_order_key() );
    update_post_meta( $subscription_id, '_order_currency', $args['currency'] );
    update_post_meta( $subscription_id, '_prices_include_tax', $args['prices_include_tax'] );
    update_post_meta( $subscription_id, '_created_via', sanitize_text_field( $args['created_via'] ) );

    // add/update the billing
    update_post_meta( $subscription_id, '_billing_period', $args['billing_period'] );
    update_post_meta( $subscription_id, '_billing_interval', absint( $args['billing_interval'] ) );

    update_post_meta( $subscription_id, '_customer_user', $args['customer_id'] );
    update_post_meta( $subscription_id, '_order_version', $args['order_version'] );

    update_post_meta( $subscription_id, '_schedule_start', $args['start_date'] );
    update_post_meta( $subscription_id, '_baby_name', $args['baby_name'] );

    /**
     * Filter the newly created subscription object.
     *
     * @since 2.2.22
     * @param WC_Subscription $subscription
     */
    $subscription = apply_filters( 'wcs_created_subscription', wcs_get_subscription( $subscription_id ) );

    /**
     * Triggered after a new subscription is created.
     *
     * @since 2.2.22
     * @param WC_Subscription $subscription
     */
    do_action( 'wcs_create_subscription', $subscription );

    return $subscription;
}

我可以确认这是可行的，因为我可以在_baby_name字段中看到meta_key字段中的行，在meta_value字段中看到我的用户id用于使用id 2898进行订阅，而我最近将其放在sites _postmeta数据库表中：

​

​

因为我知道我永远不应该修改核心插件文件，我希望我的修改在一个片段或类似的，我可以放在我的子主题function.php文件。有什么建议吗？

Answer 1

从您发布的代码可以推断，可以使用wcs_create_subscription动作钩子：

/**
 * Triggered after a new subscription is created.
 *
 * @since 2.2.22
 * @param WC_Subscription $subscription
 */
function action_wcs_create_subscription( $subscription ) {
    // Get ID
    $subscription_id = $subscription->get_id();

    update_post_meta( $subscription_id, '_baby_name', 'my_value' );
}
add_action( 'wcs_create_subscription', 'action_wcs_create_subscription', 10, 1 );

没有必要先把这作为一个论点来添加。如果需要，您可以通过$subscription获得多个数据