1. 程式人生 > 資料庫 >python使用SQLAlchemy操作MySQL

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。

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