Optimize batch order generation performance

README.md

@@ -34,6 +34,16 @@ Generate orders with random dates between `--date-start` and `--date-end`.
 Generate orders with a specific status.
 - `wp wc generate orders <nr of orders> --status=completed`
 
+Apply coupons to a percentage of generated orders (0.0-1.0). If no coupons exist, 6 will be created automatically (3 fixed cart, 3 percentage). Note: `--coupons` flag is equivalent to `--coupon-ratio=1.0`.
+
+**Important:** Decimal ratios are converted to percentages using integer rounding. For example, `0.505` becomes 50% (not 50.5%) because the random generation uses integer comparison. Use whole percentages like `0.50` for precise 50% ratios.
+- `wp wc generate orders <nr of orders> --coupon-ratio=0.5`
+
+Refund a percentage of completed orders (0.0-1.0). Refunds will be split evenly between partial and full, and 25% of partial refunds will receive a second partial refund.
+
+**Note:** The same decimal ratio behavior applies to refund ratios as described above for coupon ratios.
+- `wp wc generate orders <nr of orders> --status=completed --refund-ratio=0.3`
+
 #### Order Attribution
 
 Order Attribution represents the origin of data for an order. By default, random values are generated and assigned to the order. Orders with a creation date before 2024-01-09 will not have attribution metadata added, as the feature was not available in WooCommerce at that time.
@@ -52,6 +62,9 @@ Generate coupons with a minimum discount amount.
 Generate coupons with a maximum discount amount.
 - `wp wc generate coupons <nr of coupons> --max=50`
 
+Generate coupons with a specific discount type. Options are `fixed_cart` or `percent`. If not specified, defaults to WooCommerce default (fixed_cart).
+- `wp wc generate coupons <nr of coupons> --discount_type=percent --min=5 --max=25`
+
 ### Customers
 
 Generate customers based on the number of customers parameter.

includes/CLI.php

@@ -316,7 +316,19 @@ function () use ( $progress ) {
 		array(
 			'name'        => 'coupons',
 			'type'        => 'flag',
-			'description' => 'Create and apply a coupon to each generated order.',
+			'description' => 'Create and apply a coupon to each generated order. Equivalent to --coupon-ratio=1.0.',
+			'optional'    => true,
+		),
+		array(
+			'name'        => 'coupon-ratio',
+			'type'        => 'assoc',
+			'description' => 'Decimal ratio (0.0-1.0) of orders that should have coupons applied. If no coupons exist, 6 will be created (3 fixed value, 3 percentage). Note: Decimal values are converted to percentages using integer rounding (e.g., 0.505 becomes 50%).',
+			'optional'    => true,
+		),
+		array(
+			'name'        => 'refund-ratio',
+			'type'        => 'assoc',
+			'description' => 'Decimal ratio (0.0-1.0) of completed orders that should be refunded (wholly or partially). Note: Decimal values are converted to percentages using integer rounding (e.g., 0.505 becomes 50%).',
 			'optional'    => true,
 		),
 		array(
@@ -381,8 +393,15 @@ function () use ( $progress ) {
 			'optional'    => true,
 			'default'     => 100,
 		),
+		array(
+			'name'        => 'discount_type',
+			'type'        => 'assoc',
+			'description' => 'The type of discount for the coupon. If not specified, defaults to WooCommerce default (fixed_cart).',
+			'optional'    => true,
+			'options'     => array( 'fixed_cart', 'percent' ),
+		),
 	),
-	'longdesc'  => "## EXAMPLES\n\nwc generate coupons 10\n\nwc generate coupons 50 --min=1 --max=50",
+	'longdesc'  => "## EXAMPLES\n\nwc generate coupons 10\n\nwc generate coupons 50 --min=1 --max=50\n\nwc generate coupons 20 --discount_type=percent --min=5 --max=25",
 ) );
 
 WP_CLI::add_command( 'wc generate terms', array( 'WC\SmoothGenerator\CLI', 'terms' ), array(

includes/Generator/Coupon.php

@@ -24,13 +24,16 @@ class Coupon extends Generator {
 	public static function generate( $save = true, $assoc_args = array() ) {
 		parent::maybe_initialize_generators();
 
-		$defaults = array(
-			'min' => 5,
-			'max' => 100,
-		);
+	$defaults = array(
+		'min'           => 5,
+		'max'           => 100,
+		'discount_type' => 'fixed_cart',
+	);
+
+		$args = wp_parse_args( $assoc_args, $defaults );
 
 		list( 'min' => $min, 'max' => $max ) = filter_var_array(
-			wp_parse_args( $assoc_args, $defaults ),
+			$args,
 			array(
 				'min' => array(
 					'filter'  => FILTER_VALIDATE_INT,
@@ -68,6 +71,15 @@ public static function generate( $save = true, $assoc_args = array() ) {
 			);
 		}
 
+		// Validate discount_type if provided
+		$discount_type = ! empty( $args['discount_type'] ) ? $args['discount_type'] : '';
+		if ( ! empty( $discount_type ) && ! in_array( $discount_type, array( 'fixed_cart', 'percent' ), true ) ) {
+			return new \WP_Error(
+				'smoothgenerator_coupon_invalid_discount_type',
+				'The discount_type must be either "fixed_cart" or "percent".'
+			);
+		}
+
 		$code        = substr( self::$faker->promotionCode( 1 ), 0, -1 ); // Omit the random digit.
 		$amount      = self::$faker->numberBetween( $min, $max );
 		$coupon_code = sprintf(
@@ -76,11 +88,18 @@ public static function generate( $save = true, $assoc_args = array() ) {
 			$amount
 		);
 
-		$coupon = new \WC_Coupon( $coupon_code );
-		$coupon->set_props( array(
+		$props = array(
 			'code'   => $coupon_code,
 			'amount' => $amount,
-		) );
+		);
+
+		// Only set discount_type if explicitly provided
+		if ( ! empty( $discount_type ) ) {
+			$props['discount_type'] = $discount_type;
+		}
+
+		$coupon = new \WC_Coupon( $coupon_code );
+		$coupon->set_props( $props );
 
 		if ( $save ) {
 			$data_store = WC_Data_Store::load( 'coupon' );
@@ -125,5 +144,39 @@ public static function batch( $amount, array $args = array() ) {
 
 		return $coupon_ids;
 	}
+
+	/**
+	 * Get a random existing coupon.
+	 *
+	 * @param array|null $cached_coupon_ids Optional array of coupon IDs to use instead of querying.
+	 * @return \WC_Coupon|false Coupon object or false if none available.
+	 */
+	public static function get_random( $cached_coupon_ids = null ) {
+		// Use cached IDs if provided (batch mode optimization)
+		if ( null !== $cached_coupon_ids && ! empty( $cached_coupon_ids ) ) {
+			$coupon_ids = $cached_coupon_ids;
+		} else {
+			// Fallback to querying for coupon IDs
+			// Note: Using posts_per_page=-1 loads all coupons into memory for random selection.
+			// For stores with thousands of coupons, consider using direct SQL with RAND() for better performance.
+			// This approach was chosen for consistency with WordPress APIs and to avoid raw SQL queries.
+			$coupon_ids = get_posts(
+				array(
+					'post_type'      => 'shop_coupon',
+					'post_status'    => 'publish',
+					'posts_per_page' => -1,
+					'fields'         => 'ids',
+				)
+			);
+		}
+
+		if ( empty( $coupon_ids ) ) {
+			return false;
+		}
+
+		$random_coupon_id = $coupon_ids[ array_rand( $coupon_ids ) ];
+
+		return new \WC_Coupon( $random_coupon_id );
+	}
 }