Python简单DB认证系统
'''用户登录-添加-删除,登陆成功后执行系统命令，失败提示。通过mysql数据库认证。打印color
'''
#encoding:utf8
import os
import sys
import datetime
import MySQLdb
import time
import sh
class Colored(object):
# 显示格式: \033[显示方式;前景色;背景色m
# 只写一个字段表示前景色,背景色默认
RED = '\033[31m' # 红色
GREEN = '\033[32m' # 绿色
YELLOW = '\033[33m' # 黄色
BLUE = '\033[34m' # 蓝色
FUCHSIA = '\033[35m' # 紫红色
CYAN = '\033[36m' # 青蓝色
WHITE = '\033[37m' # 白色
#: no color
RESET = '\033[0m' # 终端默认颜色
def color_str(self, color, s):
return '{}{}{}'.format(
getattr(self, color),
s,
self.RESET
)
def red(self, s):
return self.color_str('RED', s)
def green(self, s):
return self.color_str('GREEN', s)
def yellow(self, s):
return self.color_str('YELLOW', s)
def blue(self, s):
return self.color_str('BLUE', s)
def fuchsia(self, s):
return self.color_str('FUCHSIA', s)
def cyan(self, s):
return self.color_str('CYAN', s)
def white(self, s):
return self.color_str('WHITE', s)
class initdb(object):
#conn=MySQLdb.connect(host='10.7.7.180',user='root',passwd='123456',db='test')
#cur=conn.cursor()
def __init__(self,username='NULL',password='NULL'):
self.username=username
self.password=password
self.conn=MySQLdb.connect(host='10.7.7.180',user='root',passwd='123456',db='test')
self.cur=self.conn.cursor()
def add_user(self):
itime=datetime.datetime.now()
self.cur.execute("INSERT INTO user (username, password,createtime) VALUES ('%s','%s','%s')" % (self.username,self.password,itime))
self.cur.close()
self.conn.commit()
self.conn.close()
return "Add user %s successful!" % self.username
def login_check(self):
self.cur.execute("select * from user where username='%s' and password = '%s'" % (self.username,self.password))
qur_result = self.cur.fetchall()
if qur_result == ():
return 0
else:
return 1
self.cur.close()
self.conn.close()
def del_user(self):
self.cur.execute("delete from user where username='%s'" % self.username)
self.cur.close()
self.conn.commit()
self.conn.close()
def see_user(self):
self.cur.execute("select * from user")
data = self.cur.fetchall()
self.cur.close()
self.conn.close()
d = list(data)
for x in d:
print x
while True:
color = Colored()
print color.fuchsia('*'*50)
choice=raw_input(color.green("请选择操作选项:\n\t添加用户\n\t登陆用户\n\t删除用户\n\t查看用户信息\n\t退出程序\n请选择:"))
if choice == '1':
inname=raw_input(color.yellow("请输入要添加的用户名称:"))
inpassword=raw_input(color.yellow("请输入要为%s添加的密码:" % inname))
initdb(inname,inpassword).add_user()
print color.blue("用户%s添加成功!" % inname)
elif choice == '2':
username=raw_input(color.yellow("请输入用户名:"))
password=raw_input(color.yellow("请输入密码:"))
f=initdb(username,password).login_check()
if f == 0:
print color.red("Login faild!please check your username and password!")
elif f == 1:
print color.green("Login successful!\nHello,MR %s" % username)
time.sleep(3)
filelist=open('/etc/passwd')
for i in filelist.readlines():
print i
time.sleep(3)
print sh.ifconfig('ens33')
print sh.df('-h')
print sh.free('-m')
print sh.tail('-100','/var/log/auth.log')
else:
print "something wrong!"
elif choice == '3':
dname=raw_input(color.yellow("请输入要删除的用户名称:"))
initdb(dname).del_user()
print color.red("用户%s删除成功!" % dname)
elif choice == '4':
initdb().see_user()
elif choice == '5':
print color.green("程序已退出,欢迎下次光临!")
break
else:
printcolor.red("未知的选项,请重新输入!")
页:
[1]