系统城装机大师 - 固镇县祥瑞电脑科技销售部宣传站!

当前位置:首页 > 数据库 > MsSql > 详细页面

python使用SQLAlchemy操作MySQL

时间:2020-01-02来源:系统城作者:电脑系统城

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。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持我们。

分享到:

相关信息

  • SQL Server 数据库中的收缩数据库和文件操作

    收缩数据文件通过将数据页从文件末尾移动到更靠近文件开头的未占用的空间来恢复空间,在文件末尾创建足够的空间后,可取消对文件末尾的数据页的分配并将它们返回给文件系统,本文给大家介绍SQL Server 数据库中的收缩数据...

    2023-10-27

  • dbeaver配置SQL server连接实现

    一、需要java jdk环境,我用的比较新 二、dbeaver新建连接SQL server...

    2023-10-27

系统教程栏目

栏目热门教程

人气教程排行

站长推荐

热门系统下载