CSV解析-Swift 4

沙比尔·简

我正在尝试解析CSV，但出现了一些问题。以下是我用于解析CSV的代码：

let fileURL = Bundle.main.url(forResource: "test_application_data - Sheet 1", withExtension: "csv")
let content = try String(contentsOf: fileURL!, encoding: String.Encoding.utf8)
let parsedCSV: [[String]] = content.components(separatedBy: "\n").map{ $0.components(separatedBy: ",")}

这是我正在解析的CSV中的数据：

Item 9,Description 9,image url 
"Item 10 Extra line 1 Extra line 2 Extra line 3",Description 10,image url

因此，通过使用上面的代码，我对第一行的响应正确，即，Item 9但是我对以下内容的响应格式错误Item 10

如何正确解析两行？

OOPer

CSV的RFC：逗号分隔值（CSV）文件的通用格式和MIME类型（RFC-4180）

并非所有CSV数据或CSV处理器都符合此RFC的所有描述，但是通常，双引号中包含的字段可以包含：

换行符
逗号
转义的双引号（""表示单个双引号）

这段代码比RFC-4180稍微简化了一些，但是可以处理上述三种情况：

更新此旧代码不能很好地处理CRLF。（这是RFC-4180中的有效换行符。）我在底部添加了新代码，请检查一下。感谢杰伊。

import Foundation

let csvText = """
Item 9,Description 9,image url
"Item 10
Extra line 1
Extra line 2
Extra line 3",Description 10,image url
"Item 11
Csv item can contain ""double quote"" and comma(,)", Description 11 ,image url
"""

let pattern = "[ \r\t]*(?:\"((?:[^\"]|\"\")*)\"|([^,\"\\n]*))[ \t]*([,\\n]|$)"
let regex = try! NSRegularExpression(pattern: pattern)

var result: [[String]] = []
var record: [String] = []
let offset: Int = 0
regex.enumerateMatches(in: csvText, options: .anchored, range: NSRange(0..<csvText.utf16.count)) {match, flags, stop in
    guard let match = match else {fatalError()}
    if match.range(at: 1).location != NSNotFound {
        let field = csvText[Range(match.range(at: 1), in: csvText)!].replacingOccurrences(of: "\"\"", with: "\"")
        record.append(field)
    } else if match.range(at: 2).location != NSNotFound {
        let field = csvText[Range(match.range(at: 2), in: csvText)!].trimmingCharacters(in: .whitespaces)
        record.append(field)
    }
    let separator = csvText[Range(match.range(at: 3), in: csvText)!]
    switch separator {
    case "\n": //newline
        result.append(record)
        record = []
    case "": //end of text
        //Ignoring empty last line...
        if record.count > 1 || (record.count == 1 && !record[0].isEmpty) {
            result.append(record)
        }
        stop.pointee = true
    default: //comma
        break
    }
}
print(result)

（旨在在操场上进行测试。）

新代码，CRLF准备就绪。

import Foundation

let csvText =  "Field0,Field1\r\n"

let pattern = "[ \t]*(?:\"((?:[^\"]|\"\")*)\"|([^,\"\r\\n]*))[ \t]*(,|\r\\n?|\\n|$)"
let regex = try! NSRegularExpression(pattern: pattern)

var result: [[String]] = []
var record: [String] = []
let offset: Int = 0
regex.enumerateMatches(in: csvText, options: .anchored, range: NSRange(0..<csvText.utf16.count)) {match, flags, stop in
    guard let match = match else {fatalError()}
    if let quotedRange = Range(match.range(at: 1), in: csvText) {
        let field = csvText[quotedRange].replacingOccurrences(of: "\"\"", with: "\"")
        record.append(field)
    } else if let range = Range(match.range(at: 2), in: csvText) {
        let field = csvText[range].trimmingCharacters(in: .whitespaces)
        record.append(field)
    }
    let separator = csvText[Range(match.range(at: 3), in: csvText)!]
    switch separator {
    case "": //end of text
        //Ignoring empty last line...
        if record.count > 1 || (record.count == 1 && !record[0].isEmpty) {
            result.append(record)
        }
        stop.pointee = true
    case ",": //comma
        break
    default: //newline
        result.append(record)
        record = []
    }
}
print(result) //->[["Field0", "Field1"]]

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。