excel import and export

public class Excel
    {



        /// <summary>  
        /// Import excel to datatable  
        /// </summary>  
        /// <param name="filePath">excel路径</param>  
        /// <param name="isColumnName">Whether the first row is a column name</param>  
        /// <returns>返回datatable</returns>  
        public static DataTable ExcelToDataTable(string filePath, bool isColumnName, int type = 0, int whethertitle = 0)
        {
            DataTable dataTable = null;
            FileStream fs = null;
            DataColumn column = null;
            DataRow dataRow = null;
            IWorkbook workbook = null;
            ISheet sheet = null;
            IRow row = null;
            ICell cell = null;
            int startRow = 0;
            try
            {
                #region
                if (type == 0)
                {
                    using (fs = File.OpenRead(filePath))
                    {
                        // 2007 version  
                        if (filePath.IndexOf(".xlsx") > 0)
                            workbook = new XSSFWorkbook(fs);
                        // 2003 version  
                        else if (filePath.IndexOf(".xls") > 0)
                            workbook = new HSSFWorkbook(fs);

                        if (workbook != null)
                        {
                            sheet = workbook.GetSheetAt(0);//Read the first sheet, of course, you can also read each sheet in a loop  
                            dataTable = new DataTable();
                            if (sheet != null)
                            {
                                int rowCount = sheet.LastRowNum;//Total number of rows  
                                if (rowCount > 0)
                                {
                                    IRow firstRow = sheet.GetRow(0);//The first row  
                                    if (whethertitle > 0)
                                        firstRow = sheet.GetRow(1);//The second row
                                    int cellCount = firstRow.LastCellNum;//列数  

                                    // Build the columns of the datatable  
                                    if (isColumnName)
                                    {
                                        startRow = 1;//If the first row is a column name, start reading from the second row  
                                        for (int i = firstRow.FirstCellNum; i < cellCount; ++i)
                                        {
                                            cell = firstRow.GetCell(i);
                                            cell.SetCellType (CellType.String);
                                            if (cell != null)
                                            {
                                                if (cell.StringCellValue != null)
                                                {
                                                    column = new DataColumn(cell.StringCellValue);
                                                    dataTable.Columns.Add(column);
                                                }
                                            }
                                        }
                                    }
                                    else
                                    {
                                        for (int i = firstRow.FirstCellNum; i < cellCount; ++i)
                                        {
                                            column = new DataColumn("column" + (i + 1));
                                            dataTable.Columns.Add(column);
                                        }
                                    }
                                    if (whethertitle > 0)
                                        startRow = 2;
                                    // fill row  
                                    for (int i = startRow; i <= rowCount; ++i)
                                    {
                                        row = sheet.GetRow(i);
                                        if (row == null) continue;

                                        dataRow = dataTable.NewRow();
                                        bool flag = false;
                                        if (row.FirstCellNum >= 0)
                                        {
                                            for (int j = row.FirstCellNum; j < cellCount; ++j)
                                            {
                                                cell = row.GetCell(j);
                                                if (cell == null)
                                                    dataRow[j] = "";
                                                else
                                                {
                                                    switch (cell.CellType)
                                                    {
                                                        case CellType.Blank:
                                                            dataRow[j] = "";
                                                            break;
                                                        case CellType.Numeric:
                                                            short format = cell.CellStyle.DataFormat;
                                                            //Processing of time format (2015.12.5, 2015/12/5, 2015-12-5, etc.)  
                                                            if (format == 14 || format == 31 || format == 57 || format == 58)
                                                                dataRow[j] = cell.DateCellValue;
                                                            else
                                                                dataRow[j] = cell.NumericCellValue;
                                                            flag = true;
                                                            break;
                                                        case CellType.String:
                                                            dataRow[j] = cell.StringCellValue;
                                                            flag = true;
                                                            break;
                                                        default:
                                                            dataRow[j] = row.GetCell(j).StringCellValue;
                                                            break;
                                                    }
                                                }
                                            }
                                        }
                                        if (flag)
                                            dataTable.Rows.Add(dataRow);
                                    }
                                }
                            }
                        }
                    }
                }
                #endregion

                #region
                else
                {
                    using (fs = File.OpenRead(filePath))
                    {
                        // 2007 version  
                        if (filePath.IndexOf(".xlsx") > 0)
                            workbook = new XSSFWorkbook(fs);
                        // 2003 version  
                        else if (filePath.IndexOf(".xls") > 0)
                            workbook = new HSSFWorkbook(fs);

                        if (workbook != null)
                        {
                            sheet = workbook.GetSheetAt(0);//Read the first sheet, of course, you can also read each sheet in a loop  
                            dataTable = new DataTable();
                            if (sheet != null)
                            {
                                int rowCount = sheet.LastRowNum;//Total number of rows  
                                if (rowCount > 0)
                                {
                                    IRow firstRow = sheet.GetRow(1);//The first row  
                                    if (whethertitle > 0)
                                        firstRow = sheet.GetRow(1);//The second row
                                    int cellCount = firstRow.LastCellNum;//列数  

                                    // Build the columns of the datatable  
                                    if (isColumnName)
                                    {
                                        startRow = 1;//If the first row is a column name, start reading from the second row  
                                        for (int i = firstRow.FirstCellNum; i < cellCount; ++i)
                                        {
                                            cell = firstRow.GetCell(i);
                                            if (cell != null)
                                            {
                                                if (cell.StringCellValue != null)
                                                {
                                                    column = new DataColumn(cell.StringCellValue);
                                                    dataTable.Columns.Add(column);
                                                }
                                            }
                                        }
                                    }
                                    else
                                    {
                                        for (int i = firstRow.FirstCellNum; i < cellCount; ++i)
                                        {
                                            column = new DataColumn("column" + (i + 1));
                                            dataTable.Columns.Add(column);
                                        }
                                    }
                                    startRow = 2;
                                    if (whethertitle > 0)
                                        startRow = 3;
                                    // fill row  
                                    for (int i = startRow; i <= rowCount; ++i)
                                    {
                                        row = sheet.GetRow(i);
                                        if (row == null) continue;
                                        bool flag = false;
                                        dataRow = dataTable.NewRow();
                                        for (int j = row.FirstCellNum; j < cellCount; ++j)
                                        {
                                            cell = row.GetCell(j);
                                            if (cell == null)
                                                dataRow[j] = "";
                                            else
                                            {
                                                //CellType(Unknown = -1,Numeric = 0,String = 1,Formula = 2,Blank = 3,Boolean = 4,Error = 5,)  
                                                switch (cell.CellType)
                                                {
                                                    case CellType.Blank:
                                                        dataRow[j] = "";
                                                        break;
                                                    case CellType.Numeric:
                                                        short format = cell.CellStyle.DataFormat;
                                                        //Processing of time format (2015.12.5, 2015/12/5, 2015-12-5, etc.)  
                                                        if (format == 14 || format == 31 || format == 57 || format == 58)
                                                            dataRow[j] = cell.DateCellValue;
                                                        else
                                                            dataRow[j] = cell.NumericCellValue;
                                                        flag = true;
                                                        break;
                                                    case CellType.String:
                                                        dataRow[j] = cell.StringCellValue;
                                                        flag = true;
                                                        break;
                                                }
                                            }
                                        }
                                        if (flag)
                                            dataTable.Rows.Add(dataRow);
                                    }
                                }
                            }
                        }
                    }

                }
                #endregion

                return dataTable;
            }
            catch (Exception ex)
            {
                if (fs != null)
                {
                    fs.Close();
                }
                return null;
            }
        }

        private string fileName = null; //file name
        private IWorkbook workbook = null;
        private FileStream fs = null;
        private bool disposed;
        public Excel(string filenames)
        {
            this.fileName = filenames;
            disposed = false;
        }


        /// <summary>
        /// Import datatable into excel
        /// </summary>
        /// <param name="dt">DataTable query data</param>
        /// <param name="sheetName">表名</param>
        /// <param name="isColumnName">Add table name</param>
        /// <returns></returns>
        public int DataTableToExcel(DataTable dt, string sheetName, bool isColumnName)
        {
           // NPOI.HSSF.UserModel.HSSFWorkbook workbook = new HSSFWorkbook();
            fs = new FileStream(fileName, FileMode.OpenOrCreate, FileAccess.ReadWrite);
            int count = 0; // number of rows
            ISheet sheet = null;


            /*================================================================================
            if (fileName.IndexOf(".xlsx") > 0)
            {
                workbook = new XSSFWorkbook();
            }
            else
            {
                workbook = new HSSFWorkbook();
            }



            try
            {
                if (workbook != null)
                {
                    //Create a table and give the Excel table name
                    sheet = workbook.CreateSheet(sheetName);
                }
                else
                {
                    return -1;
                }
                if (isColumnName == true) //true add column name false do not add
                {
                    //create the first row
                    IRow row = sheet.CreateRow(0);

                    // loop to fill the first row
                    for (int i = 0; i < dt.Columns.Count; ++i)
                    {
                        row.CreateCell(i).SetCellValue(dt.Columns[i].ColumnName);
                    }
                    count = 1;
                }

                //fill the data to remove the first row
                for (int i = 0; i < dt.Rows.Count; ++i)
                {
                    //create a row
                    IRow row = sheet.CreateRow(count);

                    // populate the data of the columns of the created row
                    for (int j = 0; j < dt.Columns.Count; ++j)
                    {
                        row.CreateCell(j).SetCellValue(dt.Rows[i][j].ToString());
                    }

                    count++;
                }

              
                workbook.Write(fs);


                workbook.Close();
                return count;


            }
            catch (Exception ex)
            {

                return -1;
            }




        }






        public void Dispose()
        {
            Dispose(true);
            GC.SuppressFinalize(this);
        }

        protected virtual void Dispose(bool disposing)
        {
            if (!this.disposed)
            {
                if (disposing)
                {
                    if (fs != null)
                        fs.Close();
                }

                fs = null;
                disposed = true;
            }
        }

    }

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=326114439&siteId=291194637