Python3实现excel文件转yaml格式文件（二）

发表于 分类于 阅读次数： Waline： 本文字数： 2k 阅读时长 ≈ 2 分钟

经过小伙伴的使用，第一个版本出现的很多问题，因此重构了一下脚本，思路基本不变，实现了当某一列的其中有空值，则不写入Yaml文件。

示例表格文件

这里需要注意的是，脚本判定以逗号（英文逗号）为换行，并实现自定义缩进。

转换效果图

这里由于Windows系统编码问题，导致中文乱码，只是显示问题不影响Nornir脚本执行，点击上方以GBK编码显示就可以了，我在Linux上测试了一下，显示中文是正常的。

脚本文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
import xlrd
import collections
import sys


def main(table_name, outfile_name):
    table_name = xlrd.open_workbook(table_name)
    sheet_name = table_name.sheet_by_index(0)
    title = sheet_name.row_values(0)

    for i in range(1, sheet_name.nrows):
        data = sheet_name.row_values(i)

        info_dict = collections.OrderedDict()
        for k in range(1, len(title)):

            # 如果空字符串或空置，跳过
            if data[k] == '' or data[k] is None:
                continue

            # 如果是字符串，转换为列表或字符串
            if isinstance(data[k], str):
                # 如果有逗号，识别为列表
                if ',' in data[k]:
                    dl = data[k].split(',')
                    dl = [x for x in dl if x != '' and x is not None]   # 去除列表中空字符串和空值
                    dl_type = int

                    # 检测列表中是否全是数字
                    for j in dl:
                        try:
                            int(j)
                        except:
                            dl_type = str
                            break

                    # 如果列表中全是数字，把列表转换为int类型
                    if dl_type == int:
                        dl = list(map(int, dl))

                    info_dict[title[k]] = dl
                else:
                    info_dict[title[k]] = data[k]

            # 如果为float类型，转换int
            if isinstance(data[k], float):
                info_dict[title[k]] = int(data[k])

        data_dict = collections.OrderedDict()
        data_dict[data[0]] = info_dict

        # 打开文件把字典以yaml格式写入
        with open(outfile_name, mode='a') as f:
            f.write(dict_to_yaml(data_dict))


def dict_to_yaml(dict_data):
    s = ""
    hh_code = "\n"  # 换行符
    sj_code = " " * 2  # 缩进符，两个空格
    for key, value in dict_data.items():
        s += key + ":" + hh_code
        print(s)
        for key2, value2 in value.items():
            s += sj_code + key2 + ": "
            if isinstance(value2, list):
                s += hh_code
                for value3 in value2:
                    s += sj_code * 2 + "- " + str(value3) + hh_code
            else:
                s += str(value2) + hh_code
    return s


if __name__ == '__main__':
    # 函数运行
    main(table_name='ip.xls', outfile_name='hosts.yaml')

    # 命令行参数运行  python3 test.py 主机名表格.xls yaml文件.txt
    # main(table_name=sys.argv[1], outfile_name=sys.argv[2])

经过测试，在Windows平台或者Linux平台脚本转换出来的Yaml文件都可以正常被调用。

相关文章