ECS / ECS_EquipmentMangaSystem_BingChengTaiGuang · Files

using HHECS.BllModel;
using HHECS.Infrastructure.CommonHelper;
using NPOI.HSSF.UserModel;
using NPOI.SS.Formula.Eval;
using NPOI.SS.UserModel;
using NPOI.SS.Util;
using NPOI.XSSF.UserModel;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;

namespace HHECS.Infrastructure.Excel
{
    /// <summary>
    /// 用于Excel的导入和导出,注意进行异常捕获
    /// </summary>
    public class NPIOHelper
    {
        #region 从datatable中将数据导出到excel

        /// <summary>
        /// DataTable导出到Excel的MemoryStream,xls
        /// </summary>
        /// <param name="dtSource">源DataTable</param>
        /// <param name="strHeaderText">表头文本,如果为空则没有表头</param>
        static void ExportDT(DataTable dtSource, string strHeaderText, IWorkbook workbook, ISheet sheet, IDataFormat format, FileStream fs)
        {

            #region 右击文件 属性信息

            //{
            //    DocumentSummaryInformation dsi = PropertySetFactory.CreateDocumentSummaryInformation();
            //    dsi.Company = "http://www.yongfa365.com/";
            //    workbook.DocumentSummaryInformation = dsi;

            //    SummaryInformation si = PropertySetFactory.CreateSummaryInformation();
            //    si.Author = "柳永法"; //填加xls文件作者信息
            //    si.ApplicationName = "NPOI测试程序"; //填加xls文件创建程序信息
            //    si.LastAuthor = "柳永法2"; //填加xls文件最后保存者信息
            //    si.Comments = "说明信息"; //填加xls文件作者信息
            //    si.Title = "NPOI测试"; //填加xls文件标题信息
            //    si.Subject = "NPOI测试Demo"; //填加文件主题信息
            //    si.CreateDateTime = DateTime.Now;
            //    workbook.SummaryInformation = si;
            //}

            #endregion

            //dateStyle.DataFormat = format.GetFormat("yyyy-mm-dd HH:mm:ss");

            //取得列宽
            int[] arrColWidth = new int[dtSource.Columns.Count];
            foreach (DataColumn item in dtSource.Columns)
            {
                //添加5的列宽 解决字体产生太挤的效果
                arrColWidth[item.Ordinal] = Encoding.UTF8.GetBytes(item.ColumnName.ToString()).Length + 5;
                if (item.ColumnName == "创建时间" || item.ColumnName == "更新时间")
                {
                    //添加列名的宽度 解决出现### 
                    arrColWidth[item.Ordinal] = Encoding.UTF8.GetBytes(item.ColumnName.ToString()).Length + 10;
                }
            }
            for (int i = 0; i < dtSource.Rows.Count; i++)
            {
                for (int j = 0; j < dtSource.Columns.Count; j++)
                {
                    int intTemp = Encoding.UTF8.GetBytes(dtSource.Rows[i][j].ToString()).Length;
                    if (intTemp > arrColWidth[j])
                    {
                        arrColWidth[j] = intTemp;
                    }
                }
            }
            int rowIndex = 0;
            var ColumnsStyle = workbook.CreateCellStyle();
            foreach (DataRow row in dtSource.Rows)
            {
                #region 新建表，填充表头，填充列头，样式

                if (rowIndex == 65535 || rowIndex == 0)
                {
                    if (rowIndex != 0)
                    {
                        sheet = workbook.CreateSheet();
                    }
                    int tempIndex = 0;
                    #region 表头及样式
                    if (!string.IsNullOrWhiteSpace(strHeaderText))
                    {
                        var headerRow = sheet.CreateRow(tempIndex);
                        headerRow.HeightInPoints = 25;
                        headerRow.CreateCell(0).SetCellValue(strHeaderText);

                        var headStyle = workbook.CreateCellStyle();
                        headStyle.Alignment = HorizontalAlignment.Center;
                        var font = workbook.CreateFont();
                        font.FontHeightInPoints = 20;
                        font.IsBold = true;
                        headStyle.SetFont(font);
                        headerRow.GetCell(0).CellStyle = headStyle;
                        sheet.AddMergedRegion(new CellRangeAddress(0, 0, 0, dtSource.Columns.Count - 1));
                        //headerRow.Dispose();
                        tempIndex++;
                    }

                    #endregion


                    #region 列头及样式

                    {
                        var headerRow = sheet.CreateRow(tempIndex);
                        var headStyle = workbook.CreateCellStyle();
                        headStyle.Alignment = HorizontalAlignment.Center;
                        var font = workbook.CreateFont();
                        font.FontHeightInPoints = 12;
                        font.IsBold = true;
                        headStyle.SetFont(font);
                        foreach (DataColumn column in dtSource.Columns)
                        {
                            headerRow.CreateCell(column.Ordinal).SetCellValue(column.ColumnName);
                            headerRow.GetCell(column.Ordinal).CellStyle = headStyle;
                            //设置列宽
                            if (arrColWidth[column.Ordinal] > 255)
                            {
                                arrColWidth[column.Ordinal] = 254;
                            }
                            else
                            {
                                sheet.SetColumnWidth(column.Ordinal, (arrColWidth[column.Ordinal] + 1) * 256);
                            }
                        }
                        //headerRow.Dispose();
                        tempIndex++;
                    }

                    #endregion

                    rowIndex = tempIndex;
                }

                #endregion

                #region 填充内容
                var DataRow = sheet.CreateRow(rowIndex);
                //var ColumnsStyle = workbook.CreateCellStyle();
                ColumnsStyle.Alignment = HorizontalAlignment.Center;
                foreach (DataColumn column in dtSource.Columns)
                {
                    var newCell = DataRow.CreateCell(column.Ordinal);
                    string drValue = row[column].ToString();
                    newCell.CellStyle = ColumnsStyle;
                    switch (column.DataType.ToString())
                    {
                        case "System.String": //字符串类型
                            double result;
                            if (IsNumeric(drValue, out result))
                            {
                                double.TryParse(drValue, out result);
                                newCell.SetCellValue(result);

                                break;
                            }
                            else
                            {
                                newCell.SetCellValue(drValue);
                                break;
                            }

                        case "System.DateTime": //日期类型
                            DateTime dateV;
                            DateTime.TryParse(drValue, out dateV);
                            if (drValue != "")
                            {
                                newCell.SetCellValue(dateV.ToString("yyyy-MM-dd HH:mm:ss"));
                            }
                            else
                            {
                                newCell.SetCellValue("");
                            }
                            break;
                        case "System.Boolean": //布尔型
                            bool boolV = false;
                            bool.TryParse(drValue, out boolV);
                            newCell.SetCellValue(boolV);
                            break;
                        case "System.Int16": //整型
                        case "System.Int32":
                        case "System.Int64":
                        case "System.Byte":
                            int intV = 0;
                            int.TryParse(drValue, out intV);
                            newCell.SetCellValue(intV);
                            break;
                        case "System.Decimal": //浮点型
                        case "System.Double":
                            double doubV = 0;
                            double.TryParse(drValue, out doubV);
                            newCell.SetCellValue(doubV);
                            break;
                        case "System.DBNull": //空值处理
                            newCell.SetCellValue("");
                            break;
                        default:
                            newCell.SetCellValue("");
                            break;
                    }

                }
                #endregion
                rowIndex++;
            }

            workbook.Write(fs);
            workbook.Close();
            fs.Close();
        }

