你的位置:首页 > ASP.net教程

[ASP.net教程].NET中的CSV导入导出(实例)


导入代码,从csv文件得到datatable         /// <summary>
        /// Get Data From Csv File 
        /// (Through StreamReader)
        /// </summary>
        /// <returns></returns>
        private bool GetData(Stream inputStream, out string errMessage, out DataTable dtFile)
        {
            errMessage = String.Empty;
            dtFile = new DataTable();

//this.FileUploadImport.PostedFile.InputStream.CanSeek;//StreamReader sr = new StreamReader(this.FileUploadImport.PostedFile.InputStream);//update by hwx 11/12/2010
            StreamReader sr = new StreamReader(inputStream);
            int iColumnCount = 19;//the column count in the file
            int iRow = 1;// the row number is being read
            int iColumn = 0;// the column number is being read
            string strStandardTitle = string.Empty; // the title as it is
            //read title row
            string strTitle = sr.ReadLine();
            //string[] strRowTitle = strTitle.Split(new char[] { ',' });
            string[] strRowTitle = new string[iColumnCount];
            string strCharTitle;
            int iCellNumberTitle = 0;
            bool blnQuoteTitle = false;
            for (int i = 0; i < strTitle.Length; i++)
            {
                strCharTitle = strTitle.Substring(i, 1);
                if (strCharTitle == ",")// "," is the seperation symbol of csv file,
                {
                    if (!blnQuoteTitle)
                    {
                        iCellNumberTitle++;// out of the "" range, "," is the seperation symbol
                        if (iCellNumberTitle >= iColumnCount)// too many column in this line
                        {
                            break;
                        }
                    }
                    else
                    {
                        strRowTitle[iCellNumberTitle] += strCharTitle;
                    }
                }
                else if (strCharTitle == "\"")// "\"" is the transfer symbol of csv file,
                {
                    blnQuoteTitle = !blnQuoteTitle;
                    if (blnQuoteTitle && i > 0 && strTitle.Substring(i - 1, 1) == "\"")//in the "" range and there is an transfer symbol before
                    {
                        strRowTitle[iCellNumberTitle] += strCharTitle;
                    }
                }
                else
                {
                    strRowTitle[iCellNumberTitle] += strCharTitle;
                }
            }
            //read the content
            if (strRowTitle.Length == iColumnCount)
            {
                foreach (string strCell in strRowTitle)
                {
                    iColumn++;
                    if (strCell.Trim() != string.Empty)
                    {
                        dtFile.Columns.Add(strCell);//add new column with name to the data table
                    }
                    else //file error:blank title
                    {
                        errMessage += "The cell " + iColumn.ToString() + " is blank in the header row.\r\n";
                    }
                }
                if (dtFile.Columns.Count == iColumnCount)

//make sure that no blank header or error header
                {
                    //read content row
                    string strLine;
                    while (!sr.EndOfStream)
                    {
                        iRow++;
                        iColumn = 0;
                        DataRow dr = dtFile.NewRow();
                        strLine = sr.ReadLine();

                        //read csv file line by line
                        string[] strRow = new string[iColumnCount];
                        string strChar;
                        int iCellNumber = 0;
                        bool blnQuote = false;//whether in the "" range
                        for (int i = 0; i < strLine.Length; i++)
                        {
                            strChar = strLine.Substring(i, 1);
                            if (strChar == ",")// "," is the seperation symbol of csv file,
                            {
                                if (!blnQuote)
                                {
                                    iCellNumber++;// out of the "" range, "," is the seperation symbol
                                    if (iCellNumber >= iColumnCount)//too many column in this line
                                    {
                                        break;
                                    }
                                }
                                else
                                {
                                    strRow[iCellNumber] += strChar;
                                }
                            }
                            else if (strChar == "\"")// "\"" is the transfer symbol of csv file,
                            {
                                blnQuote = !blnQuote;
                                if (blnQuote && i > 0 && strLine.Substring(i - 1, 1) == "\"")//in the "" range and there is an transfer symbol before
                                {
                                    strRow[iCellNumber] += strChar;
                                }
                            }
                            else
                            {
                                strRow[iCellNumber] += strChar;
                            }
                        }
                        if (iCellNumber + 1 == iColumnCount)
                        {
                            foreach (string strCell in strRow)
                            {
                                iColumn++;
                                if (strCell != null && strCell.Trim() != string.Empty)
                                {
                                    dr[strRowTitle[iColumn - 1]] = strCell.Trim();
                                }
                                else//file error:blank cell
                                {
                                    dr[strRowTitle[iColumn - 1]] = String.Empty;
                                    //errMessage += "The column \"" + strRowTitle[iColumn - 1] + "\" is blank in row " + iRow.ToString() + ".\r\n";
                                }
                            }
                        }
                        else// file error:the column count of current row do not equal to title's
                        {
                            errMessage += "There are more or less cells than title row in the row " + iRow.ToString() + ".\r\n";
                        }
                        dtFile.Rows.Add(dr);
                    }
                }
            }
            else //file error:the count of columns in the file don't equal it should be
            {
                errMessage += "There are an incorrect number of columns in the header row compared to the template file.\r\n";
            }
            sr.Close();
            sr.Dispose();

            errMessage = errMessage.Replace("\r\n", "<br>");
            return errMessage == String.Empty ? true : false;
        }


        /// <summary>
        /// Get dataset from csv file.
        /// </summary>
        /// <param name="filepath">http://www.nuoya118.com</param>
        /// <param name="filename"></param>
        /// <returns>Data Set</returns>
        private DataSet GetDatasetFromCsv(string filepath, string filename)
        {
            string strconn = @"driver={microsoft text driver (*.txt; *.csv)};dbq=";
            strconn += filepath;                                                        //filepath, for example: c:\
            strconn += ";extensions=asc,csv,tab,txt;";
            OdbcConnection objconn = new OdbcConnection(strconn);
            DataSet dscsv = new DataSet();
            try
            {
                string strsql = "select * from " + filename;                     //filename, for example: 1.csv
                OdbcDataAdapter odbccsvdataadapter = new OdbcDataAdapter(strsql, objconn);

                odbccsvdataadapter.Fill(dscsv);
                return dscsv;
            }
            catch (Exception ex)
            {
                throw ex;
            }
        }

 

        csv导出代码

        /// <summary>
        /// Export to Csv File from dataset
        /// </summary>
        /// <param name="src"></param>
        /// <param name="folderName">folderName</param>
        /// <param name="strFileName">strFileName</param>
        /// <returns></returns>
        public bool ExportToCsv(DataSet src, string folderName, string strFileName)
        {
            string csv = String.Empty;

            StreamWriter writer = null;

            string fileName = Server.MapPath("/") + folderName + "\\" + strFileName;

            try
            {
                if (src == null || src.Tables.Count == 0) throw new Exception("dataset is null or has not table in dataset");


                for (int i = 0; i < src.Tables.Count; i++)
                {

                    if (i > 0)

                        fileName = fileName.Substring(0, fileName.IndexOf('.')) + i + fileName.Substring(fileName.IndexOf("."));
                    writer = new StreamWriter(fileName);
                    DataTable dt = src.Tables[i];
                    StringBuilder sb = new StringBuilder();
                    for (int j = 0; j < dt.Columns.Count; j++)
                    {
                        string colName = dt.Columns[j].ColumnName;
                        if (colName.IndexOf(',') > -1)
                        colName = colName.Insert(0, "\"").Insert(colName.Length + 1, "\"");
                        sb.Append(colName);
                        if (!colName.Equals(""))
                            if (j != dt.Columns.Count - 1)
                                sb.Append(",");
                    }
                    writer.WriteLine(sb.ToString());
                    sb = new StringBuilder();
                    string temp = "";
                    for (int j = 0; j < dt.Rows.Count; j++)
                    {
                        DataRow dr = dt.Rows[j];
                        for (int k = 0; k < dt.Columns.Count; k++)
                        {
                            object o = dr[k];
                            if (o != null)
                                temp = o.ToString();
                            if (temp.IndexOf(',') > -1)
                                temp = temp.Insert(0, "\"").Insert(temp.Length + 1, "\"");
                            sb.Append(temp);
                            if (k != dt.Columns.Count - 1)
                                sb.Append(",");
                        }
                        writer.WriteLine(sb.ToString());
                        sb = new StringBuilder();
                        csv = sb.ToString();
                    }
                    writer.Close();
                }

                string strFilePath = Server.MapPath("/") + folderName;
                if (!Directory.Exists(strFilePath))
                {
                    Directory.CreateDirectory(strFilePath);
                }

                strFullFileName = Server.MapPath("/") + folderName + "\\" + fileName;
                //FullFileName = Server.MapPath(FileName);
                //FileName
                FileInfo DownloadFile = new FileInfo(strFullFileName);
                if (DownloadFile.Exists)
                {
                    Response.Clear();
                    Response.ClearHeaders();
                    Response.Buffer = false;
                    Response.ContentType = "application/octet-stream";
                    //Response.AppendHeader("Content-Disposition", "attachment;filename=" + HttpUtility.UrlEncode(DownloadFile.FullName, System.Text.Encoding.ASCII));
                    Response.AppendHeader("Content-Disposition", "attachment;filename=" + HttpUtility.UrlEncode(fileName, System.Text.Encoding.ASCII));
                    Response.AppendHeader("Content-Length", DownloadFile.Length.ToString());
                    Response.WriteFile(DownloadFile.FullName);
                    Response.Flush();
                    //Response.End();
                }
                else
                {
                    //not exist
                    throw new Exception("Export csv file does not exist!");
                }


            }

            catch (Exception ex)
            {
                throw new Exception("Save csv error", ex);
            }

            finally
            {
                if (writer != null) writer.Close();
            }
            return true;
        }

        /// <summary>
        /// List to DataTable
        /// </summary>
        /// <param name="entitys">entitys list</param>
        /// <returns></returns>
        public DataTable ListToDataTable(List<T> entitys)
        {
            if (entitys == null || entitys.Count < 1)
            {
                throw new Exception("list is null");
            }
            //get first Propertie
            Type entityType = entitys[0].GetType();
            PropertyInfo[] entityProperties = entityType.GetProperties();
            //DataTable structure
            //
            DataTable dt = new DataTable();
            for (int i = 0; i < entityProperties.Length; i++)
            {
                //dt.Columns.Add(entityProperties[i].Name, entityProperties[i].PropertyType);
                dt.Columns.Add(entityProperties[i].Name);
            }
            //add entity to DataTable
            foreach (object entity in entitys)
            {
                //check type
                if (entity.GetType() != entityType)
                {
                    throw new Exception("type not same");
                }
                object[] entityValues = new object[entityProperties.Length];
                for (int i = 0; i < entityProperties.Length; i++)
                {
                    entityValues[i] = entityProperties[i].GetValue(entity, null);
                }
                dt.Rows.Add(entityValues);
            }
            return dt;
        }