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:123456@localhost/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())