python使用SQLAlchemy操作MySQL
SQLAlchemy是Python程式語言下的一款開源軟體,提供了SQL工具包及物件關係對映(ORM)工具,使用MIT許可證發行。SQLAlchemy首次發行於2006年2月,並迅速地在Python社群中最廣泛使用的ORM工具之一,不亞於Django的ORM框架。
本文將介紹如何使用SQLAlchemy操作MySQL,完成基礎的表建立,表格資料的新增、查詢、修改、刪除(CRUD)等操作。
首先我們需要確認當前的Python環境下已經安裝sqlalchemy和pymysql模組。
新建表格
我們使用本地的MySQL資料庫,資料庫為orm_test。新建一張users表,欄位為id,name,age,place。完整的Python程式碼(create_table.py
# -*- coding: utf-8 -*- # author: Jclian91 # place: Sanya,Hainan # time: 12:46 from sqlalchemy.dialects.mysql import INTEGER,VARCHAR from sqlalchemy import Column from sqlalchemy import create_engine from sqlalchemy.ext.declarative import declarative_base Base = declarative_base() # users表結構 class Users(Base): __tablename__ = 'users' id = Column(INTEGER,primary_key=True) name = Column(VARCHAR(256),nullable=False) age = Column(INTEGER) place = Column(VARCHAR(256),nullable=False) def __init__(self,id,place): self.id = id self.name = name self.age = age self.place = place def init_db(): engine = create_engine( "mysql+pymysql://root:@localhost:3306/orm_test",encoding= "utf-8",echo=True ) Base.metadata.create_all(engine) print('Create table successfully!') if __name__ == '__main__': init_db()
執行後,生成的users表結構如下:
資料插入
接著我們插入幾條測試資料,完整的程式碼(db_insert.py
)如下:
# -*- coding: utf-8 -*- # author: Jclian91 # place: Sanya,Hainan # time: 12:52 from create_table import Users from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker def insert_data(): # 初始化資料庫連線 engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test",encoding="utf-8") # 建立DBSession型別 DBSession = sessionmaker(bind=engine) # 建立session物件 session = DBSession() # 插入單條資料 # 建立新User物件 new_user = Users(id=1,name='Jack',age=25,place='USA') # 新增到session session.add(new_user) # 提交即儲存到資料庫 session.commit() # 插入多條資料 user_list= [Users(id=2,name='Green',age=26,place='UK'),Users(id=3,name='Alex',age=31,place='GER'),Users(id=4,name='Chen',age=52,place='CHN'),Users(id=5,name='Zhang',age=42,place='CHN') ] session.add_all(user_list) session.commit() # 關閉session session.close() print('insert into db successfully!') if __name__ == '__main__': insert_data()
執行程式後,表格中的資料如下:
資料查詢
接下來演示如何進行簡單的查詢。我們需要查詢place為CHN的人名,完整的Python程式碼(db_query)如下:
# -*- coding: utf-8 -*- # author: Jclian91 # place: Sanya,Hainan # time: 12:52 from create_table import Users from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker def query_data(): # 初始化資料庫連線 engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test",encoding="utf-8") # 建立DBSession型別 DBSession = sessionmaker(bind=engine) # 建立session物件 session = DBSession() # 查詢所有place是CHN的人名 # 建立Query查詢,filter是where條件 # 呼叫one()返回唯一行,如果呼叫all()則返回所有行: users = session.query(Users).filter(Users.place == 'CHN').all() print([use.name for use in users]) # 輸出:['Chen','Zhang'] # 或者用如下查詢 users = session.query(Users.name).filter(Users.place == 'CHN').all() print(users) # 輸出:[('Chen',),('Zhang',)] session.close() if __name__ == '__main__': query_data()
輸出結果如下:
['Chen','Zhang']
[('Chen',)]
資料更新
接下來我們演示資料更新。我們需要將Jack的place修改為CHN,完整的Python程式碼(db_update)如下:
# -*- coding: utf-8 -*- # author: Jclian91 # place: Sanya,Hainan # time: 12:52 from create_table import Users from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker def update_data(): # 初始化資料庫連線 engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test",encoding="utf-8") # 建立DBSession型別 DBSession = sessionmaker(bind=engine) # 建立session物件 session = DBSession() # 資料更新,將Jack的place修改為CHN update_obj = session.query(Users).filter(Users.name=='Jack').update({"place":"CHN"}) session.commit() session.close() print("Update data successfully!") if __name__ == '__main__': update_data()
更新後的表格資料如下:
資料刪除
接著我們演示資料刪除。我們需要將Jack的記錄刪除,完整的Python程式碼(db_delete.py)如下:
# -*- coding: utf-8 -*- # author: Jclian91 # place: Sanya,Hainan # time: 12:52 from create_table import Users from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker def delete_data(): # 初始化資料庫連線 engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test",encoding="utf-8") # 建立DBSession型別 DBSession = sessionmaker(bind=engine) # 建立session物件 session = DBSession() # 資料更新,將Jack的記錄刪除 update_obj = session.query(Users).filter(Users.name=='Jack').delete() session.commit() session.close() print("Delete data successfully!") if __name__ == '__main__': delete_data()
執行程式後,表格中的資料如下:
本次分享到此結束,感謝大家閱讀~
本文的示例程式碼已經上傳至Github,網址為:https://github.com/percent4/ORM_test。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。