快捷導(dǎo)航

python使用SQLAlchemy操作MySQL

更新時間：2020年01月02日 14:41:30 作者：jclian91

這篇文章主要介紹了python使用SQLAlchemy操作MySQL，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

SQLAlchemy是Python編程語言下的一款開源軟件，提供了SQL工具包及對象關(guān)系映射（ORM）工具，使用MIT許可證發(fā)行。SQLAlchemy首次發(fā)行于2006年2月，并迅速地在Python社區(qū)中最廣泛使用的ORM工具之一，不亞于Django的ORM框架。

本文將介紹如何使用SQLAlchemy操作MySQL，完成基礎(chǔ)的表創(chuàng)建，表格數(shù)據(jù)的新增、查詢、修改、刪除（CRUD）等操作。

首先我們需要確認(rèn)當(dāng)前的Python環(huán)境下已經(jīng)安裝sqlalchemy和pymysql模塊。

新建表格

我們使用本地的MySQL數(shù)據(jù)庫，數(shù)據(jù)庫為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表結(jié)構(gòu)
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, name, age, 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()

運(yùn)行后，生成的users表結(jié)構(gòu)如下：

數(shù)據(jù)插入

接著我們插入幾條測試數(shù)據(jù)，完整的代碼(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():
  # 初始化數(shù)據(jù)庫連接
  engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
  # 創(chuàng)建DBSession類型
  DBSession = sessionmaker(bind=engine)

  # 創(chuàng)建session對象
  session = DBSession()
  # 插入單條數(shù)據(jù)
  # 創(chuàng)建新User對象
  new_user = Users(id=1, name='Jack', age=25, place='USA')
  # 添加到session
  session.add(new_user)
  # 提交即保存到數(shù)據(jù)庫
  session.commit()

  # 插入多條數(shù)據(jù)
  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()
  # 關(guān)閉session
  session.close()
  print('insert into db successfully!')

if __name__ == '__main__':
  insert_data()

運(yùn)行程序后，表格中的數(shù)據(jù)如下：

數(shù)據(jù)查詢

接下來演示如何進(jìn)行簡單的查詢。我們需要查詢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():
  # 初始化數(shù)據(jù)庫連接
  engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
  # 創(chuàng)建DBSession類型
  DBSession = sessionmaker(bind=engine)

  # 創(chuàng)建session對象
  session = DBSession()

  # 查詢所有place是CHN的人名
  # 創(chuàng)建Query查詢，filter是where條件
  # 調(diào)用one()返回唯一行，如果調(diào)用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()

輸出結(jié)果如下:

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

數(shù)據(jù)更新

接下來我們演示數(shù)據(jù)更新。我們需要將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():
  # 初始化數(shù)據(jù)庫連接
  engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
  # 創(chuàng)建DBSession類型
  DBSession = sessionmaker(bind=engine)

  # 創(chuàng)建session對象
  session = DBSession()

  # 數(shù)據(jù)更新，將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()

更新后的表格數(shù)據(jù)如下：

數(shù)據(jù)刪除

接著我們演示數(shù)據(jù)刪除。我們需要將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():
  # 初始化數(shù)據(jù)庫連接
  engine = create_engine("mysql+pymysql://root:@localhost:3306/orm_test", encoding="utf-8")
  # 創(chuàng)建DBSession類型
  DBSession = sessionmaker(bind=engine)

  # 創(chuàng)建session對象
  session = DBSession()

  # 數(shù)據(jù)更新，將Jack的記錄刪除
  update_obj = session.query(Users).filter(Users.name=='Jack').delete()
  session.commit()

  session.close()
  print("Delete data successfully!")


if __name__ == '__main__':
  delete_data()

運(yùn)行程序后，表格中的數(shù)據(jù)如下：