python操作mysql

標籤：python基礎   python操作mysql   

1、下載pymysql

pip3 install pymysql

2、匯入模組

>>> import pymysql

3、建立串連

>>> conn = pymysql.connect(host="127.0.0.1",port=3306,user="root",passwd="123456",db="test")

4、建立遊標

>>> cursor = conn.cursor()

5、執行sql語句，注意：返回的是受影響的行數

>>> effect_row = cursor.execute("create table student(ID int not null,Name varchar(10))")

6、向表中插入資料

>>> effect_row = cursor.executemany("insert into student(ID,Name)values(%s,%s)",[(1,"wilson"),(2,"david")])

>>> conn.commit()    //需要提交一下，因為預設開啟事務

還可以：

>>> date = [(5,"cc"),(6,"dd"),(7,"ee")]

>>> effect_row = cursor.executemany("insert into student(ID,Name)values(%s,%s)",date)

>>> conn.commit() 

7、查詢資料

>>> effect_row = cursor.execute("select * from student")

>>> print(cursor.fetchone())                      //fetchone()唯讀取一條

>>> print(cursor.fetchall())                      //fetchall()讀取剩下的所有

>>> print(cursor.fetchmany(3))                     //detchmany(N)擷取前N條資料

以上的代碼都是在寫一些原生sql語句，代碼的重複性無法得到好的解決，接下來所要說的sqlalchemy就是解決這一問題的方案，不需要寫原生sql語句，就和調用類方法那樣操作資料庫。

1、安裝sqlalchemy

pip3 install sqlalchemy

2、使用sqlalchemy建立表結構

import sqlalchemy

from sqlalchemy import create_engine

from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy import Column,Integer,String

from sqlalchemy.orm import sessionmaker

#建立表結構

engine = create_engine("mysql+pymysql://root:[email protected]/test",encoding="utf-8",echo=True) //串連資料庫，root為使用者名稱，123456為密碼，localhost為本機登入，test為庫

Base = declarative_base()    //產生基類

class User(Base):

  __tablename__ = "user"    //表名

  id = Column(Integer,primary_key=True) //設定id欄位並且為主鍵

  name = Column(String(32))   //設定name欄位

  password = Column(String(64))  //設定password欄位

Base.metadata.create_all(engine)   //建立表結構

3、使用sqlalchemy插入資料

Session_class = sessionmaker(bind=engine) #建立與資料庫的會話session class ,注意,這裡返回給session的是個class,不是執行個體

Session = Session_class() #產生session執行個體

user_obj = User(name="david",password="123456") #產生你要建立的資料對象

print(user_obj.name,user_obj.id)  #此時還沒建立對象呢，列印一下id是None

Session.add(user_obj) #把要建立的資料對象添加到這個session裡， 一會統一建立

print(user_obj.name,user_obj.id) #此時也依然還沒建立

Session.commit() #現此才統一提交，建立資料

print(user_obj.name,user_obj.id,user_obj.password)

4、查詢

data = Session.query(User).filter_by(name="david").all()  #first()取第一個

data2 = Session.query(User).filter(User.id>1).all()

print(data)

print(data[0].name)

print(data2)

data = Session.query(User).filter(User.id>1).filter(User.id<3).all()    #多條件查詢時，需要加filter

print(data)

5、修改

data = Session.query(User).filter(User.id>1).filter(User.id<3).first()

data.name = "Wilson"

data.password = "www.mc.com"

Session.commit()

6、復原

make_user = User(name="Licis",password="www")

Session.add(make_user)

print(Session.query(User).filter(User.name.in_(["Licis","David"])).all())

Session.rollback()

Session.commit()

7、分組統計

from sqlalchemy import func

print(Session.query(func.count(User.name),User.name).group_by(User.name).all())

python操作mysql