TrafficMonitoring/hello.py at master · Dontoronto/TrafficMonitoring · GitHub

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
81
82
83
84
#!/usr/bin/env python

import yaml
import re

class MyDumper(yaml.Dumper):

    def increase_indent(self, flow=False, indentless=False):
        return super(MyDumper, self).increase_indent(flow, False)


with open("environment.yml", "r") as stream:
    try:

        print(type(stream))
        #print(yaml.safe_dump_all(stream, line_break="   "))
        '''
        a = dict(yaml.safe_load_all(stream))
        print(a)
        for elem, key in a.items():
            #print(elem)
            print(key)'''


        a = yaml.safe_load_all(stream)

        with open('testfile.yml', 'w') as p:

            #b = yaml.dump_all(a,p, Dumper=MyDumper, default_flow_style=False)

            b = yaml.dump_all(a,indent=2, allow_unicode=True, default_flow_style=False, width=4)

            pre_text = b.split("\n")
            print(pre_text)
            #print(pre_text)
            new_text = ""

            for i in range(len(pre_text)):
                #new_text = new_text + "  " + snippet
                #snippet = "  " + snippet


                if(re.search("^[a-z]", pre_text[i])):
                    pass
                else:
                    if(re.search("=.*",pre_text[i])):
                        pre_text[i] = re.sub("=.*", "", pre_text[i])
                    pre_text[i] = "  " + pre_text[i]
                #new_text.join(str(pre_text[i]))
                #new_text.join("  " + pre_text[i])


            print('------------')
            #print(new_text)
            #print(pre_text)

            #print("\n".join(pre_text))

            finished_text = "\n".join(pre_text).replace('"',"")
            #print(finished_text)


            #yaml.safe_dump_all(finished_text,p,)
            p.write(finished_text)


            #yaml.dump_all(new_text,p)

            #print(b)
            #print(type(b))

        #print(yaml.safe_load(stream))
        #print(len(yaml.safe_load(stream)))

    except yaml.YAMLError as exc:
        print(exc)