|
"""This module provides functionality for creating a hierarchy tree and a mapping from ISCO code to node name.""" |
|
|
|
import csv |
|
import requests |
|
|
|
|
|
def create_hierarchy_dict(file: str) -> dict: |
|
""" |
|
Creates a dictionary where keys are nodes and values are sets of parent nodes representing the group level hierarchy of the ISCO-08 structure. |
|
The function assumes that the input CSV file has a column named 'unit' with the 4-digit ISCO-08 codes. |
|
A csv file with the ISCO-08 structure can be downloaded from the International Labour Organization (ILO) at [https://www.ilo.org/ilostat-files/ISCO/newdocs-08-2021/ISCO-08/ISCO-08 EN.csv](https://www.ilo.org/ilostat-files/ISCO/newdocs-08-2021/ISCO-08/ISCO-08%20EN.csv) |
|
|
|
Args: |
|
- file: A string representing the path to the CSV file containing the 4-digit ISCO-08 codes. It can be a local path or a web URL. |
|
|
|
Returns: |
|
- A dictionary where keys are ISCO-08 unit codes and values are sets of their parent codes. |
|
""" |
|
isco_hierarchy = {} |
|
|
|
if file.startswith("http://") or file.startswith("https://"): |
|
response = requests.get(file) |
|
lines = response.text.splitlines() |
|
else: |
|
with open(file, newline="") as csvfile: |
|
lines = csvfile.readlines() |
|
|
|
reader = csv.DictReader(lines) |
|
for row in reader: |
|
unit_code = row["unit"].zfill(4) |
|
minor_code = unit_code[0:3] |
|
sub_major_code = unit_code[0:2] |
|
major_code = unit_code[0] |
|
isco_hierarchy[unit_code] = {minor_code, major_code, sub_major_code} |
|
|
|
return isco_hierarchy |
|
|
|
|
|
|
|
|
|
|
|
|
