96 lines
2.6 KiB
JavaScript
96 lines
2.6 KiB
JavaScript
import mkdirp from 'mkdirp'
|
|
import path from 'path'
|
|
import fsp from 'fs/promises'
|
|
import XLSX from 'xlsx'
|
|
import { getCounty } from '../../util.js'
|
|
|
|
async function processByCounty({ date, xlsx }, countyInfo, output) {
|
|
output = {
|
|
directory: path.join(output.directory, 'by-county'),
|
|
file: county => `${county}.json`
|
|
}
|
|
|
|
const rows = XLSX.utils.sheet_to_json(xlsx.Sheets.VAX_ADMIN_BY_DAY_COUNTY)
|
|
|
|
const counties = rows.reduce((counties, row) => {
|
|
const county = getCounty(row.COUNTY_NAME)
|
|
const info = countyInfo[county]
|
|
if (!county) return counties
|
|
if (!counties[county]) counties[county] = []
|
|
counties[county].push({
|
|
report_date: row.ADMIN_DATE,
|
|
population: info.population,
|
|
vaccinations: row.VAXADMIN,
|
|
total_vaccinations: row.CUMVAXADMIN,
|
|
})
|
|
return counties
|
|
}, {})
|
|
|
|
for (const county in counties) {
|
|
const data = {
|
|
segment: { county },
|
|
headers: Object.keys(counties[county][0]),
|
|
rows: counties[county].map(row => Object.values(row))
|
|
}
|
|
|
|
await mkdirp(output.directory)
|
|
await fsp.writeFile(path.join(output.directory, output.file(county)), JSON.stringify(data))
|
|
}
|
|
}
|
|
|
|
async function processCombined({ date, xlsx }, countyInfo, output) {
|
|
output = {
|
|
directory: output.directory,
|
|
file: `combined.json`
|
|
}
|
|
|
|
const rows = XLSX.utils.sheet_to_json(xlsx.Sheets.VAX_ADMIN_BY_DAY_COUNTY)
|
|
|
|
const report_dates = new Set()
|
|
|
|
const results = rows.map(row => {
|
|
const county = getCounty(row.COUNTY_NAME)
|
|
if (!county) return null
|
|
if (['GEORGIA'].includes(county)) return null
|
|
|
|
const info = countyInfo[county]
|
|
report_dates.add(row['report_date'])
|
|
return {
|
|
county,
|
|
report_date: row.ADMIN_DATE,
|
|
population: info.population,
|
|
vaccinations: row.VAXADMIN,
|
|
total_vaccinations: row.CUMVAXADMIN,
|
|
}
|
|
}).filter(row => !!row)
|
|
|
|
const data = {
|
|
segment: { report_date: Array.from(report_dates) },
|
|
headers: Object.keys(results[0]),
|
|
rows: results.map(row => Object.values(row))
|
|
}
|
|
|
|
await mkdirp(output.directory)
|
|
await fsp.writeFile(path.join(output.directory, output.file), JSON.stringify(data))
|
|
}
|
|
|
|
|
|
async function processSingleWorkbook (xlsx, countyInfo) {
|
|
const output = {
|
|
directory: `./public/data/overall/vaccinations/`,
|
|
}
|
|
|
|
try {
|
|
await fsp.rm(output.directory, { recursive: true })
|
|
} catch (e) {}
|
|
|
|
await processByCounty(xlsx, countyInfo, output)
|
|
await processCombined(xlsx, countyInfo, output)
|
|
}
|
|
|
|
function process (vaccinationSources, countyInfo) {
|
|
return processSingleWorkbook(vaccinationSources.at(-1), countyInfo)
|
|
}
|
|
|
|
export default process
|