69 lines
1.9 KiB
TypeScript
69 lines
1.9 KiB
TypeScript
import areaTotal from './area.json';
|
||
import fs from 'fs';
|
||
|
||
const provinceName = process.argv[2] || '浙江省';
|
||
const level = process.argv[3] || 'district';
|
||
const cityNumber = process.argv[4] ? parseInt(process.argv[4]) : 2;
|
||
|
||
type Area = {
|
||
id: string;
|
||
level: 'country' | 'province' | 'city' | 'district' | 'street';
|
||
parentId: string;
|
||
name: string;
|
||
};
|
||
|
||
function outputData(output: Area[]) {
|
||
fs.writeFileSync(`${__dirname}/../src/data/areaDebug.json`, JSON.stringify(output));
|
||
console.log(`输出areaDebug.json完成，共输出了${output.length}行数据`);
|
||
}
|
||
|
||
// 输出
|
||
function main() {
|
||
console.log(`准备输出${provinceName}的前${cityNumber}个市的地区数据，输出到${level}层为止`);
|
||
const areaTotal2 = areaTotal as Area[];
|
||
const output = [] as Area[];
|
||
const province = areaTotal2.find(
|
||
ele => ele.level === 'province' && ele.name === provinceName
|
||
);
|
||
if (!province) {
|
||
console.error(`找不到名为${provinceName}的省`);
|
||
return;
|
||
}
|
||
output.push(province);
|
||
const country = areaTotal2.find(
|
||
ele => ele.id === province.parentId
|
||
);
|
||
output.push(country!);
|
||
const cities = areaTotal2.filter(
|
||
ele => ele.parentId === province.id
|
||
).sort(
|
||
(ele1, ele2) => parseInt(ele1.id) - parseInt(ele2.id)
|
||
);
|
||
const cities2 = cities.slice(0, cityNumber);
|
||
output.push(...cities2);
|
||
|
||
if (level === 'city') {
|
||
outputData(output);
|
||
return;
|
||
}
|
||
|
||
const districts = areaTotal2.filter(
|
||
ele => cities2.find(city => ele.parentId === city.id)
|
||
);
|
||
output.push(...districts);
|
||
|
||
if (level === 'district') {
|
||
outputData(output);
|
||
return;
|
||
}
|
||
|
||
const streets = areaTotal2.filter(
|
||
ele => districts.find(district => ele.parentId === district.id)
|
||
);
|
||
output.push(...streets);
|
||
outputData(output);
|
||
}
|
||
|
||
main();
|
||
|