Python 终端下的TUI开发，whiptail 的 Python 封装

 

Python 终端下的TUI开发

Mr. Neo Chen (netkiller), 陈景峰(BG7NYT)

<[email protected]>

版权 © 2011, 2012 http://netkiller.github.com

摘要

 

下面是我多年积累下来的经验总结，整理成文档供大家参考:

 

Netkiller Architect 手札	Netkiller Linux 手札	Netkiller Developer 手札	Netkiller Database 手札
Netkiller Debian 手札	Netkiller CentOS 手札	Netkiller FreeBSD 手札	Netkiller Shell 手札
Netkiller Web 手札	Netkiller Monitoring 手札	Netkiller Storage 手札	Netkiller Mail 手札
Netkiller Security 手札	Netkiller Multimedia 手札	Netkiller Writer 手札	Netkiller Version 手札
Netkiller PostgreSQL 手札	Netkiller MySQL 手札	Netkiller Cryptography 手札	Netkiller Cisco IOS 手札
Netkiller LDAP 手札	Netkiller Intranet 手札

 

 

#!/usr/bin/env python3
#/bin/env python3
#-*- coding: utf-8 -*-
##############################################
# Home  : http://netkiller.sourceforge.net/
# Author: Neo <[email protected]>
##############################################
# vim:ts=4:sw=4

import os,subprocess

class Whiptail():
	
	def __init__(self,title = None, backtitle = ""):
		if title :
			self.t = '"'+ title +'"'
		if backtitle :
			self.backtitle = '"'+ backtitle +'"'
		else:
			backtitle = ""
		self.ismenu = False
	def title(self, tmp):
		self.t = tmp
		return(self)
	def inputbox(self, label, hight, width, init = ""):
		self.form = '--inputbox' + ' "'+label+'" ' +  str(hight) + ' ' + str(width) + ' '+init
		return(self)
	def passwordbox(self, label, hight, width):
		self.form = '--passwordbox' + ' "'+label+'" ' +  str(hight) + ' ' + str(width)
		return(self)
	def menu(self,lable, item, hight, width, listheight):
		menuitem = ''
		for i in item:
			key,value = i
			menuitem = menuitem + '"'+key + '" "' + value + '" '
		self.form = '--menu' + ' "'+lable+'" ' +  str(hight) + ' ' + str(width) + ' ' + str(listheight) +' '+ menuitem
		self.ismenu = True
		return(self)
	def yesno(self, text, hight, width):
		self.form = '--yesno' + ' "'+text+'" ' +  str(hight) + ' ' + str(width)
		return(self)
	def radiolist(self, text, height, width, listheight, item):
		radioitem = ''
		for i in item:
			tag, item, status = i
			radioitem = radioitem + '"'+tag + '" "' + item + '" "' + status + '" '
		self.radiolist = '--yesno' + ' "'+text+'" ' +  str(height) + ' ' + str(width) + ' ' + str(listheight) + ' ' + radioitem
		return(self)
	def run(self):
		screen = None
		#whiptail = 'whiptail ' + ' --title ' + self.t + ' ' + self.form + ' 3>&1 1>&2 2>&3'
		#screen = os.system(whiptail)
		whiptail = 'whiptail ' + ' --title ' + self.t + ' --backtitle ' + self.backtitle +' ' + self.form + ' 3>&2 2>&1 1>&3 | tee /tmp/.whiptail'
		#screen = subprocess.getoutput(whiptail)
		try:
			#screen = subprocess.call('/usr/bin/whiptail', whiptail, shell=True)
			#p = subprocess.check_output(['/usr/bin/whiptail',whiptail], stderr = subprocess.PIPE, shell=False, universal_newlines=True)
			
			#p = subprocess.Popen('/usr/bin/whiptail', whiptail, stderr = subprocess.PIPE, shell = False)   
			#p.stdin.write('3\n')   
			#p.stdin.write('4\n')   
			#screen = p.stdout.read() 
			#print(p)
			#screen = p.stderr.read() 	
			os.system(whiptail)
			f = open('/tmp/.whiptail','r')
			screen = f.read()
			f.close()
		except OSError as e:
			print(e)

		#print(whiptail)

		return(screen)

class MySQL():
	mysql='mysql'
	def __init__(self, hosts = None):
		self.dialog = Whiptail('MySQL Adminstrator','MySQL Client')
		self.dbhost = None
		self.dbuser = None
		self.dbpass = None
		if hosts :
			self.hosts = hosts
	def test(self):
		dialog = Whiptail('Test')
		dialog.inputbox('Windows', 10, 20).run()
	def menu(self):
		hosts = []
		hosts.append(('1','MySQL Manager'))
		hosts.append(('2','MySQL Backup'))
		hosts.append(('3','MySQL Restore'))
		m = self.dbhost = self.dialog.menu('Menu', hosts, 15, 30, 3).run()
		return(m)
	def host(self, hosts):
		self.dbhost = self.dialog.menu('Database Host', hosts, 20, 40, 5).run()
	def login(self):
		hight = 8
		width = 40
		self.dbuser = self.dialog.inputbox('User', hight, width, 'root').run()
		self.dbpass = self.dialog.passwordbox('Password', hight, width).run()
		pass
	def dump(self,hosts):
		hight = 8
		width = 40
		self.host(self.hosts)
		self.login()
		dbname = self.dialog.inputbox('Database', hight, width).run()
		dbfile = self.dialog.inputbox('File', hight, width).run()
		yesno = self.dialog.yesno('Backup?', hight, width).run()
		cmd = """mysqldump -h%s -u%s -p%s %s | gzip > /opt/backup/%s""" % (self.dbhost, self.dbuser, self.dbpass, dbname, dbfile)
		#print(cmd)
		#print(yesno)
		#if yesno :
		os.system(cmd)
	def client(self,hosts):
		hight = 8
		width = 40
		self.host(hosts)
		self.login()
		dbname = self.dialog.inputbox('Database', hight, width).run()
		cmd = """mysql -h%s -u%s -p%s %s""" % (self.dbhost, self.dbuser, self.dbpass, dbname)
		#print(cmd)
		os.system( cmd )
	def restore(self,hosts):
		hight = 8
		width = 40	
		self.host(hosts)
		self.login()
		files = os.listdir('/opt/backup')
		#print(files)
		menu = []
		i = 0
		for f in files:
			menu.append((str(i),f))
			i = i + 1
		dbfile = self.dbhost = self.dialog.menu('Backup History', menu, 20, 30, 7).run()
		dbname = self.dialog.inputbox('Database', hight, width).run()
		cmd = """zcat %s | mysql -h%s -u%s -p%s %s""" % (backup, self.dbhost, self.dbuser, self.dbpass, dbname)
		#print(cmd)
		os.system( cmd )
	def logout(self):
		pass
	def close(self):
		pass
	
	def main(self):
		m = self.menu()
		print(m)
		if m == '2':
			self.dump(self.hosts)
		elif m == '3':
			self.restore(self.hosts)
		else:
			self.client(self.hosts)
	
if __name__ == '__main__':
	try:
		hosts = []
		hosts.append(('127.0.0.1','localhost'))
		hosts.append(('172.16.0.1','mysql master'))
		hosts.append(('172.16.0.2','mysql slave'))
		    
		mysql = MySQL(hosts)
		mysql.main()
	except KeyboardInterrupt:
		print ("Crtl+C Pressed. Shutting down.")
		os.exit()