asc/sportspress
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve CSV importer close #50

Browse Source
This commit is contained in:
Brian Miyaji
2014-08-26 02:17:00 +10:00
parent a72c045a5b
commit 6b4a421ec2
15 changed files with 903 additions and 811 deletions
Download Patch File Download Diff File Expand all files Collapse all files

152
includes/admin/importers/class-sp-team-importer.php
View File

@@ -5,7 +5,7 @@
* @author ThemeBoy
* @category Admin
* @package SportsPress/Admin/Importers
* @version 0.9
* @version 1.3
*/
if ( ! defined( 'ABSPATH' ) ) exit; // Exit if accessed directly
@@ -21,76 +21,71 @@ if ( class_exists( 'WP_Importer' ) ) {
*/
public function __construct() {
$this->import_page = 'sportspress_team_csv';
$this->import_label = __( 'Import Teams', 'sportspress' );
$this->columns = array(
'post_title' => __( 'Name', 'sportspress' ),
'sp_league' => __( 'Leagues', 'sportspress' ),
'sp_season' => __( 'Seasons', 'sportspress' ),
'sp_url' => __( 'Site URL', 'sportspress' ),
'sp_abbreviation' => __( 'Abbreviation', 'sportspress' ),
'sp_venue' => __( 'Home', 'sportspress' ),
);
parent::__construct();
}
/**
* import function.
*
* @access public
* @param mixed $file
* @param array $array
* @param array $columns
* @return void
*/
function import( $file ) {
global $wpdb;
function import( $array = array(), $columns = array( 'post_title' ) ) {
$this->imported = $this->skipped = 0;
if ( ! is_file($file) ):
if ( ! is_array( $array ) || ! sizeof( $array ) ):
$this->footer();
die();
endif;
ini_set( 'auto_detect_line_endings', '1' );
$rows = array_chunk( $array, sizeof( $columns ) );
if ( ( $handle = fopen( $file, "r" ) ) !== FALSE ):
foreach ( $rows as $row ):
$header = fgetcsv( $handle, 0, $this->delimiter );
if ( empty( array_filter( $row ) ) ) continue;
if ( sizeof( $header ) == 3 ):
$meta = array();
$loop = 0;
foreach ( $columns as $index => $key ):
$meta[ $key ] = sp_array_value( $row, $index );
endforeach;
while ( ( $row = fgetcsv( $handle, 0, $this->delimiter ) ) !== FALSE ):
list( $name, $leagues, $seasons ) = $row;
$team_object = get_page_by_title( $name, OBJECT, 'sp_team' );
if ( ! $name || $team_object ):
$this->skipped++;
continue;
endif;
$args = array( 'post_type' => 'sp_team', 'post_status' => 'publish', 'post_title' => $name );
$id = wp_insert_post( $args );
// Flag as import
update_post_meta( $id, '_sp_import', 1 );
// Update leagues
$leagues = explode( '|', $leagues );
wp_set_object_terms( $id, $leagues, 'sp_league', false );
// Update seasons
$seasons = explode( '|', $seasons );
wp_set_object_terms( $id, $seasons, 'sp_season', false );
$loop ++;
$this->imported++;
endwhile;
else:
echo '<p><strong>' . __( 'Sorry, there has been an error.', 'sportspress' ) . '</strong><br />';
_e( 'The CSV is invalid.', 'sportspress' ) . '</p>';
$this->footer();
die();
$name = sp_array_value( $meta, 'post_title' );
if ( ! $name ):
$this->skipped++;
continue;
endif;
fclose( $handle );
endif;
$args = array( 'post_type' => 'sp_team', 'post_status' => 'publish', 'post_title' => $name );
$id = wp_insert_post( $args );
// Flag as import
update_post_meta( $id, '_sp_import', 1 );
// Update leagues
$leagues = explode( '|', sp_array_value( $meta, 'sp_league' ) );
wp_set_object_terms( $id, $leagues, 'sp_league', false );
// Update seasons
$seasons = explode( '|', sp_array_value( $meta, 'sp_season' ) );
wp_set_object_terms( $id, $seasons, 'sp_season', false );
$this->imported++;
endforeach;
// Show Result
echo '<div class="updated settings-error below-h2"><p>
@@ -109,16 +104,6 @@ if ( class_exists( 'WP_Importer' ) ) {
do_action( 'import_end' );
}
/**
* header function.
*
* @access public
* @return void
*/
function header() {
echo '<div class="wrap"><h2>' . __( 'Import Teams', 'sportspress' ) . '</h2>';
}
/**
* greet function.
*
@@ -126,57 +111,10 @@ if ( class_exists( 'WP_Importer' ) ) {
* @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( __( 'Teams 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/teams-sample.csv' ) . '</p>';
$action = 'admin.php?import=sportspress_team_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>
</tbody>
</table>
<p class="submit">
<input type="submit" class="button" value="<?php esc_attr_e( 'Upload file and import', 'sportspress' ); ?>" />
</p>
</form>
<?php
endif;
wp_import_upload_form( 'admin.php?import=sportspress_team_csv&step=1' );
echo '</div>';
}
}