轉(zhuǎn)自：數(shù)據(jù)管道

作者：言淦，一枚喜歡淦代碼的碼農(nóng)，'言淦說(shuō)'主理人

背景

其實(shí)一開(kāi)始用的是pymysql，但是發(fā)現(xiàn)維護(hù)比較麻煩，還存在代碼注入的風(fēng)險(xiǎn)，所以就干脆直接用ORM框架。

ORM即Object Relational Mapper，可以簡(jiǎn)單理解為數(shù)據(jù)庫(kù)表和Python類之間的映射，通過(guò)操作Python類，可以間接操作數(shù)據(jù)庫(kù)。

Python的ORM框架比較出名的是SQLAlchemy和Peewee，這里不做比較，只是單純講解個(gè)人對(duì)SQLAlchemy的一些使用，希望能給各位朋友帶來(lái)幫助。

• sqlalchemy版本: 1.3.15

• pymysql版本: 0.9.3

• mysql版本: 5.7

初始化工作

一般使用ORM框架，都會(huì)有一些初始化工作，比如數(shù)據(jù)庫(kù)連接，定義基礎(chǔ)映射等。

以MySQL為例，創(chuàng)建數(shù)據(jù)庫(kù)連接只需要傳入DSN字符串即可。其中echo表示是否輸出對(duì)應(yīng)的sql語(yǔ)句，對(duì)調(diào)試比較有幫助。

個(gè)人設(shè)計(jì)

對(duì)于我個(gè)人而言，引進(jìn)ORM框架時(shí)，我的項(xiàng)目會(huì)參考MVC模式做以下設(shè)計(jì)。其中model存儲(chǔ)的是一些數(shù)據(jù)庫(kù)模型，即數(shù)據(jù)庫(kù)表映射的Python類；model_op存儲(chǔ)的是每個(gè)模型對(duì)應(yīng)的操作，即增刪查改；調(diào)用方（如main.py）執(zhí)行數(shù)據(jù)庫(kù)操作時(shí)，只需要調(diào)用model_op層，并不用關(guān)心model層，從而實(shí)現(xiàn)解耦。

├── main.py
├── model
│ ├── __init__.py
│ ├── base_model.py
│ ├── ddl.sql
│ └── py_orm_model.py
└── model_op
├── __init__.py
└── py_orm_model_op.py

映射聲明(Model介紹)

舉個(gè)栗子，如果我們有這樣一張測(cè)試表

在ORM框架中，映射的結(jié)果就是下文這個(gè)Python類

# py_orm_model.py
from .base_model import Base
from sqlalchemy import Column, Integer, String, TIMESTAMP, text, JSON
class PyOrmModel(Base):__tablename__ = 'py_orm'
id = Column(Integer, autoincrement=True, primary_key=True, comment='唯一id')
name = Column(String(255), able=False, default='', comment='名稱')
attr = Column(JSON, able=False, comment='屬性')
ct = Column(TIMESTAMP, able=False, server_default=text('CURRENT_TIMESTAMP'), comment='創(chuàng)建時(shí)間')
ut = Column(TIMESTAMP, able=False, server_default=text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), comment='更新時(shí)間')

首先，我們可以看到PyOrmModel繼承了Base類，該類是sqlalchemy提供的一個(gè)基類，會(huì)對(duì)我們聲明的Python類做一些檢查，我將其放在base_model中。

其次，每個(gè)Python類都必須包含__tablename__屬性，不然無(wú)法找到對(duì)應(yīng)的表。

第三，關(guān)于數(shù)據(jù)表的創(chuàng)建有兩種方式，第一種當(dāng)然是手動(dòng)在MySQL中創(chuàng)建，只要你的Python類定義沒(méi)有問(wèn)題，就可以正常操作；第二種是通過(guò)orm框架創(chuàng)建，比如下面

# main.py
# 注意這里的導(dǎo)入路徑，Base創(chuàng)建表時(shí)會(huì)尋找繼承它的子類，如果路徑不對(duì)，則無(wú)法創(chuàng)建成功from sqlachlemy_lab import Base, engine
if __name__ == '__main__':
Base.metadata.create_all(engine)

創(chuàng)建效果：

第四，關(guān)于字段屬性

• 1.primary_key和autoincrement比較好理解，就是MySQL的主鍵和遞增屬性。

• 2.如果是int類型，不需要指定長(zhǎng)度，而如果是varchar類型，則必須指定。

• 3.able對(duì)應(yīng)的就是MySQL中的 和 NOT

• 4.關(guān)于default和server_default: default代表的是ORM框架層面的默認(rèn)值，即插入的時(shí)候如果該字段未賦值，則會(huì)使用我們定義的默認(rèn)值；server_default代表的是數(shù)據(jù)庫(kù)層面的默認(rèn)值，即DDL語(yǔ)句中的default關(guān)鍵字。

Session介紹

在SQLAlchemy的文檔中提到，數(shù)據(jù)庫(kù)的增刪查改是通過(guò)session來(lái)執(zhí)行的。

>>> from sqlalchemy.orm import sessionmaker
>>> Session = sessionmaker(bind=engine)>>> session = Session>>> orm = PyOrmModel(id=1, name='test', attr={})
>>> session.add(orm)>>> session.commit>>> session.close

如上，我們可以看到，對(duì)于每一次操作，我們都需要對(duì)session進(jìn)行獲取，提交和釋放。這樣未免過(guò)于冗余和麻煩，所以我們一般會(huì)進(jìn)行一層封裝。

1.采用上下文管理器的方式，處理session的異?；貪L和關(guān)閉，這部分與所參考的文章是幾乎一致的。

2.在PyOrmModel中增加兩個(gè)方法，用于model和dict之間的轉(zhuǎn)換

class PyOrmModel(Base):
...@staticmethod
def fields:
return ['id', 'name', 'attr']
@staticmethod
def to_json(model):
fields = PyOrmModel.fieldsjson_data = {}for field in fields:
json_data[field] = model.__getattribute__(field)return json_data
@staticmethod
def from_json(data: dict):
fields = PyOrmModel.fieldsmodel = PyOrmModelfor field in fields:
if field in data:
model.__setattr__(field, data[field])
return model

3.數(shù)據(jù)庫(kù)操作的封裝，與參考的文章不同，我是直接調(diào)用了session，從而使調(diào)用方不需要關(guān)注model層，減少耦合。

4.調(diào)用方

# main.py
from sqlachlemy_lab.model_op import PyOrmModelOp
if __name__ == '__main__':
PyOrmModelOp.save_data({'id': 1, 'name': 'test', 'attr': {}})

完整代碼請(qǐng)參見(jiàn)：

https://github.com/yangancode/python_lab/tree/master/sqlachlemy_lab