Powershell：导入-csv，重命名所有标题

Question

在我们公司，有许多用户和许多应用程序访问受限，并且数据库具有这些访问权限的证据。我无权访问该数据库，但我所拥有的是自动生成（每天一次）的 csv 文件，其中包含我所有用户的所有访问权限。我希望他们有机会检查他们的访问情况，所以我为此编写了一个简单的 powershell 脚本。

CSV：

user;database1_dat;database2_dat;database3_dat
john;0;0;1
peter;1;0;1

我能做到：

import-csv foo.csv | where {$_.user -eq $user}

但这会向我展示原始的丑陋头饰（带有“_dat”后缀）。当我无法预测明天会有多少标题时，我可以从每个以“_dat”结尾的标题中删除最后四个字符吗？

我知道计算属性如下：

Select-Object @{ expression={$_.database1_dat}; label='database1' }

但据我所知，我必须知道所有列名。

我是否被定罪为通过单独的函数“过度设计”它并从头开始动态构建整个“计算的属性表达式”，或者我是否缺少一种简单的方法？

谢谢:-)

Answer 1

假设文件foo.csv 适合整个内存，以下解决方案表现良好：

• 如果您需要内存限制（但总是慢得多）的解决方案，请参阅Santiago Squarzon's helpful answer 或底部的替代方法。

$headerRow, $dataRows = (Get-Content -Raw foo.csv) -split '\r?\n', 2

# You can pipe the result to `where {$_.user -eq $user}`
ConvertFrom-Csv ($headerRow -replace '_dat(?=;|$)'), $dataRows -Delimiter ';' 

• Get-Content -Raw 将整个文件读入内存，比逐行读取（默认）快得多。

• -split '\r?\n', 2 将生成的多行字符串分成两部分：标题行和所有剩余行。

  • 正则表达式 \r?\n 匹配换行符（CRLF (\r\n) 和仅 LF 换行符 (\n)）
  • , 2 将要返回的标记数量限制为2，这意味着一旦找到第一个标记（标题行），拆分就会停止，输入字符串的其余部分（包括所有数据行）按原样返回作为最后一个令牌。
  • 注意$null作为多重赋值中的第一个目标变量，它用于丢弃由字符串开头的分隔符正则表达式匹配产生的空标记。

• $headerRow -replace '_dat(?=;|$)'

  • -replace '_dat(?=;|$)' 使用正则表达式删除任何 _dat 列名后缀（后跟 ; 或字符串的结尾）；如果子字符串 _dat 仅作为名称出现 suffix （而不是 inside 名称），您可以简化为 -replace '_dat'

• ConvertFrom-Csv直接接受字符串的数组，所以清理后的标题行和包含所有数据行的字符串可以按原样传递。

---

替代解决方案：对象属性的算法重命名：

注意：此解决方案缓慢，但如果您仅从 CSV 文件中提取 少数 个对象，则可能是一种选择。

正如您在问题中所指出的，在您的情况下不能选择使用 Select-Object 和 calculated properties，因为您事先既不知道列名也不知道它们的编号。

但是，您可以使用ForEach-Object 命令，其中使用.psobject.Properties，intrinsic member，用于反射输入对象：

Import-Csv -Delimiter ';' foo.csv | where { $_.user -eq $user } | ForEach-Object {
  # Initialize an aux. ordered hashtable to store the renamed
  # property name-value pairs.
  $renamedProperties = [ordered] @{}
  # Process all properties of the input object and
  # add them with cleaned-up names to the hashtable.
  foreach ($prop in $_.psobject.Properties) {
    $renamedProperties[($prop.Name -replace '_dat(?=.|$)')] = $prop.Value
  }
  # Convert the aux. hashtable to a custom object and output it.
  [pscustomobject] $renamedProperties
}

Answer 2

你可以这样做：

$textInfo = (Get-Culture).TextInfo
$headers = (Get-Content .\test.csv | Select-Object -First 1).Split(';') |
ForEach-Object {
    $textInfo.ToTitleCase($_) -replace '_dat'
}

$user = 'peter'

Get-Content .\test.csv | Select-Object -Skip 1 |
ConvertFrom-Csv -Delimiter ';' -Header $headers |
Where-Object User -EQ $user

User  Database1 Database2 Database3
----  --------- --------- ---------
peter 1         0         1        

效率不是很高，但可以解决问题。