0928_网络运维-Python GUI

#!/usr/bin/python
# -*- coding:utf-8 -*-

import tkFileDialog, Tkinter, tkMessageBox

import py_compile

# 兼容ascii和utf-8
import sys
reload(sys)
sys.setdefaultencoding('utf8')

def gui():

# 文件目录
def filedir():
filedire = tkFileDialog.askdirectory()
if filedire == '':
textdir.set('')
else:
textdir.set(filedire)

# Config文件
def filenam():
filename = tkFileDialog.askopenfilename()
if filename == '':
textfile.set('')
else:
textfile.set(filename)

# 信息确认
def panduan():
if textdir.get() == '':
tkMessageBox.showinfo("Info", "文件目录不能为空")
return
elif textfile.get() == '':
tkMessageBox.showinfo("Info", "Config文件不能为空")
return
else:
test.destroy()

# 主窗口
test = Tkinter.Tk()
test.geometry('400x180')
test.title('生成脚本文件')

# host信息录入系统
label = Tkinter.Label(test, text = "选择包含host信息的文件：")
label.pack()

textfile = Tkinter.StringVar()
textfile.set("")
entryfile = Tkinter.Entry(test, textvariable = textfile)
entryfile.pack()

buttonfile = Tkinter.Button(test, text = "选择Config文件", command = filenam)
buttonfile.pack()

# 日常config保存路径
label = Tkinter.Label(test, text = "每日配置自动备份到如下目录：")
label.pack()

textdir = Tkinter.StringVar()
textdir.set("C:/Config")
entrydir = Tkinter.Entry(test, textvariable = textdir)
entrydir.pack()

buttondir = Tkinter.Button(test, text = "打开文件目录", command = filedir)
buttondir.pack()

# 检查确认录入信息
Confirmbutton = Tkinter.Button(test, text = "确认", command = panduan)
Confirmbutton.pack()
test.mainloop()
return (textdir.get(), textfile.get())

if __name__ == '__main__':

# gui得到保存的路径和需要的配置信息文件
(dirname, filename) = gui()

# 导出最终py文件

# readline: config.txt
# writeline: [new].py
# mainfile: main.txt
readfile = open(filename, 'r')
mainfile = open("main.txt", 'r')
writefile = open(filename + "_config.py", 'w')
writefile.write('''#!/usr/bin/python
# -*- coding:utf-8 -*-

import paramiko
import os,sys,time
import re
''')
writefile.write("dirname = \"" + dirname + '\"\n')
writefile.write("device = []\n")
for line in readfile:
if not line.strip() == '':
writefile.write("device.append(\"" + line.strip('\n') + "\")\n")

for line in mainfile:
writefile.write(line)

readfile.close()
mainfile.close()
writefile.close()

# 显示成功
test = Tkinter.Tk()
test.geometry("0x0")
tkMessageBox.showinfo("Info", "生成文件保存成" + filename + "_config.pyc")
test.destroy()
py_compile.compile(filename + "_config.py")

# 删不掉【debug】
# if os.path.exists(filename + "_config.py"):
# os.remove(filename + "_config.py")