mysql知識樹整理【4】---pymysql

鷹眼2號發表於2018-10-28

安裝引入模組

  • 安裝mysql模組
sudo apt-get install python-mysql
  • 在檔案中引入模組
import Mysqldb

Connection物件

  • 用於建立與資料庫的連線
  • 建立物件:呼叫connect()方法
conn=connect(引數列表)
  • 引數host:連線的mysql主機,如果本機是'localhost'
  • 引數port:連線的mysql主機的埠,預設是3306
  • 引數db:資料庫的名稱
  • 引數user:連線的使用者名稱
  • 引數password:連線的密碼
  • 引數charset:通訊採用的編碼方式,預設是'gb2312',要求與資料庫建立時指定的編碼一致,否則中文會亂碼

物件的方法

  • close()關閉連線
  • commit()事務,所以需要提交才會生效
  • rollback()事務,放棄之前的操作
  • cursor()返回Cursor物件,用於執行sql語句並獲得結果

Cursor物件

  • 執行sql語句
  • 建立物件:呼叫Connection物件的cursor()方法
cursor1=conn.cursor()

物件的方法

  • close()關閉
  • execute(operation [, parameters ])執行語句,返回受影響的行數
  • fetchone()執行查詢語句時,獲取查詢結果集的第一個行資料,返回一個元組
  • next()執行查詢語句時,獲取當前行的下一行
  • fetchall()執行查詢時,獲取結果集的所有行,一行構成一個元組,再將這些元組裝入一個元組返回
  • scroll(value[,mode])將行指標移動到某個位置
    • mode表示移動的方式
    • mode的預設值為relative,表示基於當前行移動到value,value為正則向下移動,value為負則向上移動
    • mode的值為absolute,表示基於第一條資料的位置,第一條資料的位置為0

物件的屬性

  • rowcount只讀屬性,表示最近一次execute()執行後受影響的行數
  • connection獲得當前連線物件

------------------------------------------------------------------------------------------------------------------------------------------------------------------------

增加

  • 建立testInsert.py檔案,向學生表中插入一條資料
#encoding=utf-8
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    count=cs1.execute("insert into students(sname) values('張良')")
    print count
    conn.commit()
    cs1.close()
    conn.close()
except Exception,e:
    print e.message

修改

  • 建立testUpdate.py檔案,修改學生表的一條資料
#encoding=utf-8
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    count=cs1.execute("update students set sname='劉邦' where id=6")
    print count
    conn.commit()
    cs1.close()
    conn.close()
except Exception,e:
    print e.message

刪除

  • 建立testDelete.py檔案,刪除學生表的一條資料
#encoding=utf-8
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    count=cs1.execute("delete from students where id=6")
    print count
    conn.commit()
    cs1.close()
    conn.close()
except Exception,e:
    print e.message

sql語句引數化

  • 建立testInsertParam.py檔案,向學生表中插入一條資料
#encoding=utf-8
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cs1=conn.cursor()
    sname=raw_input("請輸入學生姓名:")
    params=[sname]
    count=cs1.execute('insert into students(sname) values(%s)',params)
    print count
    conn.commit()
    cs1.close()
    conn.close()
except Exception,e:
    print e.message

其它語句

  • cursor物件的execute()方法,也可以用於執行create table等語句
  • 建議在開發之初,就建立好資料庫表結構,不要在這裡執行

------------------------------------------------------------------------------------------------------------------------------------------------------------------------

查詢一行資料

  • 建立testSelectOne.py檔案,查詢一條學生資訊
#encoding=utf8
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cur=conn.cursor()
    cur.execute('select * from students where id=7')
    result=cur.fetchone()
    print result
    cur.close()
    conn.close()
except Exception,e:
    print e.message

查詢多行資料

  • 建立testSelectMany.py檔案,查詢一條學生資訊
#encoding=utf8
import MySQLdb
try:
    conn=MySQLdb.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
    cur=conn.cursor()
    cur.execute('select * from students')
    result=cur.fetchall()
    print result
    cur.close()
    conn.close()
except Exception,e:
    print e.message

-----------------------------------------------------------------------------------------------------------------------------------------------------------------------

