zoukankan      html  css  js  c++  java
  • C#操作CSV

    //这里写了一个通用的类

    using System;
    using System.Data;
    using System.Configuration;
    using System.Collections.Generic;
    using System.Web;
    using System.Web.Security;
    using System.Web.UI;
    using System.Web.UI.WebControls;
    using System.Web.UI.WebControls.WebParts;
    using System.Web.UI.HtmlControls;
    using System.Text;
    using System.Reflection;
    using System.IO;
    using System.Data.Odbc;

    namespace Com.DRPENG.SDXY.UI.Common
    {
        public class CSVHelper
        {
            #region Fields
            string _fileName;
            DataTable _dataSource;//数据源
            string[] _titles = null;//列标题
            string[] _fields = null;//字段名

            #endregion

            #region .ctor
            /// <summary>
            /// 构造函数
            /// </summary>
            /// <param name="dataSource">数据源</param>
            public CSVHelper()
            {
              
            }

            /// <summary>
            /// 构造函数
            /// </summary>
            /// <param name="titles">要输出到 Excel 的列标题的数组</param>
            /// <param name="fields">要输出到 Excel 的字段名称数组</param>
            /// <param name="dataSource">数据源</param>
            public CSVHelper(string[] titles, string[] fields, DataTable dataSource)
                : this(titles, dataSource)
            {
                if (fields == null || fields.Length == 0)
                    throw new ArgumentNullException("fields");
                if (titles.Length != fields.Length)
                    throw new ArgumentException("titles.Length != fields.Length", "fields");

                _fields = fields;
            }

            /// <summary>
            /// 构造函数
            /// </summary>
            /// <param name="titles">要输出到 Excel 的列标题的数组</param>
            /// <param name="dataSource">数据源</param>
            public CSVHelper(string[] titles, DataTable dataSource)
                : this(dataSource)
            {
                if (titles == null || titles.Length == 0)
                    throw new ArgumentNullException("titles");

                _titles = titles;
            }

            /// <summary>
            /// 构造函数
            /// </summary>
            /// <param name="dataSource">数据源</param>
            public CSVHelper(DataTable dataSource)
            {
                if (dataSource == null)
                    throw new ArgumentNullException("dataSource");
                // maybe more checks needed here (IEnumerable, IList, IListSource, ) ???
                // 很难判断,先简单的使用 DataTable

                _dataSource = dataSource;
            }

            #endregion

            #region public Methods

            #region  导出到CSV文件并且提示下载
            /// <summary>
            /// 导出到CSV文件并且提示下载
            /// </summary>
            /// <param name="fileName"></param>
            public void DataToCSV(string fileName)
            {
                // 确保有一个合法的输出文件名
                //if (fileName == null || fileName == string.Empty || !(fileName.ToLower().EndsWith(".csv")))
                //    fileName = GetRandomFileName();
                string data = ExportCSV();
                HttpContext.Current.Response.ClearHeaders();
                HttpContext.Current.Response.Clear();
                HttpContext.Current.Response.Expires = 0;
                HttpContext.Current.Response.BufferOutput = true;
                HttpContext.Current.Response.Charset = "GB2312";
                HttpContext.Current.Response.ContentEncoding = System.Text.Encoding.GetEncoding("GB2312");
                HttpContext.Current.Response.AppendHeader("Content-Disposition", string.Format("attachment;filename={0}.csv", System.Web.HttpUtility.UrlEncode(fileName, System.Text.Encoding.UTF8)));
                HttpContext.Current.Response.ContentType = "text/h323;charset=gbk";
                HttpContext.Current.Response.Write(data);
                HttpContext.Current.Response.End();
            }

            #endregion

            /// <summary>
            /// 获取CSV导入的数据
            /// </summary>
            /// <param name="filePath">文件路径</param>
            /// <param name="fileName">文件名称(.csv不用加)</param>
            /// <returns></returns>
            public DataTable GetCsvData(string filePath,string fileName)
            {
                string path = Path.Combine(filePath, fileName + ".csv");
                string connString = @"Driver={Microsoft Text Driver (*.txt; *.csv)};Dbq=" + filePath + ";Extensions=asc,csv,tab,txt;";
                try
                {
                    using (OdbcConnection odbcConn = new OdbcConnection(connString))
                    {
                        odbcConn.Open();
                        OdbcCommand oleComm = new OdbcCommand();
                        oleComm.Connection = odbcConn;
                        oleComm.CommandText = "select * from [" + fileName + "#csv]";
                        OdbcDataAdapter adapter = new OdbcDataAdapter(oleComm);
                        DataSet ds = new DataSet();
                        adapter.Fill(ds, fileName);
                        return ds.Tables[0];
                        odbcConn.Close();
                    }
                    if (File.Exists(path))
                    {
                        File.Delete(path);
                    }
                }
                catch (Exception ex)
                {
                    if (File.Exists(path))
                    {
                        File.Delete(path);
                    }
                    throw ex;
                }
            }
            #endregion

            #region 返回写入CSV的字符串
            /// <summary>
            /// 返回写入CSV的字符串
            /// </summary>
            /// <returns></returns>
            private string ExportCSV()
            {   
                if(_dataSource==null)
                    throw new ArgumentNullException("dataSource");

                StringBuilder strbData = new StringBuilder();
                if (_titles == null)
                {
                    //添加列名
                    foreach (DataColumn column in _dataSource.Columns)
                    {
                        strbData.Append(column.ColumnName + ",");
                    }
                    strbData.Append("\n");
                    foreach (DataRow dr in _dataSource.Rows)
                    {
                        for (int i = 0; i < _dataSource.Columns.Count; i++)
                        {
                            strbData.Append(dr[i].ToString() + ",");
                        }
                        strbData.Append("\n");
                    }
                    return strbData.ToString();
                }
                else
                {
                    foreach (string columnName in _titles)
                    {
                        strbData.Append(columnName + ",");
                    }
                    strbData.Append("\n");
                    if (_fields == null)
                    {
                        foreach (DataRow dr in _dataSource.Rows)
                        {
                            for (int i = 0; i < _dataSource.Columns.Count; i++)
                            {
                                strbData.Append(dr[i].ToString() + ",");
                            }
                            strbData.Append("\n");
                        }
                        return strbData.ToString();
                    }
                    else
                    {
                        foreach (DataRow dr in _dataSource.Rows)
                        {
                            for (int i = 0; i < _fields.Length; i++)
                            {
                                strbData.Append(_fields[i].ToString() + ",");
                            }
                            strbData.Append("\n");
                        }
                        return strbData.ToString();
                    }
                }
            }
            #endregion

            #region 得到一个随意的文件名
            /// <summary>
            /// 得到一个随意的文件名
            /// </summary>
            /// <returns></returns>
            private string GetRandomFileName()
            {
                Random rnd = new Random((int)(DateTime.Now.Ticks));
                string s = rnd.Next(Int32.MaxValue).ToString();
                return DateTime.Now.ToShortDateString() + "_" + s + ".csv";
            }
            #endregion
        }

    }

  • 相关阅读:
    POJ2778 DNA Sequence AC自动机上dp
    codeforces732F Tourist Reform 边双联通分量
    codeforces786B Legacy 线段树优化建图
    洛谷P3588 PUS 线段树优化建图
    codeforces1301D Time to Run 模拟
    codeforces1303B National Project 二分或直接计算
    codeforces1303C Perfect Keyboard 模拟或判断欧拉路
    codeforces1303D Fill The Bag 二进制应用+贪心
    python之路——使用python操作mysql数据库
    python之路——mysql索引原理
  • 原文地址:https://www.cnblogs.com/zhangzt/p/1624818.html
Copyright © 2011-2022 走看看