使用PHP導入和導出CSV文件

項目開發中，很多時候要將外部CSV文件導入到數據庫中或者將數據導出為CSV文件，那么具體該如何實現呢？本文將使用PHP并結合mysql，實現了CSV格式數據的導入和導出功能。

我們先準備mysql數據表，假設項目中有一張記錄學生信息的表student，并有id，name，sex，age分別記錄學生的姓名、性別、年齡等信息。

CREATE TABLE `student` (
  `id` int(11) NOT NULL auto_increment,
  `name` varchar(50) NOT NULL,
  `sex` varchar(10) NOT NULL,
  `age` smallint(3) NOT NULL default '0',
  PRIMARY KEY  (`id`)
) ENGINE=MyISAM  DEFAULT CHARSET=utf8;

我們還需要一個html交互頁面，放置導入表單和導出按鈕。

<form id="addform" action="do.php?action=import" method="post" enctype="multipart/form-data">
    <p>請選擇要導入的CSV文件：<br/><input type="file" name="file"> <input type="submit"
    class="btn" value="導入CSV">
    <input type="button" class="btn" value="導出CSV" onclick="window.location.href='do.php?
    action=export'"></p>
</form>

選擇好本地csv文件后，點擊導入，提交到do.php?action=import處理，而點擊導出按鈕則請求地址do.php?action=export進行數據導出處理。

1.導入CSV

do.php需要根據get過來的參數，分別處理導入和導出過程，php結構如下：

include_once ("connect.php"); //連接數據庫

$action = $_GET['action'];
if ($action == 'import') { //導入CSV
   //導入處理
} elseif ($action=='export') { //導出CSV
   //導出處理
}</pre>

導入CSV處理流程：校驗csv文件合法性（本文忽略）->打開讀入并解析csv文件中的字段->循環獲取各字段值->批量添加到數據表中->完成。



if ($action == 'import') { //導入CSV
    $filename = $_FILES['file']['tmp_name'];
    if (empty ($filename)) {
        echo '請選擇要導入的CSV文件！';
        exit;
    }
    $handle = fopen($filename, 'r');
    $result = input_csv($handle); //解析csv
    $len_result = count($result);
    if($len_result==0){
        echo '沒有任何數據！';
        exit;
    }
    for ($i = 1; $i < $len_result; $i++) { //循環獲取各字段值
        $name = iconv('gb2312', 'utf-8', $result[$i][0]); //中文轉碼
        $sex = iconv('gb2312', 'utf-8', $result[$i][1]);
        $age = $result[$i][2];
        $data_values .= "('$name','$sex','$age'),";
    }
    $data_values = substr($data_values,0,-1); //去掉最后一個逗號
    fclose($handle); //關閉指針
    $query = mysql_query("insert into student (name,sex,age) values $data_values");//批量插入數據表中
    if($query){
        echo '導入成功！';
    }else{
        echo '導入失敗！';
    }
}


注意php自帶的fgetcsv函數可以輕松處理csv，使用該函數可以從文件指針中讀入一行并解析CSV字段。下面的函數將csv文件字段解析并以數組的形式返回。



function input_csv($handle) {
    $out = array ();
    $n = 0;
    while ($data = fgetcsv($handle, 10000)) {
        $num = count($data);
        for ($i = 0; $i < $num; $i++) {
            $out[$n][$i] = $data[$i];
        }
        $n++;
    }
    return $out;
}


此外在導入到數據庫中時，我們采用的是批量插入而不是一條條插入的，因此在構建SQL語句時，要稍作處理，見代碼。



2.導出CSV



我們知道csv文件是由逗號分割符組成的純文本文件，你可以用excel打開，效果跟xls表個一樣。



導出CSV處理流程：讀取學生信息表->循環記錄構建逗號分隔的字段信息->設置header信息->導出文件（下載）到本地



...
} elseif ($action=='export') { //導出CSV
    $result = mysql_query("select * from student order by id asc");
    $str = "姓名,性別,年齡\n";
    $str = iconv('utf-8','gb2312',$str);
    while($row=mysql_fetch_array($result)){
        $name = iconv('utf-8','gb2312',$row['name']); //中文轉碼
        $sex = iconv('utf-8','gb2312',$row['sex']);
        $str .= $name.",".$sex.",".$row['age']."\n"; //用引文逗號分開
    }
    $filename = date('Ymd').'.csv'; //設置文件名
    export_csv($filename,$str); //導出
}


要將數據導出到本地即下在，需要修改header信息，代碼如下：



function export_csv($filename,$data) {
    header("Content-type:text/csv");
    header("Content-Disposition:attachment;filename=".$filename);
    header('Cache-Control:must-revalidate,post-check=0,pre-check=0');
    header('Expires:0');
    header('Pragma:public');
    echo $data;
}


注意導入和導出的過程中，因為我們使用的是統一UTF-8編碼，遇到中文字符一定要記得轉碼，否則可能會出現中文亂碼的情況。