python3 自动生成配置

python3 自动生成配置

三月 20, 2020

因为平时线上发版要提前整理一个操作时的文档出来，之前都是手动整理，手动做过3次就再也不想弄了，所以写这个脚本来代替之前手动的工作。。。。
#写了2天还被吐槽代码不够优雅，我好南(ಥ﹏ಥ)
又学到了一个新的用法，游标!

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
#!/usr/bin/env python
#coding=utf-8
#by Dawn on 20200318
'''
此脚本的作用是读取server下各个模块信息，将要发版的模块生成如下配置，代替手动粘贴的过程
sh remote_restart.sh 10.101.178.13 oauth2
curl 10.101.178.13:8004/check.do
'''
import os
import toml
import time

now = time.strftime("%Y%m%d%H%M", time.localtime())
allpath=[]
allfile=[]
fileByModule={}
def getallfileByModule(path):
    allfilelist=os.listdir(path)
    for file in allfilelist:
        filepath=os.path.join(path,file)
        if os.path.isdir(filepath):
            getallfileByModule(filepath)
        elif os.path.isfile(filepath):
            module=filepath.split("/")[-2]
            if module in fileByModule:
                fileByModule[module].append(filepath)
            else:
                fileByModule[module]=[filepath]

def readallfile(file):
    serverinfo=toml.load(file)
    modulename=(serverinfo['server'])
    moduleip = (serverinfo['addr'])
    moduleport = (serverinfo['port'])
    return modulename,moduleip,moduleport

def writeallfile(server,addr,port):
    opremote=open(remotepath,"a+")
    operatingfile=opremote.write("sh remote_restart.sh {} {}\n".format(addr,server) )
    opremote.close()
    opcurl=open(curlpath,"a+")
    checkfile=opcurl.write("curl {}:{}/check.do\n".format(addr,port))
    opcurl.close()

if __name__ == '__main__':
    local='kr'#生成配置的地区#
    dir_path = '/Users/zhaodan/Downloads/LumiWorking/AIOT/online-operating/servers/{}/'.format(local)# 模块配置路径#
    modulefile = './OpModules'# 发版模块文件#
    remotepath = './{}_remotefile_{}'.format(local,now)# 生成remote文件#
    curlpath = './{}_checkfile_{}'.format(local,now)# 生成check文件#
    moduleSet=set()
    with open(modulefile) as file:
        for line in file:
            moduleSet.add(line.strip('\n'))
    getallfileByModule(dir_path)
    t=0
    while t!=-1:
        flag=False
        for module,pathList in fileByModule.items():
            if len(pathList)>t:
                modulename, moduleip, moduleport=readallfile(pathList[t])
                if modulename in moduleSet:
                    writeallfile(modulename, moduleip, moduleport)
                flag=True
        if not flag:
            break
        else:
            t+=1
            RemoteWrap=open(remotepath,"a+")
            wrap1=RemoteWrap.write("\n")
            RemoteWrap.close()
            CheckWrap=open(curlpath,"a+")
            wrap2=CheckWrap.write("\n")
            CheckWrap.close()

图为目录结构，脚本读取目录下的模块信息，生成指定格式的配置文件
python3.png