使用peewee模块操作数据库

1. 简介

Peewee 是一个 Python ORM（Object-Relational Mapping）库，支持 SQLite、MySQL、PostgreSQL 和 Cockroach 数据库。在 ORM 系统中，每个类都映射到底层数据库中的一个表。ORM可以处理这些问题，我们就不需要自己编写乏味的数据库接口代码，可以专注于对系统逻辑进行编程。

2. 定义模型

类名为表名，在类中定义字段名称、类型、是否有空值等信息

from peewee import *

db = SqliteDatabase('people.db')

class Person(Model):
    name = CharField()
    birthday = DateField(null=True)

    class Meta:
        database = db # This model uses the "people.db" database.

class Pet(Model):
    owner = ForeignKeyField(Person, backref='pets')
    name = CharField()
    animal_type = CharField()

    class Meta:
        database = db

如果操作已经存在的数据库表，我们可以通过以下命令生成模型文件
指定数据库名，host，用户名，密码，数据库，模板文件名

python -m pwiz -e mysql -H localhost -p 3306 -u user -P 123456 database > db.py

3. 表格操作

3.1 创建表格

db.create_tables([Person, Pet])

3.2 删除表格

db.drop_tables([Person, Pet])

4. 数据增删改查

4.1 新增数据

• 保存单条数据

from datetime import date
uncle_bob = Person(name='Bob', birthday=date(1960, 1, 15))
uncle_bob.save()

• 一次性插入多条数据

Person.insert_many([{'name':'Bob', 'birthday':'2002-01-01'},
                    {'name':'Tom', 'birthday':'1999-05007'}])
                  .execute()

4.2 按条件删除数据

删除Tweet实例中创建日期为一年以前的数据

Tweet.delete().where(Tweet.creation_date < one_year_ago).execute()

teewee中in的使用

Person.delect().where(Person.name.in_(['Bob','Tom'])).execute()

4.3 更新/修改数据

Table.update(**kwargs).where((Table.datadate==datadate) & (Table.account_id==account_id)).execute()

4.4 查询数据

grandma = Person.select().where(Person.name == 'Grandma L.').get()

等价于

grandma = Person.get(Person.name == 'Grandma L.')

将查询到的数据转为DataFrame:
先用dicts()方法将数据转化为字典格式，再转为DatFrame

df = pd.DataFrame(Table.select().where(Table.datadate==today).dicts())

df = pd.DataFrame(Table.select(Table.col1, Table.col2, Table.col3).where(Table.datadate==today).dicts())

更多内容可以查看官方文档