Forked from strangerstudios/requires_membership_categories_columns.php
Last active
January 15, 2020 02:56
-
-
Save kimwhite/e7102f45f4dfadc3a197862af663117a to your computer and use it in GitHub Desktop.
Add a "Requires Membership" column to the Categories dashboard page to show what levels are required to view.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
<?php | |
/** | |
* Show a “Requires Membership” Column on the Categories Screen | |
* This is an easy way to see how your content is being restricted for members. | |
* | |
* You can add this recipe to your site by creating a custom plugin | |
* or using the Code Snippets plugin available for free in the WordPress repository. | |
* Read this companion article for step-by-step directions on either method. | |
* https://www.paidmembershipspro.com/create-a-plugin-for-pmpro-customizations/ | |
*/ | |
//add a new column to the categories view | |
function requires_membership_categories_columns_head( $defaults ) { | |
$defaults['requires_membership'] = 'Requires Membership'; | |
return $defaults; | |
} | |
//get the column data | |
function requires_membership_categories_columns_content( $content, $column_name, $term_id ) { | |
if ($column_name == 'requires_membership') { | |
global $membership_levels, $wpdb; | |
$protected_levels = array(); | |
foreach($membership_levels as $level) | |
{ | |
$protectedcategories = $wpdb->get_col("SELECT category_id FROM $wpdb->pmpro_memberships_categories WHERE membership_id = $level->id"); | |
if( in_array( $term_id, $protectedcategories ) ) { | |
$protected_levels[] = $level->name; | |
} | |
} | |
echo implode( ', ', $protected_levels); | |
} | |
} | |
add_filter( 'manage_edit-category_columns', 'requires_membership_categories_columns_head' ); | |
add_action( 'manage_category_custom_column', 'requires_membership_categories_columns_content', 10, 3 ); |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment