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)如下：

1. # -*- coding: utf-8 -*-
2. # author: Jclian91
3. # place: Sanya, Hainan
4. # time: 12:46
5.  
6. from sqlalchemy.dialects.mysql import INTEGER, VARCHAR
7. from sqlalchemy import Column
8. from sqlalchemy import create_engine
9. from sqlalchemy.ext.declarative import declarative_base
10. Base = declarative_base()
11.  
12.  
13. # users表結構
14. class Users(Base):
15. __tablename__ = 'users'
16.  
17. id = Column(INTEGER, primary_key=True)
18. name = Column(VARCHAR(256), nullable=False)
19. age = Column(INTEGER)
20. place = Column(VARCHAR(256), nullable=False)
21.  
22.  
23. def __init__(self, id, name, age, place):
24. self.id = id
25. self.name = name
26. self.age = age
27. self.place = place
28.  
29. def init_db():
30. engine = create_engine(
31. "mysql+pymysql://root:@localhost:3306/orm_test",
32. encoding= "utf-8",
33. echo=True
34. )
35. Base.metadata.create_all(engine)
36. print('Create table successfully!')
37.  
38. if __name__ == '__main__':
39. init_db()

運行后，生成的users表結構如下：

數據插入

接著我們插入幾條測試數據，完整的代碼(db_insert.py)如下：

1. # -*- coding: utf-8 -*-
2. # author: Jclian91
3. # place: Sanya, Hainan
4. # time: 12:52
5.  
6. from create_table import Users
7. from sqlalchemy import create_engine
8. from sqlalchemy.orm import sessionmaker
9.  
10. def insert_data():
11. # 初始化數據庫連接
12. engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
13. # 創建DBSession類型
14. DBSession = sessionmaker(bind=engine)
15.  
16. # 創建session對象
17. session = DBSession()
18. # 插入單條數據
19. # 創建新User對象
20. new_user = Users(id=1, name='Jack', age=25, place='USA')
21. # 添加到session
22. session.add(new_user)
23. # 提交即保存到數據庫
24. session.commit()
25.  
26. # 插入多條數據
27. user_list= [Users(id=2, name='Green', age=26, place='UK'),
28. Users(id=3, name='Alex', age=31, place='GER'),
29. Users(id=4, name='Chen', age=52, place='CHN'),
30. Users(id=5, name='Zhang', age=42, place='CHN')
31. ]
32. session.add_all(user_list)
33. session.commit()
34. # 關閉session
35. session.close()
36. print('insert into db successfully!')
37.  
38. if __name__ == '__main__':
39. insert_data()

運行程序后，表格中的數據如下：

數據查詢

接下來演示如何進行簡單的查詢。我們需要查詢place為CHN的人名，完整的Python代碼（db_query）如下：

1. # -*- coding: utf-8 -*-
2. # author: Jclian91
3. # place: Sanya, Hainan
4. # time: 12:52
5.  
6. from create_table import Users
7. from sqlalchemy import create_engine
8. from sqlalchemy.orm import sessionmaker
9.  
10. def query_data():
11. # 初始化數據庫連接
12. engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
13. # 創建DBSession類型
14. DBSession = sessionmaker(bind=engine)
15.  
16. # 創建session對象
17. session = DBSession()
18.  
19. # 查詢所有place是CHN的人名
20. # 創建Query查詢，filter是where條件
21. # 調用one()返回唯一行，如果調用all()則返回所有行:
22. users = session.query(Users).filter(Users.place == 'CHN').all()
23. print([use.name for use in users])
24. # 輸出：['Chen', 'Zhang']
25.  
26. # 或者用如下查詢
27. users = session.query(Users.name).filter(Users.place == 'CHN').all()
28. print(users)
29. # 輸出：[('Chen',), ('Zhang',)]
30.  
31. session.close()
32.  
33.  
34. if __name__ == '__main__':
35. query_data()

輸出結果如下:

['Chen', 'Zhang']
[('Chen',), ('Zhang',)]

數據更新

接下來我們演示數據更新。我們需要將Jack的place修改為CHN，完整的Python代碼（db_update）如下：

1. # -*- coding: utf-8 -*-
2. # author: Jclian91
3. # place: Sanya, Hainan
4. # time: 12:52
5.  
6. from create_table import Users
7. from sqlalchemy import create_engine
8. from sqlalchemy.orm import sessionmaker
9.  
10. def update_data():
11. # 初始化數據庫連接
12. engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
13. # 創建DBSession類型
14. DBSession = sessionmaker(bind=engine)
15.  
16. # 創建session對象
17. session = DBSession()
18.  
19. # 數據更新，將Jack的place修改為CHN
20. update_obj = session.query(Users).filter(Users.name=='Jack').update({"place":"CHN"})
21. session.commit()
22.  
23. session.close()
24. print("Update data successfully!")
25.  
26.  
27. if __name__ == '__main__':
28. update_data()

更新后的表格數據如下：

數據刪除

接著我們演示數據刪除。我們需要將Jack的記錄刪除，完整的Python代碼（db_delete.py）如下：

1. # -*- coding: utf-8 -*-
2. # author: Jclian91
3. # place: Sanya, Hainan
4. # time: 12:52
5.  
6. from create_table import Users
7. from sqlalchemy import create_engine
8. from sqlalchemy.orm import sessionmaker
9.  
10. def delete_data():
11. # 初始化數據庫連接
12. engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
13. # 創建DBSession類型
14. DBSession = sessionmaker(bind=engine)
15.  
16. # 創建session對象
17. session = DBSession()
18.  
19. # 數據更新，將Jack的記錄刪除
20. update_obj = session.query(Users).filter(Users.name=='Jack').delete()
21. session.commit()
22.  
23. session.close()
24. print("Delete data successfully!")
25.  
26.  
27. if __name__ == '__main__':
28. delete_data()

運行程序后，表格中的數據如下：

本次分享到此結束，感謝大家閱讀~

本文的示例代碼已經上傳至Github，網址為：https://github.com/percent4/ORM_test。

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持我們。