How to create all tables defined in models using peewee

Question

I define a lot of model classes using peewee. ClassName.create_table() can generate the table,but only one table. How could I create all tables using a single statement?

coleifer · Answer 1 · 2016-03-31T06:11:13.947

22

Peewee has a helper that will create the tables in the correct order, but you still need to explicitly pass in all the models:

from peewee import *
db = SqliteDatabase(':memory:')
db.create_tables([ModelA, ModelB, ModelC])

edited Mar 31 '16 at 06:11

answered Feb 07 '14 at 15:40

coleifer

24,887
6
60
75

1

Maybe this function's docstring could appear in the API docs? – Cilyan Feb 07 '14 at 15:50

score 16 · Answer 2 · edited May 23 '17 at 11:47

16

An update for Python 3 (and everyone who comes across this question via Google like I did). If you have all models based on the main peewee Model class you can simply use:

import peewee    
models = peewee.Model.__subclasses__()

Credit to this question for the idea. They also go into more detail on how to get it to work recursively if your model is more complex.

edited May 23 '17 at 11:47

Community

1
1

answered May 06 '17 at 13:14

ninetynine

327
2
10

2

Hands down best solution! – SkyWalker Mar 23 '20 at 16:03
Fails in some instances, e.g. if using the `SqliteQueueDatabase` extension, you get an `ImproperlyConfigured` error for the `VirtualModel` model, which is used internally but not actually part of your database. – fdmillion Jan 26 '22 at 07:41

score 10 · Accepted Answer · answered Feb 07 '14 at 16:13

10

Extending coleifer's answer with the assumption that all tables are grouped in one module:

import inspect
import peewee
import tables
models = [
    obj for name, obj in inspect.getmembers(
        tables, lambda obj: type(obj) == type and issubclass(obj, peewee.Model)
    )
]
peewee.create_model_tables(models)

answered Feb 07 '14 at 16:13

Cilyan

7,883
1
29
37

Seems create_model_tables is deprecated in newer peewee versions. – Reck Oct 30 '18 at 17:40
http://docs.peewee-orm.com/en/latest/peewee/example.html#creating-tables – Reck Oct 30 '18 at 17:41
Possibly, I no longer use peewee for a very long time now. – Cilyan Nov 05 '18 at 17:39

score 2 · Answer 4 · answered Jun 30 '14 at 06:34

2

This snipnet will create all tables which objects are defined in the current module:

import sys

for cls in sys.modules[__name__].__dict__.values():
    try:
        if BaseModel in cls.__bases__:
            cls.create_table()
    except:
        pass

answered Jun 30 '14 at 06:34

Mathieu Rodic

6,637
2
43
49

score 2 · Answer 5 · answered Mar 27 '16 at 12:09

2

for cls in globals().values():
    if type(cls) == peewee.BaseModel:
        try:
            cls.create_table()
        except peewee.OperationalError as e:
            print(e)

answered Mar 27 '16 at 12:09

thinker3

12,771
5
30
36

score 1 · Answer 6 · answered Oct 03 '18 at 12:26

def create_all_tables():
    for cls in sys.modules[__name__].__dict__.values():
        if hasattr(cls, '__bases__') and issubclass(cls, peewee.Model):
            if cls is not BaseModel:
                cls.create_table()

This also works for ManyToManyField's get_through_model().

score 0 · Answer 7 · answered Oct 21 '21 at 16:26

I found the existing answers did not support the BaseModel pattern recommended in the docs. I created a helper function to recursively find Models

def all_subclasses(base: type) -> list[type]:
    return [
        cls
        for sub in base.__subclasses__()
        for cls in [sub] + all_subclasses(sub)
    ]

I then filtered out base models, by prefixing them with an underscore.

class _BaseModel(Model):
    class Meta:
        database = db

models = [
    sub for sub in all_subclasses(Model)
    if not sub.__name__.startswith('_')
]

db.create_tables(models)

How to create all tables defined in models using peewee

7 Answers7

Linked