Paid Memberships Pro - Get Discount Code

paid-memberships-pro-get-discount-code.php

<?php
function pmpro_custom_add_discount_code( $atts ) {
	if ( function_exists( 'pmprosm_getCodeByUserID' ) && is_user_logged_in() ) {
		global $current_user;
		$code_id = pmprosm_getCodeByUserID( $current_user->ID ); // Replace $current_user->ID with sponsoree ID.
		if ( $code_id ) {
			$discount_code = pmprosm_getDiscountCodeByCodeID( $code_id );
			if ( isset( $discount_code->code ) ) {
				return $discount_code->code;
			}
		}
	}
	return '';
}
add_shortcode( 'pmpro_discount_code', 'pmpro_custom_add_discount_code' );