qri transform for storing mta elevator statuses

transform.star

load("encoding/csv.star", "csv")
load("encoding/json.star", "json")
load("http.star", "http")

def download(ctx):
  # send an http get request to the elevator outages page
  raw = http.get('https://traintime.lirr.org/api/ElevEsc?api_key=dwrxfpcsp42pcwxfbpd90zde5fjeb05pcdaqjw3j').body()
  data = json.loads(raw)

  stations = data['Stations']

  stationKeys = stations.keys()

  # shape the data into an array of arrays
  csvData = [['station', 'type', 'location', 'status' ]]

  for key in stationKeys:
    stationData = stations[key]
    if 'elevator' in stationData:
      for elevatorRecord in stationData['elevator']:
        csvData.append([key, 'elevator', elevatorRecord['LOC'], elevatorRecord['STAT']])
    if 'escalator' in stationData:
      for elevatorRecord in stationData['escalator']:
        csvData.append([key, 'escalator', elevatorRecord['LOC'], elevatorRecord['STAT']])

  return csvData
  # print('%s' % csvData)
  # print('%s' % stationKeys)


# set the body
def transform(ds, ctx):
  # ctx.download is whatever download() returned
  csvData = ctx.download
  # set the dataset body
  print('%s' % csvData)
  ds.set_body(csv.write_all(csvData,comma=","), parse_as='csv')