如何通过python在.yaml文件中存在的现有数据之间插入数据？

Question

我正在尝试使用Python和YAML创建日志记录软件。 我需要测试多个板，每个板可能需要进行1个以上的试用。 因此，我想按板编号的升序排列YAML文件中的数据，以使分析过程自动化。

#!/usr/bin/python3

import yaml
import io
board_no = (input('Enter the board number: '))
trial_no = (input('Enter the trial no: '))
reading1 = int(input('Reading1: '))
status = input('Pass/Fail: ')
yaml.allow_duplicate_keys = True
basic_info = { 'Board ' + board_no:{
        'Trial ' + trial_no:{
            'Reading1 '  :  reading1,
            'Status ': status
           }
        }
   }
fp = "../yaml_101/test_list.yaml"
outfile = open (fp,'a')
yaml.dump(basic_info, outfile, indent = 4, default_flow_style = False)
outfile.close()

我已经编写了这个相当简单的代码来创建日志条目。 我一直在寻找在2个块之间插入数据的方法，但没有找到一个。 我当前的输出是（手动输入后）：

Board 1:
    Trial 1:
        'Reading1 ': 450
        'Status ': Pass
Board 2:
    Trial 1:
        'Reading1 ': 758
        'Status ': Fail
Board 3:
    Trial 1:
        'Reading1 ': 450
        'Status ': Pass
Board 2:
    Trial 2:
        'Reading1 ': 450
        'Status ': Pass

我想要的输出是：

Board 1:
    Trial 1:
        'Reading1 ': 450
        'Status ': Pass
Board 2:
    Trial 1:
        'Reading1 ': 758
        'Status ': Fail
Board 2:
    Trial 2:
        'Reading1 ': 450
        'Status ': Pass
Board 3:
    Trial 1:
        'Reading1 ': 450
        'Status ': Pass

有人可以指导我吗？ 我对YAML和Python完全陌生。

Answer 1

您的代码有几个问题：

1. 您打开了要追加的文件，但是不想追加，而是要插入。 当然可以，但是随后您需要知道文件指针的位置，在缓冲区中读取所有内容，然后编写新内容，然后编写缓冲区。 了解文件指针的位置将是困难的部分。

2. 您执行yaml.allow_duplicate_keys = True但是在PyYAML中不执行任何操作。 这是ruamel.yaml功能，它允许您使用重复的密钥读取文件（YAML规范不允许，但不合格的PyYAML库会静默忽略）。

3. 您不应直接操作YAML文件，而应始终使用从YAML加载数据，对其进行更新和转储。 这样，即使您的输入之一包含单引号或双引号或空格（如您从'Trial '所见，由于空格而需要在YAML中加引号），您的YAML也会保持一致。

4. 您只在YAML文件中使用映射，但是这样的试验列表更适合于序列。

5. 您的密钥具有枚举信息，这几乎总是一个坏主意，因为您无法预先知道密钥的外观。 因此，而不是一个关键Board 1 ，你应该有一个键Board与值1 ，或值与板为重点的数字映射。 这样，就很容易获取特定板号的数据（或以相同方式处理或更新试验）。

---

首先，您应该使status.yaml文件如下所示：

- Board: 1
  Trial: 1
  Reading1: 450
  Status: Pass
- Board: 2
  Trial: 1
  Reading1: 758
  Status: Fail
- Board: 3
  Trial: 1
  Reading1: 450
  Status: Pass

或使用：

Boards:
  - nr: 1
    Trials:
    - nr: 1
      Reading1: 450
      Status: Pass
  - nr: 2
    Trials:
    - nr: 1
      Reading1: 758
      Status: Fail
  - nr: 3
    Trials:
    - nr: 1
      Reading1: 450
      Status: Pass

或（我的最爱）：

Boards:
  1:
    Trials:
      1:
        Reading: 450
        Status: Pass
  2:
    Trials:
      1:
        Reading: 758
        Status: Fail
  3:
    Trials:
      1:
         Reading: 450
         Status: Pass

在第一个示例中，您应该遍历Python列表，并在Board的值更改为大于输入的Board编号之前插入一个新的试用版。

要更新最后一个示例，请使用ruamel.yaml （免责声明，我是该软件包的作者），因此映射中的键保持相同的顺序：

from pathlib import Path
import ruamel.yaml

status_file = Path('status.yaml')
yaml = ruamel.yaml.YAML()
# yaml.indent(mapping=2)
yaml.preserve_quotes = True

data = yaml.load(status_file)

board_no = 2
 # no need to input the trial_no, it should be the next number for this board
trials = data['Boards'][board_no]['Trials']
trial_no = sorted(trials.keys())[-1] + 1
reading1 = 450
status = 'Pass'

trials[trial_no] = dict(Reading1=reading1, Status=status)

yaml.dump(data, status_file)

它将status.yaml更新为：

Boards:
  1:
    Trials:
      1:
        Reading1: 450
        Status: Pass
  2:
    Trials:
      1:
        Reading1: 758
        Status: Fail
      2:
        Reading1: 450
        Status: Pass
  3:
    Trials:
      1:
        Reading1: 450
        Status: Pass