1、准备安装Python的mysql模块"MySQLdb"和Python表格模块“prettytable”(之前文章已介绍,在此不在赘述)
[root@lnmp ~]# python Python 2.6.6 (r266:84292, Jan 22 2014, 09:42:36) [GCC 4.4.7 20120313 (Red Hat 4.4.7-4)] on linux2 Type "help", "copyright", "credits" or "license" for more information. >>> import MySQLdb >>> from prettytable import PrettyTable # 确保模块已安装
2、程序主体
[root@lnmp ~]# vim xt_forbidacc_del.py #!/usr/bin/python import os,sys,MySQLdb from prettytable import PrettyTable hostip = '127.0.0.1' username = 'root' passwd = 'redhat' db = 'mysql' try: acc = raw_input('Please input accname:').strip() conn=MySQLdb.connect(hostip,username,passwd,db,port=3306) cur=conn.cursor() cur.execute("select host,user,password from user where user='%s'" %acc) jilu = cur.fetchone() if jilu == None: print '######## No such user:\033[33;3m%s\033[0m'%acc sys.exit() else: x = PrettyTable() x.field_names = ['host','user','password'] x.padding_width = 1 x.add_row([jilu[0],jilu[1],jilu[2]]) print x sure = raw_input('Are you sure delete \033[33;3m%s\033[0m:[y/n]' %acc).strip() if sure == 'y' or sure == 'Y': cur.execute("delete from user where user='%s'" %acc) conn.commit() else: cur.close() conn.close() sys.exit() cur.close() conn.close() except MySQLdb.Error,e: print "Mysql Error %d: %s" % (e.args[0], e.args[1])
程序测试
[root@lnmp ~]# mysql -uroot -predhat
mysql> select host,user,password from mysql.user; # 删除前数据库数据 +-----------+------+-------------------------------------------+ | host | user | password | +-----------+------+-------------------------------------------+ | localhost | root | *84BB5DF4823DA319BBF86C99624479A198E6EEE9 | | lnmp | root | | | 127.0.0.1 | root | | | localhost | | | | lnmp | | | | localhost | shaw | *84BB5DF4823DA319BBF86C99624479A198E6EEE9 | +-----------+------+-------------------------------------------+ 6 rows in set (0.00 sec)
[root@lnmp ~]# python xt_forbidacc_del.py # 删除账号“shaw” Please input accname:shaw +-----------+------+-------------------------------------------+ | host | user | password | +-----------+------+-------------------------------------------+ | localhost | shaw | *84BB5DF4823DA319BBF86C99624479A198E6EEE9 | +-----------+------+-------------------------------------------+ Are you sure delete shaw:[y/n]y
[root@lnmp ~]# mysql -uroot -predhat mysql> select host,user,password from mysql.user; # 删除后数据库数据 +-----------+------+-------------------------------------------+ | host | user | password | +-----------+------+-------------------------------------------+ | localhost | root | *84BB5DF4823DA319BBF86C99624479A198E6EEE9 | | lnmp | root | | | 127.0.0.1 | root | | | localhost | | | | lnmp | | | +-----------+------+-------------------------------------------+ 5 rows in set (0.01 sec)