1 #! /usr/bin/env python
  2 # -*- coding: utf-8 -*-
  3 import MySQLdb
  4 
  5 class Database:
  6     def __init__(self,user,passwd,db,host,port):
  7         self.user=user
  8         self.passwd=passwd
  9         self.db=db
 10         self.host=host
 11         self.port=port
 12         try:
 
 13             self.conn=MySQLdb.connect(user=self.user,passwd=self.passwd,db=self.db,host=self.host,port=self.port)
 14             self.cur=self.conn.cursor()
 15         except MySQLdb.Error as e:
 16             print(e)
 17         
 18     def is_User_exists(self,username):
 19         sql="select salary from employee where name=‘%s‘;
 
" %username
 20         try:
 21             self.cur.execute(sql)
 22             data=self.cur.fetchall()
 23             if len(data)==0:
 24                 return False
 25             else:
 26                 return True
 27         except MySQLdb.Error as e:
 28                 print (e)
 29         
 30
 
     #查詢數據
 31     def opt_Select(self,username):
 32         sql="select salary from employee where name=‘%s‘;" %username
 33         try:
 34             self.cur.execute(sql)
 35             data=self.cur.fetchall()
 36             for i in data:
 37                 print (‘工資：        %d‘) %(i[0])
 38             print(‘查詢數據‘)
 39         except MySQLdb.Error as e:
 40             print(e)
 41     
 42     #插入數據   
 43     def opt_Insert(self,username,value):
 44         sql=‘insert into employee(name,salary) values(\‘%s\‘,%s);‘ %(username,value)
 45         try:
 46             self.cur.execute(sql)
 47             self.conn.commit()
 48             print (‘插入數據‘)
 49         except MySQLdb.Error as e:
 50             print(e)
 51      
 52      #修改員工工資      
 53     def opt_Update(self,username,value):
 54         sql=‘update  employee set salary=%s where name=\‘%s\‘;‘ %(value,username)
 55         try:
 56             self.cur.execute(sql)
 57             self.conn.commit()
 58             print (‘修改數據‘)
 59         except MySQLdb.Error as e:
 60             print(e)
 61     
 62     #刪除員工數據
 63     def opt_Delete(self,username):
 64         sql=‘delete from employee where name=\‘%s\‘;‘ % username
 65         try:
 66             self.cur.execute(sql)
 67             self.conn.commit()
 68             print(‘刪除數據‘)
 69         except MySQLdb.Error as e:
 70             print(e)
 71             
 72     def close_Conn(self):
 73         try:
 74             self.cur.close()
 75             self.conn.close()
 76         except MySQLdb.Error as e:
 77             print(e)
 78 
 79 
 80 tips=‘‘‘
 81 
 82 1: 查詢
 83 2: 新增
 84 3: 修改 
 85 4: 刪除
 86 5: 退出
 87 
 88 請選擇:   
 89 ‘‘‘
 90 options=[1,2,3,4,5]
 91 
 92 if __name__==‘__main__‘:
 93     conn_mysql=Database(‘root‘,‘1‘,‘test‘,‘localhost‘,3306)
 94     while True:       
 95         # 選擇操作
 96         try:
 97             choose=input(tips)
 98         except(EOFError,KeyboardInterrupt):
 99             choose=5
100         if choose not in options:
101             continue         
102         elif choose==1:
103             username = raw_input(‘輸入要查詢員工姓名:  ‘)
104             if conn_mysql.is_User_exists(username) is False:
105                 print (‘員工不存在‘)
106             else:
107                 conn_mysql.opt_Select(username)
108         elif choose==2:
109             username,value = raw_input(‘輸入新增員工姓名,工資:  ‘).split(‘,‘)
110             conn_mysql.opt_Insert(username,value)
111         elif choose==3:
112             username,value= raw_input(‘輸入要更新員工的姓名,工資:  ‘).split(‘,‘)
113             conn_mysql.opt_Update(username,value)
114         elif choose==4:
115             username= raw_input(‘輸入要刪除員工的姓名:  ‘)
116             conn_mysql.opt_Delete(username)
117         elif choose==5:
118             print(‘退出程序......‘)
119             sys.exit()

python MySQLdb包 增刪改查簡單應用