Wordpress - 使用列标题将mysql表导出到csv

时间:2017-06-28 13:46:23

标签: php mysql wordpress export-to-csv

我可以为我的表下载csv格式的文件,但是如何将列标题添加到同一个文件中。

目前的代码如下 -

// load wpdb
    $path = $_SERVER['DOCUMENT_ROOT'];
    include_once $path . '/wp-load.php';

     global $wpdb;

   $table = $_POST["table_name"];// table name
   $file = 'database_csv'; // csv file name
   $results = $wpdb->get_results("SELECT * FROM $wpdb->prefix$table",ARRAY_A );

   if(count($results) > 0){
      foreach($results as $result){
      $result = array_values($result);
      $result = implode(", ", $result);
      $csv_output .= $result."\n";
    }
  }

  $filename = $file."_".date("Y-m-d_H-i",time());
  header("Content-type: application/vnd.ms-excel");
  header("Content-disposition: csv" . date("Y-m-d") . ".csv");
  header( "Content-disposition: filename=".$filename.".csv");
  header("Pragma: no-cache");
  header("Expires: 0");
  print $csv_output;
  exit;

1 个答案:

答案 0 :(得分:0)

我能够通过首先获取列名然后将其分配给最终输出来实现:

  $table_name = $wpdb->prefix.$_POST["table_name"];// table name
                       $file = 'database_csv'; // csv file name
                       $results = $wpdb->get_results("SELECT * FROM $table_name",ARRAY_A );

                        // get column names
                        $query = "SELECT `COLUMN_NAME` FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `TABLE_SCHEMA`='".$wpdb->dbname."' AND `TABLE_NAME`='".$table_name."'";

                        $columnNamesList = $wpdb->get_results($query);  


                        foreach ( $columnNamesList as $column_name ) {
                            $csv_output.=$column_name->COLUMN_NAME.",";
                        }


                        // remove last additional comma 
                        $csv_output = substr($csv_output,0,strlen($csv_output)-1);

                        // start dumping csv rows in new line
                        $csv_output.="\n";

                       if(count($results) > 0){
                          foreach($results as $result){
                          $result = array_values($result);
                          $result = implode(", ", $result);
                          $csv_output .= $result."\n";
                        }
                      }

                      $filename = $file."_".date("Y-m-d_H-i",time());
                      header("Content-type: application/vnd.ms-excel");
                      header("Content-disposition: csv" . date("Y-m-d") . ".csv");
                      header( "Content-disposition: filename=".$filename.".csv");
                      header("Pragma: no-cache");
                      header("Expires: 0");
                      print $csv_output;
                      exit;