Deferred database initialization for flask-peewee.db.Database

app.py

from flask import Flask
from models import Base


app = Flask('tets')

db.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from flask_peewee.db import Database
from peewee import Model


class DeferredDatabase(Database):
    def __init__(self, app=None):
        if app:
            super(DeferredDatabase, self).__init__(app)
        else:
            class BaseModel(Model):
                pass
            self.Model = BaseModel

    def init_app(self, app):
        self.app = app

        self.load_database()
        self.register_handlers()

        class Meta:
            database = self.database

        self.Model.Meta = Meta

models.py

# database irrelevant model definitions


class User(Base):
    class Meta:
        db_table = 'test_user'
    name = CharField()