封裝

  • 觀察前面的檔案發現,除了sql語句及引數不同,其它語句都是一樣的
  • 建立MysqlHelper.py檔案,定義類
#encoding=utf8
import MySQLdb

class MysqlHelper():
    def __init__(self,host,port,db,user,passwd,charset='utf8'):
        self.host=host
        self.port=port
        self.db=db
        self.user=user
        self.passwd=passwd
        self.charset=charset

    def connect(self):
        self.conn=MySQLdb.connect(host=self.host,port=self.port,db=self.db,user=self.user,passwd=self.passwd,charset=self.charset)
        self.cursor=self.conn.cursor()

    def close(self):
        self.cursor.close()
        self.conn.close()

    def get_one(self,sql,params=()):
        result=None
        try:
            self.connect()
            self.cursor.execute(sql, params)
            result = self.cursor.fetchone()
            self.close()
        except Exception, e:
            print e.message
        return result

    def get_all(self,sql,params=()):
        list=()
        try:
            self.connect()
            self.cursor.execute(sql,params)
            list=self.cursor.fetchall()
            self.close()
        except Exception,e:
            print e.message
        return list

    def insert(self,sql,params=()):
        return self.__edit(sql,params)

    def update(self, sql, params=()):
        return self.__edit(sql, params)

    def delete(self, sql, params=()):
        return self.__edit(sql, params)

    def __edit(self,sql,params):
        count=0
        try:
            self.connect()
            count=self.cursor.execute(sql,params)
            self.conn.commit()
            self.close()
        except Exception,e:
            print e.message
        return count

新增

  • 建立testInsertWrap.py檔案,使用封裝好的幫助類完成插入操作
#encoding=utf8
from MysqlHelper import *

sql='insert into students(sname,gender) values(%s,%s)'
sname=raw_input("請輸入使用者名稱:")
gender=raw_input("請輸入性別,1為男,0為女")
params=[sname,bool(gender)]

mysqlHelper=MysqlHelper('localhost',3306,'test1','root','mysql')
count=mysqlHelper.insert(sql,params)
if count==1:
    print 'ok'
else:
    print 'error'

查詢一個

  • 建立testGetOneWrap.py檔案,使用封裝好的幫助類完成查詢最新一行資料操作
#encoding=utf8
from MysqlHelper import *

sql='select sname,gender from students order by id desc'

helper=MysqlHelper('localhost',3306,'test1','root','mysql')
one=helper.get_one(sql)
print one

----------------------------------------------------------------------------------------------------------------------------------------------------------------------

例項:使用者登入

建立使用者表userinfos

  • 表結構如下
    • id
    • uname
    • upwd
    • isdelete
  • 注意:需要對密碼進行加密
  • 如果使用md5加密,則密碼包含32個字元
  • 如果使用sha1加密,則密碼包含40個字元,推薦使用這種方式
create table userinfos(
id int primary key auto_increment,
uname varchar(20),
upwd char(40),
isdelete bit default 0
);

加入測試資料

  • 插入如下資料,使用者名稱為123,密碼為123,這是sha1加密後的值
insert into userinfos values(0,'123','40bd001563085fc35165329ea1ff5c5ecbdbbeef',0);

接收輸入並驗證

  • 建立testLogin.py檔案,引入hashlib模組、MysqlHelper模組
  • 接收輸入
  • 根據使用者名稱查詢,如果未查到則提示使用者名稱不存在
  • 如果查到則匹配密碼是否相等,如果相等則提示登入成功
  • 如果不相等則提示密碼錯誤
#encoding=utf-8
from MysqlHelper import MysqlHelper
from hashlib import sha1

sname=raw_input("請輸入使用者名稱:")
spwd=raw_input("請輸入密碼:")

s1=sha1()
s1.update(spwd)
spwdSha1=s1.hexdigest()

sql="select upwd from userinfos where uname=%s"
params=[sname]

sqlhelper=MysqlHelper('localhost',3306,'test1','root','mysql')
userinfo=sqlhelper.get_one(sql,params)
if userinfo==None:
    print '使用者名稱錯誤'
elif userinfo[0]==spwdSha1:
    print '登入成功'
else:
    print '密碼錯誤'