1.DbUtil操作類
#!/usr/bin/python
import pymysql;
'''
數據庫操作類
'''
class DbUtil:
dbconnect = None;
hostname = '';
username = '';
password = '';
dbname = '';
def __init__(self,hostname,username,password,dbname):
self.hostname = hostname;
self.username = username;
self.password = password;
self.dbname = dbname;
'''
獲取mysql連接對象
'''
def open_connect(self):
try:
# 創建數據庫連接
self.dbconnect = pymysql.connect(self.hostname,self.username,self.password,self.dbname);
except:
print('打開連接異常');
'''
數據添加操作
'''
def insert_data(self,sql):
# 使用cursor()方法獲取操作遊標
cursor = self.dbconnect.cursor();
try:
# 執行sql語句
cursor.execute(sql);
# 提交到數據庫執行
self.dbconnect.commit();
last_id = self.get_last_insert_id(cursor);
cursor.close();
return last_id;
except:
# 如果發生錯誤則回滾
self.dbconnect.rollback();
print('請檢查sql語法是否正確');
return 0;
'''
獲取數據添加成功後的自增ID
'''
def get_last_insert_id(self,cursor):
sql = 'SELECT LAST_INSERT_ID() AS id;';
cursor.execute(sql);
database = cursor.fetchone();
return database[0];
'''
查詢單條數據
'''
def find_one(self,sql):
# 使用cursor()方法獲取操作遊標
cursor = self.dbconnect.cursor();
try:
# 執行SQL語句
cursor.execute(sql)
# 獲取所有記錄列表
results = cursor.fetchone()
cursor.close();
return results;
except:
print ("Error: unable to fetch data")
return None;
'''
關閉數據庫連接
'''
def close_connect(self):
if self.dbconnect != None:
self.dbconnect.close();
2.DbUtil類截圖
3.使用DbUtil類
#!/usr/bin/python
from DbUtil import DbUtil;
from RegionModel import RegionModel;
dbUtil = DbUtil("localhost","root","","enquiry");
regionModel = RegionModel(dbUtil);
region = regionModel.getAreaCode();
print(region[0]);