Python3 集成SQLAlchemy ORM 框架:实现简单增删改查功能封装

SQLAlchemy ORM 内部组件结构图及其说明:

组成部分:

  • Engine,框架的引擎
  • Connection Pooling ,数据库连接池
  • Dialect,选择连接数据库的DB API种类
  • Schema/Types,架构和类型
  • SQL Exprression Language,SQL表达式语言,将创建的类翻译为SQl语言

SQLAlchemy本身无法操作数据库,其必须以来pymsql等第三方插件,Dialect用于和数据API进行交流,根据配置文件的不同调用不同的数据库API,从而实现对数据库的操作:

基于pymysql连接
    mysql+pymysql://<username>:<password>@<host>/<dbname>[?<options>]

Python 基于SQLAlchemy 初始化表结构

import datetime
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime
from sqlalchemy.orm import relationship, sessionmaker

Base = declarative_base()
engine = create_engine(
    "mysql+pymysql://root:123456@127.0.0.1:3306/hotel",
    max_overflow=0,  # 超过连接池大小外最多创建的连接
    pool_size=5,  # 连接池大小
    pool_timeout=30,  # 池中没有线程最多等待的时间,否则报错
    pool_recycle=-1  # 多久之后对线程池中的线程进行一次连接的回收(重置)
)
Session = sessionmaker(engine)

# ##################### 单表示例 #########################
class Users(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(String(32), index=True)
    age = Column(Integer, default=18)
    email = Column(String(32), unique=True)
    ctime = Column(DateTime, default=datetime.datetime.now)
    extra = Column(Text, nullable=True)

    __table_args__ = (
        # UniqueConstraint('id', 'name', name='uix_id_name'),
        # Index('ix_id_name', 'name', 'extra'),
    )


class Hosts(Base):
    __tablename__ = 'hosts'

    id = Column(Integer, primary_key=True)
    name = Column(String(32), index=True)
    ctime = Column(DateTime, default=datetime.datetime.now)


# ##################### 一对多示例 #########################
class Hobby(Base):
    __tablename__ = 'hobby'
    id = Column(Integer, primary_key=True)
    caption = Column(String(50), default='篮球')


class Person(Base):
    __tablename__ = 'person'
    nid = Column(Integer, primary_key=True)
    name = Column(String(32), index=True, nullable=True)
    hobby_id = Column(Integer, ForeignKey("hobby.id"))

    # 与生成表结构无关,仅用于查询方便
    hobby = relationship("Hobby", backref='pers')


# ##################### 多对多示例 #########################

class Server2Group(Base):
    __tablename__ = 'server2group'
    id = Column(Integer, primary_key=True, autoincrement=True)
    server_id = Column(Integer, ForeignKey('server.id'))
    group_id = Column(Integer, ForeignKey('group.id'))


class Group(Base):
    __tablename__ = 'group'
    id = Column(Integer, primary_key=True)
    name = Column(String(64), unique=True, nullable=False)

    # 与生成表结构无关,仅用于查询方便
    servers = relationship('Server', secondary='server2group', backref='groups')


class Server(Base):
    __tablename__ = 'server'

    id = Column(Integer, primary_key=True, autoincrement=True)
    hostname = Column(String(64), unique=True, nullable=False)


def init_db():
    Base.metadata.create_all(engine)


def drop_db():
    Base.metadata.drop_all(engine)


if __name__ == '__main__':
    drop_db()
    init_db()

Python 基于SQLAlchemy 实现对表基础操作:增删改查

from orm import Users, Hosts, Session
# 原生SQL查询
from sqlalchemy.sql import text

# 全局session 实例化
session = Session()


# 简单插入
def insert():
    # ############# 执行ORM操作 #############
    obj1 = Users(name="alex1")
    session.add(obj1)
    # 提交事务
    session.commit()


# 批量插入
def batch_insert():
    session.add_all([
        Users(name="wupeiqi"),
        Users(name="alex"),
        Hosts(name="c1.com"),
    ])
    session.commit()


# 简单条件删除
def delete():
    session.query(Users).filter(Users.id > 2).delete()
    session.commit()


# 简单条件更新
def update():
    session.query(Users).filter(Users.id > 0).update({"name": "099"})
    session.query(Users).filter(Users.id > 0).update({Users.name: Users.name + "099"}, synchronize_session=False)
    session.query(Users).filter(Users.id > 0).update({"age": Users.age + 1}, synchronize_session="evaluate")
    session.commit()


# 简单查询
def select():
    r1 = session.query(Users).all()
    print(r1)
    r2 = session.query(Users.name.label('xx'), Users.age).all()
    print(r2)
    r3 = session.query(Users).filter(Users.name == "alex").all()
    print(r3)
    r4 = session.query(Users).filter_by(name='alex').all()
    print(r4)
    r5 = session.query(Users).filter_by(name='alex').first()
    print(r5)
    r6 = session.query(Users).filter(text("id<:value and name=:name")).params(value=224, name='fred').order_by(
        Users.id).all()
    print(r6)
    r7 = session.query(Users).from_statement(text("SELECT * FROM users where name=:name")).params(name='ed').all()
    print(r7)


# 全局session 关闭
session.close()

if __name__ == '__main__':
    # insert()
    # batch_insert()
    # delete()
    # update()
    select()

 

 

  • 1
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在使用 Flask 框架进行开发时,我们通常会使用 ORM 工具来进行数据库操作。常用的 ORM 工具有 SQLAlchemy 和 Peewee。这里以 SQLAlchemy 为例,介绍如何封装 MySQL 的增删查改操作。 首先,我们需要在 Flask 中初始化 SQLAlchemy: ```python from flask_sqlalchemy import SQLAlchemy app = Flask(__name__) app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://user:password@host/db_name' db = SQLAlchemy(app) ``` 接下来,我们可以定义一个基础的 Model 类,所有的数据模型都继承自它: ```python class BaseModel(db.Model): __abstract__ = True id = db.Column(db.Integer, primary_key=True) created_at = db.Column(db.DateTime, default=datetime.utcnow) updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow) ``` 这里的 BaseModel 类定义了一个 id 字段和两个时间戳字段,所有的数据模型都会继承这些字段。 接着,我们可以定义一个封装了增删查改操作的基础 Repository 类: ```python class BaseRepository: model = None @classmethod def add(cls, data): db.session.add(data) db.session.commit() @classmethod def delete(cls, data): db.session.delete(data) db.session.commit() @classmethod def find(cls, **kwargs): return cls.model.query.filter_by(**kwargs).first() @classmethod def find_all(cls, **kwargs): return cls.model.query.filter_by(**kwargs).all() @classmethod def update(cls, data): db.session.commit() ``` 这里的 BaseRepository 类定义了 add、delete、find、find_all 和 update 五个方法,分别对应了增、删、查、查全部和改操作。这些方法都是基于 SQLAlchemy 提供的 API 进行封装的。 最后,我们可以定义一个具体的 Repository 类,继承自 BaseRepository,并指定对应的 model: ```python class UserRepository(BaseRepository): model = User ``` 这里的 UserRepository 类继承了 BaseRepository 类,并指定了 model 为 User。这样,我们就可以通过 UserRepository 类来进行 User 数据模型的增删查改操作了。 ```python user = User(name='John') UserRepository.add(user) user = UserRepository.find(name='John') user.name = 'Tom' UserRepository.update(user) UserRepository.delete(user) users = UserRepository.find_all() ``` 以上就是一个简单的 MySQL 增删查改操作的封装示例。在实际开发中,我们可以根据具体的需求对 Repository 类进行拓展,并且可以结合 Flask 的 Blueprint 等功能来进行模块化的开发。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值