413 lines
17 KiB
C#
413 lines
17 KiB
C#
using NPOI.HSSF.UserModel;
|
||
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;
|
||
|
||
namespace WaterCloud.Code
|
||
{
|
||
/// <summary>
|
||
/// List导出到Excel文件
|
||
/// </summary>
|
||
/// <typeparam name="T"></typeparam>
|
||
public class ExcelHelper<T> where T : new()
|
||
{
|
||
#region List导出到Excel文件
|
||
/// <summary>
|
||
/// List导出到Excel文件
|
||
/// </summary>
|
||
/// <param name="sFileName"></param>
|
||
/// <param name="sHeaderText"></param>
|
||
/// <param name="list"></param>
|
||
[Obsolete]
|
||
public string ExportToExcel(string sFileName, string sHeaderText, List<T> list, string[] columns)
|
||
{
|
||
sFileName = string.Format("{0}_{1}", Utils.GetGuid(), sFileName);
|
||
string sRoot = GlobalContext.HostingEnvironment.ContentRootPath;
|
||
string partDirectory = string.Format("Resource{0}Export{0}Excel", Path.DirectorySeparatorChar);
|
||
string sDirectory = Path.Combine(sRoot, partDirectory);
|
||
string sFilePath = Path.Combine(sDirectory, sFileName);
|
||
if (!Directory.Exists(sDirectory))
|
||
{
|
||
Directory.CreateDirectory(sDirectory);
|
||
}
|
||
using (MemoryStream ms = CreateExportMemoryStream(list, sHeaderText, columns))
|
||
{
|
||
using (FileStream fs = new FileStream(sFilePath, FileMode.Create, FileAccess.Write))
|
||
{
|
||
byte[] data = ms.ToArray();
|
||
fs.Write(data, 0, data.Length);
|
||
fs.Flush();
|
||
}
|
||
}
|
||
return partDirectory + Path.DirectorySeparatorChar + sFileName;
|
||
}
|
||
|
||
/// <summary>
|
||
/// List导出到Excel的MemoryStream
|
||
/// </summary>
|
||
/// <param name="list">数据源</param>
|
||
/// <param name="sHeaderText">表头文本</param>
|
||
/// <param name="columns">需要导出的属性</param>
|
||
[Obsolete]
|
||
private MemoryStream CreateExportMemoryStream(List<T> list, string sHeaderText, string[] columns)
|
||
{
|
||
HSSFWorkbook workbook = new HSSFWorkbook();
|
||
ISheet sheet = workbook.CreateSheet();
|
||
|
||
Type type = typeof(T);
|
||
PropertyInfo[] properties = ReflectionHelper.GetProperties(type, columns);
|
||
|
||
ICellStyle dateStyle = workbook.CreateCellStyle();
|
||
IDataFormat format = workbook.CreateDataFormat();
|
||
dateStyle.DataFormat = format.GetFormat("yyyy-MM-dd");
|
||
|
||
#region 取得每列的列宽（最大宽度）
|
||
int[] arrColWidth = new int[properties.Length];
|
||
for (int columnIndex = 0; columnIndex < properties.Length; columnIndex++)
|
||
{
|
||
//GBK对应的code page是CP936
|
||
arrColWidth[columnIndex] = properties[columnIndex].Name.Length;
|
||
}
|
||
#endregion
|
||
for (int rowIndex = 0; rowIndex < list.Count; rowIndex++)
|
||
{
|
||
#region 新建表，填充表头，填充列头，样式
|
||
if (rowIndex == 65535 || rowIndex == 0)
|
||
{
|
||
if (rowIndex != 0)
|
||
{
|
||
sheet = workbook.CreateSheet();
|
||
}
|
||
|
||
#region 表头及样式
|
||
{
|
||
IRow headerRow = sheet.CreateRow(0);
|
||
headerRow.HeightInPoints = 25;
|
||
headerRow.CreateCell(0).SetCellValue(sHeaderText);
|
||
|
||
ICellStyle headStyle = workbook.CreateCellStyle();
|
||
headStyle.Alignment = HorizontalAlignment.Center;
|
||
IFont font = workbook.CreateFont();
|
||
font.FontHeightInPoints = 20;
|
||
font.Boldweight = 700;
|
||
headStyle.SetFont(font);
|
||
|
||
headerRow.GetCell(0).CellStyle = headStyle;
|
||
|
||
sheet.AddMergedRegion(new CellRangeAddress(0, 0, 0, properties.Length - 1));
|
||
}
|
||
#endregion
|
||
|
||
#region 列头及样式
|
||
{
|
||
IRow headerRow = sheet.CreateRow(1);
|
||
ICellStyle headStyle = workbook.CreateCellStyle();
|
||
headStyle.Alignment = HorizontalAlignment.Center;
|
||
IFont font = workbook.CreateFont();
|
||
font.FontHeightInPoints = 10;
|
||
font.Boldweight = 700;
|
||
headStyle.SetFont(font);
|
||
|
||
for (int columnIndex = 0; columnIndex < properties.Length; columnIndex++)
|
||
{
|
||
// 类属性如果有Description就用Description当做列名
|
||
DescriptionAttribute customAttribute = (DescriptionAttribute)Attribute.GetCustomAttribute(properties[columnIndex], typeof(DescriptionAttribute));
|
||
string description = properties[columnIndex].Name;
|
||
if (customAttribute != null)
|
||
{
|
||
description = customAttribute.Description;
|
||
}
|
||
headerRow.CreateCell(columnIndex).SetCellValue(description);
|
||
headerRow.GetCell(columnIndex).CellStyle = headStyle;
|
||
|
||
//设置列宽
|
||
sheet.SetColumnWidth(columnIndex, (arrColWidth[columnIndex] + 1) * 256);
|
||
}
|
||
}
|
||
#endregion
|
||
}
|
||
#endregion
|
||
|
||
#region 填充内容
|
||
ICellStyle contentStyle = workbook.CreateCellStyle();
|
||
contentStyle.Alignment = HorizontalAlignment.Left;
|
||
IRow dataRow = sheet.CreateRow(rowIndex + 2); // 前面2行已被占用
|
||
for (int columnIndex = 0; columnIndex < properties.Length; columnIndex++)
|
||
{
|
||
ICell newCell = dataRow.CreateCell(columnIndex);
|
||
newCell.CellStyle = contentStyle;
|
||
|
||
string drValue = properties[columnIndex].GetValue(list[rowIndex], null).ParseToString();
|
||
switch (properties[columnIndex].PropertyType.ToString())
|
||
{
|
||
case "System.String":
|
||
newCell.SetCellValue(drValue);
|
||
break;
|
||
|
||
case "System.DateTime":
|
||
case "System.Nullable`1[System.DateTime]":
|
||
newCell.SetCellValue(drValue.ToDate());
|
||
newCell.CellStyle = dateStyle; //格式化显示
|
||
break;
|
||
|
||
case "System.Boolean":
|
||
case "System.Nullable`1[System.Boolean]":
|
||
newCell.SetCellValue(drValue.ToBool());
|
||
break;
|
||
|
||
case "System.Byte":
|
||
case "System.Nullable`1[System.Byte]":
|
||
case "System.Int16":
|
||
case "System.Nullable`1[System.Int16]":
|
||
case "System.Int32":
|
||
case "System.Nullable`1[System.Int32]":
|
||
newCell.SetCellValue(drValue.ToInt());
|
||
break;
|
||
|
||
case "System.Int64":
|
||
case "System.Nullable`1[System.Int64]":
|
||
newCell.SetCellValue(drValue.ParseToString());
|
||
break;
|
||
|
||
case "System.Double":
|
||
case "System.Nullable`1[System.Double]":
|
||
newCell.SetCellValue(drValue.ToDouble());
|
||
break;
|
||
|
||
case "System.Decimal":
|
||
case "System.Nullable`1[System.Decimal]":
|
||
newCell.SetCellValue(drValue.ToDouble());
|
||
break;
|
||
|
||
case "System.DBNull":
|
||
newCell.SetCellValue(string.Empty);
|
||
break;
|
||
|
||
default:
|
||
newCell.SetCellValue(string.Empty);
|
||
break;
|
||
}
|
||
}
|
||
#endregion
|
||
}
|
||
|
||
using (MemoryStream ms = new MemoryStream())
|
||
{
|
||
workbook.Write(ms);
|
||
ms.Flush();
|
||
ms.Position = 0;
|
||
return ms;
|
||
}
|
||
}
|
||
#endregion
|
||
|
||
#region Excel导入
|
||
/// <summary>
|
||
/// Excel导入
|
||
/// </summary>
|
||
/// <param name="filePath"></param>
|
||
/// <returns></returns>
|
||
public List<T> ImportFromExcel(string filePath)
|
||
{
|
||
List<T> list = new List<T>();
|
||
HSSFWorkbook hssfWorkbook = null;
|
||
XSSFWorkbook xssWorkbook = null;
|
||
ISheet sheet = null;
|
||
using (FileStream file = new FileStream(filePath, FileMode.Open, FileAccess.Read))
|
||
{
|
||
switch (Path.GetExtension(filePath))
|
||
{
|
||
case ".xls":
|
||
hssfWorkbook = new HSSFWorkbook(file);
|
||
sheet = hssfWorkbook.GetSheetAt(0);
|
||
break;
|
||
|
||
case ".xlsx":
|
||
xssWorkbook = new XSSFWorkbook(file);
|
||
sheet = xssWorkbook.GetSheetAt(0);
|
||
break;
|
||
|
||
default:
|
||
throw new Exception("不支持的文件格式");
|
||
}
|
||
}
|
||
IRow columnRow = sheet.GetRow(1); // 第二行为字段名
|
||
Dictionary<int, PropertyInfo> mapPropertyInfoDict = new Dictionary<int, PropertyInfo>();
|
||
for (int j = 0; j < columnRow.LastCellNum; j++)
|
||
{
|
||
ICell cell = columnRow.GetCell(j);
|
||
PropertyInfo propertyInfo = MapPropertyInfo(cell.ParseToString());
|
||
if (propertyInfo != null)
|
||
{
|
||
mapPropertyInfoDict.Add(j, propertyInfo);
|
||
}
|
||
}
|
||
|
||
for (int i = (sheet.FirstRowNum + 2); i <= sheet.LastRowNum; i++)
|
||
{
|
||
IRow row = sheet.GetRow(i);
|
||
T entity = new T();
|
||
for (int j = row.FirstCellNum; j < columnRow.LastCellNum; j++)
|
||
{
|
||
if (mapPropertyInfoDict.ContainsKey(j))
|
||
{
|
||
if (row.GetCell(j) != null)
|
||
{
|
||
PropertyInfo propertyInfo = mapPropertyInfoDict[j];
|
||
switch (propertyInfo.PropertyType.ToString())
|
||
{
|
||
case "System.DateTime":
|
||
case "System.Nullable`1[System.DateTime]":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString().ToDate());
|
||
break;
|
||
|
||
case "System.Boolean":
|
||
case "System.Nullable`1[System.Boolean]":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString().ToBool());
|
||
break;
|
||
|
||
case "System.Byte":
|
||
case "System.Nullable`1[System.Byte]":
|
||
mapPropertyInfoDict[j].SetValue(entity, Byte.Parse(row.GetCell(j).ParseToString()));
|
||
break;
|
||
case "System.Int16":
|
||
case "System.Nullable`1[System.Int16]":
|
||
mapPropertyInfoDict[j].SetValue(entity, Int16.Parse(row.GetCell(j).ParseToString()));
|
||
break;
|
||
case "System.Int32":
|
||
case "System.Nullable`1[System.Int32]":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString().ToInt());
|
||
break;
|
||
|
||
case "System.Int64":
|
||
case "System.Nullable`1[System.Int64]":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString().ToLong());
|
||
break;
|
||
|
||
case "System.Double":
|
||
case "System.Nullable`1[System.Double]":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString().ToDouble());
|
||
break;
|
||
|
||
case "System.Single":
|
||
case "System.Nullable`1[System.Single]":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString().ParseToFloat());
|
||
break;
|
||
|
||
case "System.Decimal":
|
||
case "System.Nullable`1[System.Decimal]":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString().ToDecimal());
|
||
break;
|
||
|
||
default:
|
||
case "System.String":
|
||
mapPropertyInfoDict[j].SetValue(entity, row.GetCell(j).ParseToString());
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
list.Add(entity);
|
||
}
|
||
return list;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 查找Excel列名对应的实体属性
|
||
/// </summary>
|
||
/// <param name="columnName"></param>
|
||
/// <returns></returns>
|
||
private PropertyInfo MapPropertyInfo(string columnName)
|
||
{
|
||
PropertyInfo[] propertyList = ReflectionHelper.GetProperties(typeof(T));
|
||
PropertyInfo propertyInfo = propertyList.Where(p => p.Name == columnName).FirstOrDefault();
|
||
if (propertyInfo != null)
|
||
{
|
||
return propertyInfo;
|
||
}
|
||
else
|
||
{
|
||
foreach (PropertyInfo tempPropertyInfo in propertyList)
|
||
{
|
||
DescriptionAttribute[] attributes = (DescriptionAttribute[])tempPropertyInfo.GetCustomAttributes(typeof(DescriptionAttribute), false);
|
||
if (attributes.Length > 0)
|
||
{
|
||
if (attributes[0].Description == columnName)
|
||
{
|
||
return tempPropertyInfo;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return null;
|
||
}
|
||
#endregion
|
||
|
||
#region excle转datatable
|
||
/// <summary>
|
||
/// 读取excel文件数据到DataTable
|
||
/// </summary>
|
||
/// <param name="filePath"></param>
|
||
/// <param name="deleteFile"></param>
|
||
/// <returns></returns>
|
||
public static DataTable ReadExcelToTable(string filePath, bool deleteFile = false)
|
||
{
|
||
var dataTable = new DataTable();
|
||
using (var tempFile = new FileStream(filePath, FileMode.Open))
|
||
{
|
||
IWorkbook workbook = null;
|
||
if (Path.GetExtension(filePath)==".xls")
|
||
{
|
||
workbook = new HSSFWorkbook(tempFile);
|
||
}
|
||
else
|
||
{
|
||
workbook = new XSSFWorkbook(tempFile);
|
||
}
|
||
|
||
var sheet = workbook.GetSheetAt(0);
|
||
var tableHeadRow = sheet.GetRow(0);
|
||
for (int i = 0; i < tableHeadRow.PhysicalNumberOfCells; i++)
|
||
{
|
||
if (tableHeadRow.Cells[i] != null)
|
||
{
|
||
tableHeadRow.Cells[i].SetCellType(CellType.String);
|
||
}
|
||
var headCell = tableHeadRow.Cells[i];
|
||
dataTable.Columns.Add(new DataColumn(headCell.StringCellValue));
|
||
}
|
||
for (int i = 1; i < sheet.PhysicalNumberOfRows; i++)
|
||
{
|
||
var row = sheet.GetRow(i);
|
||
var newRow = dataTable.NewRow();
|
||
for (int j = 0; j < row.PhysicalNumberOfCells; j++)
|
||
{
|
||
if (row.Cells[j] != null)
|
||
{
|
||
row.Cells[j].SetCellType(CellType.String);
|
||
}
|
||
var cell = row.Cells[j];
|
||
|
||
newRow[j] = cell.StringCellValue;
|
||
}
|
||
dataTable.Rows.Add(newRow);
|
||
}
|
||
workbook.Close();
|
||
}
|
||
if (deleteFile)
|
||
{
|
||
File.Delete(filePath);
|
||
}
|
||
return dataTable;
|
||
}
|
||
#endregion
|
||
}
|
||
}
|