        /// <summary>
        /// DataTable导出到Excel文件
        /// </summary>
        /// <param name="dtSource">源DataTable</param>
        /// <param name="strHeaderText">表头文本</param>
        /// <param name="strFileName">保存位置</param>
        /// <param name="Path">文件路径</param>
        public static Task<BllResult> ExportDTtoExcelAsync<T>(List<T> datas, string strHeaderText, string strFileName, string Path)
        {
            return Task.Run(() =>
            {
                try
                {
                    //是否有路径文件
                    ExistsPath(Path);
                    string[] temp = strFileName.Split('.');
                    IWorkbook workbook;
                    var dtSource = DtToList.ListToDt(datas);
                    if (temp[^1].Trim() == "xls" && dtSource.Columns.Count < 256 && dtSource.Rows.Count < 65536)
                    {
                        workbook = new HSSFWorkbook();
                    }
                    else
                    {
                        if (temp[^1] == "xls")
                            strFileName += "x";
                        workbook = new XSSFWorkbook();
                    }

                    var sheet = workbook.CreateSheet();
                    var dateStyle = workbook.CreateCellStyle();
                    var format = workbook.CreateDataFormat();
                    using FileStream fs = new FileStream(strFileName, FileMode.Create, FileAccess.ReadWrite);
                    ExportDT(dtSource, strHeaderText, workbook, sheet, format, fs);
                    return BllResultFactory.Success();
                }
                catch (Exception ex)
                {
                    return BllResultFactory.Error(ex.Message);
                }
            });
        }

