library/module/load_classify.py

#!/usr/bin/python3.6
# -*- coding: utf-8 -*-
# @Time    : 2021/3/22 15:47
# @Author  : lijunliang
# @Email   : lijunliang@topnet.net.cn
# @File    : load_classify.py
# @Software: PyCharm

def load_classify(classify_path: str):
    '''
    加载分类
    :param classify_path:
    :return:
    '''
    number2classify = {}
    f = open(classify_path, "r")
    classifies = f.read()
    for val in classifies.split("\n")[1:]:
        val = val.split(",")
        number2classify[val[1].strip()] = val[0].strip()
    return create_classify(number2classify)


def create_classify(number2classify: dict) -> dict:
    last_classify = {}
    for key, val in number2classify.items():
        val_len = len(key)
        if val_len <= 2:
            last_classify[val] = key
        this_key = []
        for ep in range(1, int(val_len / 2) + 1):
            tier = key[:ep * 2]
            this_key.append(number2classify[tier])
        tier_title = "/".join(this_key)
        last_classify[tier_title] = key
    return last_classify


if __name__ == '__main__':
    number2classify = load_classify("../data/classify.csv")
    # ret = create_classify(number2classify)
    print(number2classify)