SiteServer-CMS/SiteServer.Utils/CsvUtils.cs

using System.Collections.Generic;
using System.Text;
using SiteServer.Utils.Enumerations;

namespace SiteServer.Utils
{
    public class CsvUtils
	{
        public static void Export(string filePath, List<string> head, List<List<string>> rows)
        {
            DirectoryUtils.CreateDirectoryIfNotExists(DirectoryUtils.GetDirectoryPath(filePath));
            FileUtils.DeleteFileIfExists(filePath);

            var builder = new StringBuilder();
            foreach (var name in head)
            {
                builder.Append(name).Append(",");
            }
            builder.Length -= 1;
            builder.Append("\n");

            foreach (var row in rows)
            {
                foreach (var r in row)
                {
                    var value = r.Replace(@"""", @"""""");
                    builder.Append(@"""" + value + @"""").Append(",");
                }
                builder.Length -= 1;
                builder.Append("\n");
            }

            FileUtils.WriteText(filePath, ECharset.gb2312, builder.ToString());
        }

        public static void Import(string filePath, out List<string> head, out List<List<string>> rows)
        {
            head = new List<string>();
            rows = new List<List<string>>();

            var valueList = new List<string>();

            //string content = FileUtils.ReadText(filePath, ECharset.utf_8);
            var content = FileUtils.ReadText(filePath, ECharset.gb2312);
            if (!string.IsNullOrEmpty(content))
            {
                valueList = TranslateUtils.StringCollectionToStringList(content, '\n');
            }

            if (valueList.Count > 1)
            {
                head = TranslateUtils.StringCollectionToStringList(valueList[0]);
                valueList = valueList.GetRange(1, valueList.Count - 1);
            }

            foreach (var str in valueList)
            {
                var row = new List<string>();

                var value = str.Replace(@"""""", @"""");
                var list = TranslateUtils.StringCollectionToStringList(value);

                if (list.Count != head.Count) continue;
                foreach (var r in list)
                {
                    row.Add(r.Trim('"'));
                }
                rows.Add(row);
            }
        }
	}
}