Files
sportspress/includes/admin/importers/class-sp-event-importer.php
Brian Miyaji 10ad135093 echo __ => _e
2014-06-23 16:59:28 +10:00

500 lines
15 KiB
PHP

<?php
/**
* Event importer - import events into SportsPress.
*
* @author ThemeBoy
* @category Admin
* @package SportsPress/Admin/Importers
* @version 0.9
*/
if ( ! defined( 'ABSPATH' ) ) exit; // Exit if accessed directly
if ( class_exists( 'WP_Importer' ) ) {
class SP_Event_Importer extends SP_Importer {
/**
* __construct function.
*
* @access public
* @return void
*/
public function __construct() {
$this->import_page = 'sportspress_event_csv';
}
/**
* import function.
*
* @access public
* @param mixed $file
* @return void
*/
function import( $file ) {
global $wpdb;
$this->imported = $this->skipped = 0;
if ( ! is_file($file) ):
$this->footer();
die();
endif;
ini_set( 'auto_detect_line_endings', '1' );
if ( ( $handle = fopen( $file, "r" ) ) !== FALSE ):
$header = fgetcsv( $handle, 0, $this->delimiter );
if ( sizeof( $header ) >= 3 ):
$loop = 0;
// Get event format, league, and season from post vars
$event_format = ( empty( $_POST['sp_format'] ) ? false : $_POST['sp_format'] );
$league = ( empty( $_POST['sp_league'] ) ? false : $_POST['sp_league'] );
$season = ( empty( $_POST['sp_season'] ) ? false : $_POST['sp_season'] );
// Get labels from result and performance post types
$result_labels = sp_get_var_labels( 'sp_result' );
$performance_labels = sp_get_var_labels( 'sp_performance' );
while ( ( $row = fgetcsv( $handle, 0, $this->delimiter ) ) !== FALSE ):
// Slice array into event, team, and player
$event = array_slice( $row, 0, 3 );
$team = array_slice( $row, 3, 3 );
$player = array_slice( $row, 6 );
// Add new event if date is given
if ( sizeof( $event ) > 0 && ! empty( $event[0] ) ):
// Add player performance to last event if available
if ( isset( $id ) && isset( $players ) && sizeof( $players ) > 0 ):
update_post_meta( $id, 'sp_players', $players );
endif;
// List event columns
list( $date, $time, $venue ) = $event;
// Format date by replacing slashes with dashes
$date = str_replace( '/', '-', trim( $date ) );
// Add time to date if given
if ( ! empty( $time ) ):
$date .= ' ' . trim( $time );
endif;
// Define post type args
$args = array( 'post_type' => 'sp_event', 'post_status' => 'publish', 'post_date' => $date );
// Insert event
$id = wp_insert_post( $args );
// Initialize performance array
$players = array();
// Flag as import
update_post_meta( $id, '_sp_import', 1 );
// Update event format
if ( $event_format ):
update_post_meta( $id, 'sp_format', $event_format );
endif;
// Update league
if ( $league ):
wp_set_object_terms( $id, $league, 'sp_league', false );
endif;
// Update season
if ( $season ):
wp_set_object_terms( $id, $season, 'sp_season', false );
endif;
// Update venue
wp_set_object_terms( $id, $venue, 'sp_venue', false );
// Increment
$loop ++;
$this->imported ++;
endif;
// Add new team if team name is given
if ( sizeof( $team ) > 0 && ! empty( $team[0] ) ):
// List team columns
list( $team_name, $result, $outcome ) = $team;
// Find out if team exists
$team_object = get_page_by_title( $team_name, OBJECT, 'sp_team' );
// Get or insert team
if ( $team_object ):
// Make sure team is published
if ( $team_object->post_status != 'publish' ):
wp_update_post( array( 'ID' => $team_object->ID, 'post_status' => 'publish' ) );
endif;
// Get team ID
$team_id = $team_object->ID;
else:
// Insert team
$team_id = wp_insert_post( array( 'post_type' => 'sp_team', 'post_status' => 'publish', 'post_title' => $team_name ) );
// Flag as import
update_post_meta( $team_id, '_sp_import', 1 );
endif;
// Update league
if ( $league ):
wp_set_object_terms( $team_id, $league, 'sp_league', true );
endif;
// Update season
if ( $season ):
wp_set_object_terms( $team_id, $season, 'sp_season', true );
endif;
// Add to event if exists
if ( isset( $id ) ):
// Add team to event
add_post_meta( $id, 'sp_team', $team_id );
// Add empty player to event
add_post_meta( $id, 'sp_player', 0 );
// Explode results into array
$results = explode( '|', $result );
// Create team results array from result keys
$team_results = array();
if ( sizeof( $result_labels ) > 0 ):
foreach( $result_labels as $key => $label ):
$team_results[ $key ] = trim( array_shift( $results ) );
endforeach;
$team_results[ 'outcome' ] = array();
endif;
// Explode outcomes into array
$outcomes = explode( '|', $outcome );
// Add outcome slugs to team outcomes array
foreach ( $outcomes as $outcome ):
// Continue if outcome doesn't exist
if ( $outcome == null ):
continue;
endif;
// Remove whitespace
$outcome = trim( $outcome );
// Get or insert outcome
$outcome_object = get_page_by_title( $outcome, OBJECT, 'sp_outcome' );
if ( $outcome_object ):
// Make sure outcome is published
if ( $outcome_object->post_status != 'publish' ):
wp_update_post( array( 'ID' => $outcome_object->ID, 'post_status' => 'publish' ) );
endif;
// Get outcome slug
$outcome_slug = $outcome_object->post_name;
else:
// Insert outcome
$outcome_id = wp_insert_post( array( 'post_type' => 'sp_outcome', 'post_status' => 'publish', 'post_title' => $outcome ) );
// Get outcome slug
$post_data = get_post( $outcome_id, ARRAY_A );
$outcome_slug = $post_data['post_name'];
// Flag as import
update_post_meta( $outcome_id, '_sp_import', 1 );
endif;
// Add to team results array
$team_results[ 'outcome' ][] = $outcome_slug;
endforeach;
// Get existing results
$event_results = get_post_meta( $id, 'sp_results', true );
// Create new array if results not exists
if ( ! $event_results ):
$event_results = array();
endif;
// Add team results to existing results
$event_results[ $team_id ] = $team_results;
// Update event results
update_post_meta( $id, 'sp_results', $event_results );
// Get event name
$title = get_the_title( $id );
// Add delimiter if event name is set
if ( $title ):
$title .= ' ' . get_option( 'sportspress_event_teams_delimiter', 'vs' ) . ' ';
endif;
// Append team name to event name
$title .= $team_name;
// Update event with new name
$post = array(
'ID' => $id,
'post_title' => $title,
);
wp_update_post( $post );
endif;
endif;
// Add new player if player name is given
if ( sizeof( $player ) > 0 && ! empty( $player[0] ) ):
// Get and unset player name leaving us with the performance
$player_name = $player[0];
unset( $player[0] );
// Find out if player exists
$player_object = get_page_by_title( $player_name, OBJECT, 'sp_player' );
// Get or insert player
if ( $player_object ):
// Make sure player is published
if ( $player_object->post_status != 'publish' ):
wp_update_post( array( 'ID' => $player_object->ID, 'post_status' => 'publish' ) );
endif;
// Get player ID
$player_id = $player_object->ID;
else:
// Insert player
$player_id = wp_insert_post( array( 'post_type' => 'sp_player', 'post_status' => 'publish', 'post_title' => $player_name ) );
// Flag as import
update_post_meta( $player_id, '_sp_import', 1 );
// Update number
update_post_meta( $player_id, 'sp_number', null );
endif;
// Update league
if ( $league ):
wp_set_object_terms( $player_id, $league, 'sp_league', true );
endif;
// Update season
if ( $season ):
wp_set_object_terms( $player_id, $season, 'sp_season', true );
endif;
// Add to event if exists
if ( isset( $id ) ):
// Add player to event
add_post_meta( $id, 'sp_player', $player_id );
// Add player performance to array if team is available
if ( isset( $team_id ) ):
// Initialize performance array
$performance = array();
// Map keys to player performance
foreach ( $performance_labels as $key => $label ):
$performance[ $key ] = array_shift( $player );
endforeach;
$players[ $team_id ][ $player_id ] = $performance;
// Get player teams
$player_teams = get_post_meta( $player_id, 'sp_team', false );
$current_team = get_post_meta( $player_id, 'sp_current_team', true );
$past_teams = get_post_meta( $player_id, 'sp_past_team', false );
// Add team if not exists in player
if ( ! in_array( $team_id, $player_teams ) ):
add_post_meta( $player_id, 'sp_team', $team_id );
endif;
// Add as past team or set current team if not set
if ( ! $current_team ):
update_post_meta( $player_id, 'sp_current_team', $team_id );
elseif ( $current_team != $team_id && ! in_array( $team_id, $past_teams ) ):
add_post_meta( $player_id, 'sp_past_team', $team_id );
endif;
endif;
endif;
endif;
endwhile;
// Add player performance to last event if available
if ( isset( $id ) && isset( $players ) && sizeof( $players ) > 0 ):
update_post_meta( $id, 'sp_players', $players );
endif;
else:
echo '<p><strong>' . __( 'Sorry, there has been an error.', 'sportspress' ) . '</strong><br />';
_e( 'The CSV is invalid.', 'sportspress' ) . '</p>';
$this->footer();
die();
endif;
fclose( $handle );
endif;
// Show Result
echo '<div class="updated settings-error below-h2"><p>
'.sprintf( __( 'Import complete - imported <strong>%s</strong> events and skipped <strong>%s</strong>.', 'sportspress' ), $this->imported, $this->skipped ).'
</p></div>';
$this->import_end();
}
/**
* Performs post-import cleanup of files and the cache
*/
function import_end() {
echo '<p>' . __( 'All done!', 'sportspress' ) . ' <a href="' . admin_url('edit.php?post_type=sp_event') . '">' . __( 'View Events', 'sportspress' ) . '</a>' . '</p>';
do_action( 'import_end' );
}
/**
* header function.
*
* @access public
* @return void
*/
function header() {
echo '<div class="wrap"><h2>' . __( 'Import Events', 'sportspress' ) . '</h2>';
}
/**
* greet function.
*
* @access public
* @return void
*/
function greet() {
echo '<div class="narrow">';
echo '<p>' . __( 'Hi there! Choose a .csv file to upload, then click "Upload file and import".', 'sportspress' ).'</p>';
echo '<p>' . sprintf( __( 'Events need to be defined with columns in a specific order (3+ columns). <a href="%s">Click here to download a sample</a>.', 'sportspress' ), plugin_dir_url( SP_PLUGIN_FILE ) . 'dummy-data/events-sample.csv' ) . '</p>';
$action = 'admin.php?import=sportspress_event_csv&step=1';
$bytes = apply_filters( 'import_upload_size_limit', wp_max_upload_size() );
$size = size_format( $bytes );
$upload_dir = wp_upload_dir();
if ( ! empty( $upload_dir['error'] ) ) :
?><div class="error"><p><?php _e('Before you can upload your import file, you will need to fix the following error:', 'sportspress'); ?></p>
<p><strong><?php echo $upload_dir['error']; ?></strong></p></div><?php
else :
?>
<form enctype="multipart/form-data" id="import-upload-form" method="post" action="<?php echo esc_attr(wp_nonce_url($action, 'import-upload')); ?>">
<table class="form-table">
<tbody>
<tr>
<th>
<label for="upload"><?php _e( 'Choose a file from your computer:', 'sportspress' ); ?></label>
</th>
<td>
<input type="file" id="upload" name="import" size="25" />
<input type="hidden" name="action" value="save" />
<input type="hidden" name="max_file_size" value="<?php echo $bytes; ?>" />
<small><?php printf( __( 'Maximum size: %s', 'sportspress' ), $size ); ?></small>
</td>
</tr>
<tr>
<th>
<label for="file_url"><?php _e( 'OR enter path to file:', 'sportspress' ); ?></label>
</th>
<td>
<?php echo ' ' . ABSPATH . ' '; ?><input type="text" id="file_url" name="file_url" size="25" />
</td>
</tr>
<tr>
<th><label><?php _e( 'Delimiter', 'sportspress' ); ?></label><br/></th>
<td><input type="text" name="delimiter" placeholder="," size="2" /></td>
</tr>
<tr>
<th><label><?php _e( 'Format', 'sportspress' ); ?></label><br/></th>
<td id="sp_formatdiv">
<div id="post-formats-select">
<input type="radio" name="sp_format" class="post-format" id="post-format-league" value="league" checked="checked"> <label for="post-format-league" class="post-format-icon post-format-league">League</label>
<br><input type="radio" name="sp_format" class="post-format" id="post-format-friendly" value="friendly"> <label for="post-format-friendly" class="post-format-icon post-format-friendly">Friendly</label>
<br>
</div>
</td>
</tr>
<tr>
<th><label><?php _e( 'League', 'sportspress' ); ?></label><br/></th>
<td><?php
$args = array(
'taxonomy' => 'sp_league',
'name' => 'sp_league',
'values' => 'slug',
'show_option_none' => __( '-- Not set --', 'sportspress' ),
);
if ( ! sp_dropdown_taxonomies( $args ) ):
echo '<p>' . __( 'None', 'sportspress' ) . '</p>';
sp_taxonomy_adder( 'sp_league', 'sp_team', __( 'Add New', 'sportspress' ) );
endif;
?></td>
</tr>
<tr>
<th><label><?php _e( 'Season', 'sportspress' ); ?></label><br/></th>
<td><?php
$args = array(
'taxonomy' => 'sp_season',
'name' => 'sp_season',
'values' => 'slug',
'show_option_none' => __( '-- Not set --', 'sportspress' ),
);
if ( ! sp_dropdown_taxonomies( $args ) ):
echo '<p>' . __( 'None', 'sportspress' ) . '</p>';
sp_taxonomy_adder( 'sp_season', 'sp_team', __( 'Add New', 'sportspress' ) );
endif;
?></td>
</tr>
</tbody>
</table>
<p class="submit">
<input type="submit" class="button" value="<?php esc_attr_e( 'Upload file and import', 'sportspress' ); ?>" />
</p>
</form>
<?php
endif;
echo '</div>';
}
}
}