Python简单DB认证系统

hmzone

'''　　
用户登录-添加-删除,登陆成功后执行系统命令，失败提示。通过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[1]添加用户\n\t[2]登陆用户\n\t[3]删除用户\n\t[4]查看用户信息\n\t[5]退出程序\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:
　　
        print  color.red("未知的选项,请重新输入!")