update asyncpg example

This commit is contained in:
Raphael Deem 2017-03-07 19:31:44 -08:00
parent 19592e8eea
commit 694207a86d

View File

@ -5,14 +5,14 @@ import os
import asyncio import asyncio
import uvloop import uvloop
from asyncpg import create_pool from asyncpg import connect
from sanic import Sanic from sanic import Sanic
from sanic.response import json from sanic.response import json
DB_CONFIG = { DB_CONFIG = {
'host': '<host>', 'host': '<host>',
'user': '<username>', 'user': '<user>',
'password': '<password>', 'password': '<password>',
'port': '<port>', 'port': '<port>',
'database': '<database>' 'database': '<database>'
@ -22,8 +22,7 @@ def jsonify(records):
""" """
Parse asyncpg record response into JSON format Parse asyncpg record response into JSON format
""" """
return [{key: value for key, value in return [dict(r.items()) for r in records]
zip(r.keys(), r.values())} for r in records]
app = Sanic(__name__) app = Sanic(__name__)
@ -32,26 +31,22 @@ async def create_db(app, loop):
""" """
Create some table and add some data Create some table and add some data
""" """
async with create_pool(**DB_CONFIG) as pool: conn = await connect(**DB_CONFIG)
async with pool.acquire() as connection: await conn.execute('DROP TABLE IF EXISTS sanic_post')
async with connection.transaction(): await conn.execute("""CREATE TABLE sanic_post (
await connection.execute('DROP TABLE IF EXISTS sanic_post')
await connection.execute("""CREATE TABLE sanic_post (
id serial primary key, id serial primary key,
content varchar(50), content varchar(50),
post_date timestamp post_date timestamp);"""
);""") )
for i in range(0, 100): for i in range(0, 100):
await connection.execute(f"""INSERT INTO sanic_post await conn.execute(f"""INSERT INTO sanic_post
(id, content, post_date) VALUES ({i}, {i}, now())""") (id, content, post_date) VALUES ({i}, {i}, now())""")
@app.route("/") @app.route("/")
async def handler(request): async def handler(request):
async with create_pool(**DB_CONFIG) as pool: conn = await connect(**DB_CONFIG)
async with pool.acquire() as connection: results = await conn.fetch('SELECT * FROM sanic_post')
async with connection.transaction():
results = await connection.fetch('SELECT * FROM sanic_post')
return json({'posts': jsonify(results)}) return json({'posts': jsonify(results)})