63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
import os
|
|
import asyncio
|
|
|
|
import uvloop
|
|
import aiopg
|
|
|
|
from sanic import Sanic
|
|
from sanic.response import json
|
|
|
|
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
|
|
|
|
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)
|
|
loop = asyncio.get_event_loop()
|
|
|
|
|
|
async def get_pool():
|
|
return await aiopg.create_pool(connection)
|
|
|
|
app = Sanic(name=__name__)
|
|
pool = loop.run_until_complete(get_pool())
|
|
|
|
|
|
async def prepare_db():
|
|
""" Let's create some table and add some data
|
|
|
|
"""
|
|
async with pool.acquire() as conn:
|
|
async with conn.cursor() as cur:
|
|
await cur.execute('DROP TABLE IF EXISTS sanic_polls')
|
|
await cur.execute("""CREATE TABLE sanic_polls (
|
|
id serial primary key,
|
|
question varchar(50),
|
|
pub_date timestamp
|
|
);""")
|
|
for i in range(0, 100):
|
|
await cur.execute("""INSERT INTO sanic_polls
|
|
(id, question, pub_date) VALUES ({}, {}, now())
|
|
""".format(i, i))
|
|
|
|
|
|
@app.route("/")
|
|
async def handle(request):
|
|
async with pool.acquire() as conn:
|
|
async with conn.cursor() as cur:
|
|
result = []
|
|
await cur.execute("SELECT question, pub_date FROM sanic_polls")
|
|
async for row in cur:
|
|
result.append({"question": row[0], "pub_date": row[1]})
|
|
return json({"polls": result})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
loop.run_until_complete(prepare_db())
|
|
app.run(host="0.0.0.0", port=8100, workers=3, loop=loop)
|