function wcm_member_badge () {
$user_id = wc_memberships_get_user_memberships();
if(wc_membership_hasMembershipLevel ('1', $user_id)) {
?>
<span class='wcm-member-badge' data-balloon-pos="up" data-balloon="Pro Member">
<p>Level 1</p>
</span>
<?php
}
if(wc_membership_hasMembershipLevel ('2', $user_id)) {
?>
<span class='wcm-member-badge' data-balloon-pos="up" data-balloon="Pro Member">
<p>Level 2</p>
</span>
<?php
}
else {
?>
<span class='wcm-member-badge' data-balloon-pos="up" data-balloon="Pro Member">
<p>No Active Membership</p>
</span>
<?php
}
}
add_action( 'bp_member_members_list_item', 'wcm_member_badge' );
add_action( 'bp_profile_field_item', 'wcm_member_badge' );
