Python對資料庫進行簡單操作

Python對資料庫進行簡單操作

來自專欄自動化測試5 人贊了文章

Python對資料庫進行簡單操作

#!/usr/bin/python

#-*- coding:UTF-8 -*-

import MySQLdb

#打開資料庫連接localhost--伺服器；testuser--連接用戶名； test123--連接密碼； TESTDB--資料庫名

db=MySQLdb.connet("localhost","testuser","test123","TESTDB")

#使用cursor（）方法獲取操作游標

cursor=db.cursor()

#如果數據表已經存在使用execute（）方法刪除表。

cursor.exectute("DROP TABLE IF EXISTS EMPLOYEE")

#創建數據表SQL 語句

sql="""CREATE TABLE EMPLOYEE(

FIRST_NAME CHAR(20) NOT NULL,

LAST_NAME CHAR(20),

AGE INT,

SEX CHAR(1)

INCOME FLOAT)"""

cursor.execute(sql)

#SQL插入語句

sqlinsert="""INSERT INTO EMPLOYEE(FIRST_NAME,

LAST_NAME,AGE,SEX,INCOME)

VALUES(Mac,Mohan,20,M,2000)"""

#與上一句sql語句等效 --向SQL語句中傳遞參數

sqlinsert1="INSERT INTO EMPLOYEE(FIRST_NAME,

LAST_NAME,AGE,SEX,INCOME)

VALUES(%s,%s,%d,%c,%d)" %

(Mac,Mohan,20,M,2000)

try:

#執行sql語句

cursor.excute(sqlinsert)

#提交數據到資料庫執行

db.commit()

except:

#Rollback in case there is any error

db.rollback()

#SQL查詢語句

sqlsel="SELECT * FROM EMPLOYEE WHERE INCOME >%d"%(1000)

try:

#執行sql語句

cursor.execute(sqlsel)

#獲取所有記錄列表

results=cursor.fetchall()

for row in results:

fname=row[0]

lname=row[1]

age=row[2]

sex=row[3]

income=row[4]

#列印結果

print("fname=%s,lname=%s,age=%d,sex=%s,income=%d" %(fname,lname,age,sex,income))

except:

print("Error:unable to fecth data")

#SQL更新語句

sqlupd="UPDATE EMPLOYEE SET AGE=AGE+1 WHERE SEX=%c" %(M)

try:

cursor.excute(sqlupd)

db.commit()

except:

db.rollback()

#關閉資料庫連接

db.close()

推薦閱讀：