Peewee 使用手册

2021-03-04 10:33:07 浏览数 (1)

文章目录

    • Peewee 是什么
    • 如何使用
      • 基本步骤
      • 连接数据库
      • 创建数据表
      • 插入一条数据记录
      • 获取条件过滤后的数据记录
      • 更新数据记录
      • 查询单条数据记录
      • 其他要点

Peewee 是什么

Peewee 即 Python OMR 框架之一。

如何使用

基本步骤

  • 通过 pip3 下载 peewee
  • 新建 models.py 模型文件
  • 在 models.py 中加入以下基础代码
  • 运行 models.py,在 mysql 中生成 new_record 表

连接数据库

代码语言:javascript复制
from peewee import *
from settings import DATABASES

# 连接数据库
MYSQL_DB = MySQLDatabase(
    host=DATABASES['DEFAULT']['HOST'],
    port=DATABASES['DEFAULT']['PORT'],
    user=DATABASES['DEFAULT']['USER'],
    passwd=DATABASES['DEFAULT']['PASSWORD'],
    database=DATABASES['DEFAULT']['NAME'],
)

class BaseModel(Model):
    """基础模型类"""

    class Meta:
        database = MYSQL_DB

创建数据表

代码语言:javascript复制
class NewRecord(BaseModel):
    """NewRecord 模型类"""

    field_1 = CharField(max_length=10, null=True)
    field_2 = CharField(max_length=100, null=True)
    field_3 = IntegerField(default=0)
    field_4 = TextField()

    class Meta:
        db_table = 'new_record'

if __name__ == '__main__':
    # 创建表
    NewRecord.create_table()

插入一条数据记录

代码语言:javascript复制
new_record = NewRecord()
new_record.field_1 = "field_1"
new_record.field_2 = "field_2"
new_record.field_3 = 0
new_record.field_4 = "field_4"
new_record.save()

获取条件过滤后的数据记录

代码语言:javascript复制
from modules.models import NewRecord

def get_data_list():
    """获取数据列表

    :return:
    """

    data_list = []
    new_records = NewRecord.select().where(
        NewRecord.status == 0).order_by(
        NewRecord.id)

    for new_record in new_records:
        _data = {
            'field_1': new_record.field_1,
            'field_2': new_record.field_2,
            'field_3': new_record.field_3
        }
        data_list.append(_data)

    return data_list

更新数据记录

代码语言:javascript复制
def update_new_record_field_3(
        field_1, field_2,
        field_3):
    """更新特定查询数据字段 field_3

    :param field_1: 待更新 field_1
    :param field_2: 待更新 field_2
    :param field_3: 待更新 field_3
    :return:
    """

    update_instance = NewRecord.update(
        {
            NewRecord.field_3: field_3
        }
    ).where(NewRecord.field_1 == field_1
            and NewRecord.field_2 == field_2)
    update_instance.execute()

查询单条数据记录

代码语言:javascript复制
_record = NewRecord.get(NewRecord.field_1 == field_1 and NewRecord.field_2 == field_2)
print(_record.field_1, _record.field_2)

其他要点

  • 联合约束的创建
代码语言:javascript复制
class Meta:
        db_table = 'new_record'
        constraints = [SQL('UNIQUE KEY(field_1, field_2)')]

0 人点赞