67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
""" To run this example you need additional aiopg package
|
|
|
|
"""
|
|
import os
|
|
import asyncio
|
|
import datetime
|
|
|
|
import uvloop
|
|
from aiopg.sa import create_engine
|
|
import sqlalchemy as sa
|
|
|
|
from sanic import Sanic
|
|
from sanic.response import json
|
|
|
|
database_name = os.environ['DATABASE_NAME']
|
|
database_host = os.environ['DATABASE_HOST']
|
|
database_user = os.environ['DATABASE_USER']
|
|
database_password = os.environ['DATABASE_PASSWORD']
|
|
|
|
connection = 'postgres://{0}:{1}@{2}/{3}'.format(database_user,
|
|
database_password,
|
|
database_host,
|
|
database_name)
|
|
|
|
metadata = sa.MetaData()
|
|
|
|
polls = sa.Table('sanic_polls', metadata,
|
|
sa.Column('id', sa.Integer, primary_key=True),
|
|
sa.Column('question', sa.String(50)),
|
|
sa.Column("pub_date", sa.DateTime))
|
|
|
|
|
|
app = Sanic(name=__name__)
|
|
|
|
|
|
async def prepare_db(app, loop):
|
|
""" Let's add some data
|
|
|
|
"""
|
|
async with create_engine(connection) as engine:
|
|
async with engine.acquire() as conn:
|
|
await conn.execute('DROP TABLE IF EXISTS sanic_polls')
|
|
await conn.execute("""CREATE TABLE sanic_polls (
|
|
id serial primary key,
|
|
question varchar(50),
|
|
pub_date timestamp
|
|
);""")
|
|
for i in range(0, 100):
|
|
await conn.execute(
|
|
polls.insert().values(question=i,
|
|
pub_date=datetime.datetime.now())
|
|
)
|
|
|
|
|
|
@app.route("/")
|
|
async def handle(request):
|
|
async with create_engine(connection) as engine:
|
|
async with engine.acquire() as conn:
|
|
result = []
|
|
async for row in conn.execute(polls.select()):
|
|
result.append({"question": row.question, "pub_date": row.pub_date})
|
|
return json({"polls": result})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=8000, before_start=prepare_db)
|