improvements for aiopg example

This commit is contained in:
Anton Zhyrney 2016-12-11 14:04:24 +02:00
parent 154f8570f0
commit 721044b378

View File

@ -1,6 +1,5 @@
import os import os
import asyncio import asyncio
import datetime
import uvloop import uvloop
import aiopg import aiopg
@ -32,13 +31,14 @@ async def prepare_db():
""" """
async with pool.acquire() as conn: async with pool.acquire() as conn:
async with conn.cursor() as cur: async with conn.cursor() as cur:
await cur.execute("""CREATE TABLE sanic_poll ( await cur.execute('DROP TABLE IF EXISTS sanic_polls')
await cur.execute("""CREATE TABLE sanic_polls (
id integer primary key, id integer primary key,
question varchar(50), question varchar(50),
pub_date timestamp pub_date timestamp
);""") );""")
for i in range(0, 100): for i in range(0, 100):
await cur.execute("""INSERT INTO sanic_poll (id, question, pub_date) VALUES ({}, {}, now()) await cur.execute("""INSERT INTO sanic_polls (id, question, pub_date) VALUES ({}, {}, now())
""".format(i, i)) """.format(i, i))
@ -47,7 +47,7 @@ async def handle(request):
async with pool.acquire() as conn: async with pool.acquire() as conn:
async with conn.cursor() as cur: async with conn.cursor() as cur:
result = [] result = []
await cur.execute("SELECT question, pub_date FROM sanic_poll") await cur.execute("SELECT question, pub_date FROM sanic_polls")
async for row in cur: async for row in cur:
result.append({"question": row[0], "pub_date": row[1]}) result.append({"question": row[0], "pub_date": row[1]})
return json({"polls": result}) return json({"polls": result})