        #endregion

        #region Excel到DT

        /// <summary>
        /// 将制定sheet中的数据导入到datatable中
        /// </summary>
        /// <param name="sheet">需要导出的sheet</param>
        /// <param name="headerRowIndex">列头所在行号，-1表示没有列头，会忽略needHeader参数，填充1到n为列头</param>
        /// <param name="needHeader">表示是否需要表头，如果headerRowIndex不为-1，则needHeader为false，填充1到n为列头；如果为true，则取headerRowIndex所在行的cell值为列头，要求此列名不能重复，否则用数字代替</param>
        /// <returns></returns>
        static DataTable ImportDt(ISheet sheet, int headerRowIndex, bool needHeader)
        {
            DataTable table = new DataTable();
            IRow headerRow;
            int cellCount;
            if (headerRowIndex < 0 || !needHeader)
            {
                headerRow = sheet.GetRow(0);
                cellCount = headerRow.LastCellNum;

                for (int i = headerRow.FirstCellNum; i <= cellCount; i++)
                {
                    DataColumn column = new DataColumn(Convert.ToString(i));
                    table.Columns.Add(column);
                }
            }
            else
            {
                headerRow = sheet.GetRow(headerRowIndex);
                cellCount = headerRow.LastCellNum;

                for (int i = headerRow.FirstCellNum; i < cellCount; i++)
                {
                    if (headerRow.GetCell(i) == null)
                    {
                        if (table.Columns.IndexOf(Convert.ToString(i)) > 0)
                        {
                            DataColumn column = new DataColumn(Convert.ToString("重复列名" + i));
                            table.Columns.Add(column);
                        }
                        else
                        {
                            DataColumn column = new DataColumn(Convert.ToString(i));
                            table.Columns.Add(column);
                        }

                    }
                    else if (table.Columns.IndexOf(headerRow.GetCell(i).ToString()) > 0)
                    {
                        DataColumn column = new DataColumn(Convert.ToString("重复列名" + i));
                        table.Columns.Add(column);
                    }
                    else
                    {
                        DataColumn column = new DataColumn(headerRow.GetCell(i).ToString());
                        table.Columns.Add(column);
                    }
                }
            }
            for (int i = headerRowIndex + 1; i <= sheet.LastRowNum; i++)
            {
                IRow row;
                if (sheet.GetRow(i) == null)
                {
                    row = sheet.CreateRow(i);
                }
                else
                {
                    row = sheet.GetRow(i);
                }

                DataRow dataRow = table.NewRow();

                for (int j = row.FirstCellNum; j <= cellCount; j++)
                {

                    if (row.GetCell(j) != null)
                    {
                        switch (row.GetCell(j).CellType)
                        {
                            case CellType.String:
                                string str = row.GetCell(j).StringCellValue;
                                if (str != null && str.Length > 0)
                                {
                                    dataRow[j] = str.ToString();
                                }
                                else
                                {
                                    dataRow[j] = null;
                                }
                                break;
                            case CellType.Numeric:
                                if (DateUtil.IsCellDateFormatted(row.GetCell(j)))
                                {
                                    dataRow[j] = DateTime.FromOADate(row.GetCell(j).NumericCellValue);
                                }
                                else
                                {
                                    dataRow[j] = Convert.ToDouble(row.GetCell(j).NumericCellValue);
                                }
                                break;
                            case CellType.Boolean:
                                dataRow[j] = Convert.ToString(row.GetCell(j).BooleanCellValue);
                                break;
                            case CellType.Error:
                                dataRow[j] = ErrorEval.GetText(row.GetCell(j).ErrorCellValue);
                                break;
                            case CellType.Formula:
                                switch (row.GetCell(j).CachedFormulaResultType)
                                {
                                    case CellType.String:
                                        string strFORMULA = row.GetCell(j).StringCellValue;
                                        if (strFORMULA != null && strFORMULA.Length > 0)
                                        {
                                            dataRow[j] = strFORMULA.ToString();
                                        }
                                        else
                                        {
                                            dataRow[j] = null;
                                        }
                                        break;
                                    case CellType.Numeric:
                                        dataRow[j] = Convert.ToString(row.GetCell(j).NumericCellValue);
                                        break;
                                    case CellType.Boolean:
                                        dataRow[j] = Convert.ToString(row.GetCell(j).BooleanCellValue);
                                        break;
                                    case CellType.Error:
                                        dataRow[j] = ErrorEval.GetText(row.GetCell(j).ErrorCellValue);
                                        break;
                                    default:
                                        dataRow[j] = "";
                                        break;
                                }
                                break;
                            default:
                                dataRow[j] = "";
                                break;
                        }
                    }

                }
                table.Rows.Add(dataRow);

            }

            return table;
        }

