.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 - 11== "\"")//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 - 11== "\"")//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"></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 == 0throw 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;
        }

 

 

posted on 2010-12-21 14:07  blair0807  阅读(1918)  评论(0编辑  收藏  举报