starlette-plugins 0.0.6

Requirements

• python >= 3.7

Installation

starlette-plugins should be installed using pip:

pip install starlette-plugins

Usage

Let’s imagine that we need to write Starlette plugins for Peewee ORM.

from aiopeewee import db_url

from starlette_plugins import StarlettePlugin

class Peewee(StarlettePlugin):

    # Give your plugin a name
    name = 'db'

    # Prepare a default configuration
    config = {
        'url': 'sqlite+async:///db.sqlite',
        'connection_params': {},
    }

def __init__(self, app=None, **settings):
    super(Peewee, self).__init__(app, **settings)
    self.database = None

def setup(self, app, **settings):
    """Setup the plugin."""
    super(Peewee, self).setup(app, **settings)
    self.database = db_url.connect(self.config.url, **self.config.connection_params)

async def middleware(self, scope, receive, send, app):
    """An optional ASGI middleware."""
    try:
        await self.database.connect_async()
        await app(scope, receive, send)
    finally:
        await self.database.close_async()

async def shutdown(self, scope):
    """ The methods are supported: `on_startup`, `on_shutdown`."""
    if hasattr(self.database, 'close_all'):
        self.database.close_all()

Use the plugin

from starlette.applications import Starlette

db = Peewee(url='postgres+async://database')
db.setup(app)

Bug tracker

If you have any suggestions, bug reports or annoyances please report them to the issue tracker at https://github.com/klen/starlette-plugins/issues

Contributing

Development of the project happens at: https://github.com/klen/starlette-plugins

License

Licensed under a MIT license.