        /// <summary>
        /// 读取excel 默认第一行为标头
        /// </summary>
        /// <param name="strFileName">excel文档路径</param>
        /// <returns></returns>
        public static Task<DataTable> ImportExceltoDtAsync(string strFileName)
        {
            return Task.Run(() =>
            {
                DataTable dt;
                IWorkbook wb;
                using (FileStream file = new FileStream(strFileName, FileMode.Open, FileAccess.Read))
                {
                    wb = WorkbookFactory.Create(file);
                }
                ISheet sheet = wb.GetSheetAt(0);
                dt = ImportDt(sheet, 0, true);
                return dt;
            });

        }

        /// <summary>
        /// 读取Excel流到DataSet
        /// </summary>
        /// <param name="stream">Excel流</param>
        /// <returns>Excel中的数据</returns>
        public static Task<DataSet> ImportExceltoDsAsync(string strFileName)
        {
            return Task.Run(() =>
            {
                DataSet ds = new DataSet();
                using (FileStream file = new FileStream(strFileName, FileMode.Open, FileAccess.Read))
                {
                    IWorkbook wb;
                    wb = WorkbookFactory.Create(file);
                    for (int i = 0; i < wb.NumberOfSheets; i++)
                    {
                        DataTable dt = new DataTable();
                        ISheet sheet = wb.GetSheetAt(i);
                        dt = ImportDt(sheet, 0, true);
                        ds.Tables.Add(dt);
                    }
                    return ds;
                }
            });
        }

