如何根据IF条件从另一个JSON文件创建JSON文件！Python3

Question

我有一个json文件，如下所示：

[
      {
            "id": "john",
            "title": "tech",
            "date": "2020/1/1",
            "shift": "DAYS"
      },
      {
            "id": "tom",
            "title": "tech",
            "date": "2021/5/5",
            "shift": "NIGHT"
      }
]

我需要根据这些信息重新创建另一个JSON文件。例如：

如果班次==夜间：

shiftHourStart=22:00:00

班次结束=06:00:00

所以输出应该是这样的：

[
      {
            "id": "john",
            "title": "tech",
            "shiftHourStart" = "22:00:00",
            "shiftHourEnd" = "06:00:00"
            "shift": "NIGHT"
      },
      {
            "id": "tom",
            "title": "tech",
            "date": "2021/5/5",
            "shiftHourStart" = "06:00:00",
            "shiftHourEnd" = "15:00:00",
            "shift": "DAY"          
      }
]

Answer 1

您可以使用dict.update

data = [
      {
            "id": "john",
            "title": "tech",
            "date": "2020/1/1",
            "shift": "DAYS"
      },
      {
            "id": "tom",
            "title": "tech",
            "date": "2021/5/5",
            "shift": "NIGHT"
      }
]

shift_mapping = {
    'DAYS': { # or DAY whatever you have
        "shiftHourStart": "22:00:00",
            "shiftHourEnd": "06:00:00"
    },
    'NIGHT':{
        "shiftHourStart": "06:00:00",
        "shiftHourEnd": "15:00:00",
    }
}

for each_shift in data:
    if 'shift' in each_shift:
        each_shift.update(shift_mapping[each_shift['shift']])
    else:
        print('Warning!! shift key does not exists!')
    
print(data)

[{'id': 'john',
  'title': 'tech',
  'date': '2020/1/1',
  'shift': 'DAYS',
  'shiftHourStart': '22:00:00',
  'shiftHourEnd': '06:00:00'},
 {'id': 'tom',
  'title': 'tech',
  'date': '2021/5/5',
  'shift': 'NIGHT',
  'shiftHourStart': '06:00:00',
  'shiftHourEnd': '15:00:00'}]