WordPress 后台添加数据库导入导出CSV功能

2019独角兽企业重金招聘Python工程师标准>>>

/***************************************************

* Application Actions

* *************************************************/

function application_admin_pages() {

   add_menu_page(__('Application'), __('Application'), 'manage_options', basename(__FILE__), 'application_landing_function');

}

add_action('admin_menu', 'application_admin_pages');

function application_admin_submenu() {

    add_submenu_page( 'functions.php', 'import_export', 'Import & Export', 'manage_options', 'backup-page', 'application_import_export_page');

}

add_action('admin_menu', 'application_admin_submenu');

/* application_landing_function */

function application_landing_function() {?>

<div class="wrap">

<h1>Application</h1>

<table>

<?php

$result = mysql_query("SELECT * FROM _cwcustom_rebate_devices");

while ($table = mysql_fetch_assoc($result)){

$p = $p + 1;

echo '<tr>';

foreach ($table AS $tableName){

echo '<td>'.$tableName.'</td>';

}

echo '</tr>';

} ?>

</table>

</div>

<?php  }

/* application_import_export_page */

function application_import_export_page() {?>

<div class="wrap">

<h1>Application</h1>

<h2>Import CSV</h2>

<form method="post" action="" method="post" enctype="multipart/form-data">

<fieldset>

<legend><strong>Choose a CSV (.csv) file to upload, then click Import .</strong></legend>

<input type="file" name="file" size="50" maxlength="100000" /><br/>

</fieldset>

<p class="submit">

<input type="submit" name="import" class="button button-primary button-import" value="Import" />

</p>

</form>

<!-- End  Import from HTML-->

<?php echo export_UI(); ?><!-- Export CSV  HTML-->

<p></p>

<?php import_form_action();?><!-- Import form action -->

</div>

<script type="text/javascript" charset="utf-8">

$(document).on('click','input.button-import', function(){

if (!confirm("Are You Sure?")) {

return false;

}

})

</script>

<?php }

/* Import Form Action */

function import_form_action() {

define('UPLOAD_PATH', dirname(__FILE__).'/upload/');

global $wpdb;

if ( count($_POST) > 0 && isset($_POST['import']) ) {

$up_info = $_FILES['file'];

if($up_info['error']>0) {

echo "Error: " . $up_info['error']. "<br />";

} else {

$file = fopen($up_info["tmp_name"],"r");

$count = 0;

$table_name = array();

$wpdb->query ('TRUNCATE TABLE _cwcustom_rebate_devices') ;

while(!feof($file) && $data = fgetcsv($file)) {

$column  = count($data);

$result = array();

if($count==0 && !empty($data)) {

for($i=0;$i<$column;$i++){

array_push($table_name,$data[$i]);

}

} elseif ($count>0 && !empty($data)) {

for($i=0;$i<$column;$i++) {

array_push($result,$data[$i]);

}

$temp_result = array();

$i=0;

foreach($table_name as $table_name_key => $table_name_val) {

$temp_result[$table_name_val] = $result[$i];

$i++;

}

$res = $wpdb->insert( "_cwcustom_rebate_devices", $temp_result );

echo $res ? $count.' Succeed<br/>' : $count.' False<br/>';

}

$count++;

}

fclose($file);

}

}

}

/* Export */

define('CONST_CSV_SLUG', 'admin.php?page=backup-page&action=');

function export_UI() {

?>

<div id="wrapper">

<h2>Export CSV</h2>

<div id="page">

<table cellspacing="0" class="wp-list-table widefat">

<thead>

<tr>

<th>Sr.</th>

<th>Export Tables to CSV</th>

</tr>

</thead>

<tbody>

<?php

$array_table_name = array('1' => '_cwcustom_rebate_applications','2' => '_cwcustom_rebate_application_devices','3' => '_cwcustom_rebate_devices');

foreach ($array_table_name as $serial_number => $table_name){?>

<tr>

<td><?php echo $serial_number;?></td>

<td><a class="button button-large" href="<?php echo CONST_CSV_SLUG.$table_name;?>">Export &nbsp; <strong><?php echo $table_name;?></strong></a></td>

</tr>

<?php }

?>

</tbody>

</table>

</div>

</div>

<?php

}

/*Prompt to open/save exported data as .CSV file*/

function application_export(){
$getTable = isset($_REQUEST['action']) ? $_REQUEST['action'] : '';
if ($getTable){
echo csv_generate($getTable);
exit;
}
}
add_action('init', 'application_export');

/* Convert table data into CSV format */

function csv_generate($getTable){

ob_clean();

global $wpdb;

$field='';

$getField ='';

if($getTable){

$result = $wpdb->get_results("SELECT * FROM $getTable");

$requestedTable = mysql_query("SELECT * FROM ".$getTable);

$fieldsCount = mysql_num_fields($requestedTable);

for($i=0; $i<$fieldsCount; $i++){

$field = mysql_fetch_field($requestedTable);

$field = (object) $field;

$getField .= $field->name.',';

}

$sub = substr_replace($getField, '', -1);

$fields = $sub; # GET FIELDS NAME

$each_field = explode(',', $sub);

$csv_file_name = $getTable.'_'.date('Ymd_His').'.csv'; # CSV FILE NAME WILL BE table_name_yyyymmdd_hhmmss.csv

# GET FIELDS VALUES WITH LAST COMMA EXCLUDED

foreach($result as $row){

for($j = 0; $j < $fieldsCount; $j++){

if($j == 0) $fields .= "\n"; # FORCE NEW LINE IF LOOP COMPLETE

$value = str_replace(array("\n", "\n\r", "\r\n", "\r"), "\t", $row->$each_field[$j]); # REPLACE NEW LINE WITH TAB

$value = str_getcsv ( $value , ",", "\"" , "\\"); # SEQUENCING DATA IN CSV FORMAT, REQUIRED PHP >= 5.3.0

$fields .= $value[0].','; # SEPARATING FIELDS WITH COMMA

}

$fields = substr_replace($fields, '', -1); # REMOVE EXTRA SPACE AT STRING END

}

header("Content-type: text/x-csv"); # DECLARING FILE TYPE

header("Content-Transfer-Encoding: binary");

header("Content-Disposition: attachment; filename=".$csv_file_name); # EXPORT GENERATED CSV FILE

header("Pragma: no-cache");

header("Expires: 0");

return $fields;

  }

}

/* End Export*/

转载于:https://my.oschina.net/startphp/blog/271015