        /// <summary>
        /// 读取Excel流到DataSet
        /// </summary>
        /// <param name="stream">Excel流</param>
        /// <param name="dict">字典参数，key：sheet名，value：列头所在行号，-1表示没有列头</param>
        /// <returns>Excel中的数据</returns>
        public static Task<DataSet> ImportExceltoDsAsync(string strFileName, Dictionary<string, int> dict)
        {
            return Task.Run(() =>
            {
                DataSet ds = new DataSet();
                IWorkbook wb;
                using (FileStream file = new FileStream(strFileName, FileMode.Open, FileAccess.Read))
                {
                    wb = WorkbookFactory.Create(file);
                    foreach (string key in dict.Keys)
                    {
                        DataTable dt = new DataTable();
                        ISheet sheet = wb.GetSheet(key);
                        dt = ImportDt(sheet, dict[key], true);
                        ds.Tables.Add(dt);
                    }
                    return ds;
                }
            });
        }

        /// <summary>
        /// 读取excel
        /// </summary>
        /// <param name="strFileName">excel文件路径</param>
        /// <param name="sheet">需要导出的sheet</param>
        /// <param name="HeaderRowIndex">列头所在行号，-1表示没有列头</param>
        /// <returns></returns>
        public static Task<DataTable> ImportExceltoDtAsync(string strFileName, string SheetName, int HeaderRowIndex, bool needHeader)
        {
            return Task.Run(() =>
            {
                IWorkbook wb;
                using (FileStream file = new FileStream(strFileName, FileMode.Open, FileAccess.Read))
                {
                    wb = WorkbookFactory.Create(file);
                }
                ISheet sheet = wb.GetSheet(SheetName);
                return ImportDt(sheet, HeaderRowIndex, needHeader);
            });

        }

        /// <summary>
        /// 读取excel
        /// </summary>
        /// <param name="strFileName">excel文件路径</param>
        /// <param name="sheet">需要导出的sheet序号</param>
        /// <param name="HeaderRowIndex">列头所在行号，-1表示没有列头</param>
        /// <returns></returns>
        public static Task<DataTable> ImportExceltoDtAsync(string strFileName, int SheetIndex, int HeaderRowIndex, bool needHeader)
        {
            return Task.Run(() =>
            {
                IWorkbook wb;
                using (FileStream file = new FileStream(strFileName, FileMode.Open, FileAccess.Read))
                {
                    wb = WorkbookFactory.Create(file);
                }
                ISheet sheet = wb.GetSheetAt(SheetIndex);
                return ImportDt(sheet, HeaderRowIndex, needHeader);
            });

        }

        #endregion

        private static bool IsNumeric(string message, out double result)
        {
            Regex rex = new Regex(@"^[-]?\d+[.]?\d*$");
            result = -1;
            if (rex.IsMatch(message))
            {
                result = double.Parse(message);
                return true;
            }
            else
                return false;

        }
        public static BllResult ExistsPath(string Path)
        {
            try
            {
                if (!Directory.Exists(Path))
                {
                    Directory.CreateDirectory(Path);
                }
                return BllResultFactory.Success();
            }
            catch (Exception ex)
            {
                return BllResultFactory.Error(ex.Message);
            }
        }

        public static IWorkbook GetWorkbook<T>(List<T> data) where T : class
        {
            IWorkbook workbook = new XSSFWorkbook();
            ISheet sheet = workbook.CreateSheet();
            var rowNum = 0;
            foreach (var item in data)
            {
                var props = typeof(T).GetProperties().ToList();
                if (rowNum == 0)
                {
                    var headRow = sheet.CreateRow(rowNum);
                    foreach (var prop in props)
                    {
                        var i = props.IndexOf(prop);
                        var attr = prop.GetCustomAttribute<DescriptionAttribute>(true);
                        var cell = headRow.CreateCell(i);
                        cell.SetCellValue(attr == null ? prop.Name : attr.Description);
                    }
                    rowNum++;
                }

                var row = sheet.CreateRow(rowNum);
                foreach (var prop in props)
                {
                    row.CreateCell(props.IndexOf(prop)).SetCellValue($"{prop.GetValue(item)}");
                }
                rowNum++;
            }

            return workbook;
        }
    }
}