async def handle_thread_update(request):
data = await request.json()
thread_slug_or_id = request.match_info['slug_or_id']
message = data.get('message', False)
title = data.get('title', False)
connection_pool = request.app['pool']
connection = connection_pool.getconn()
cursor = connection.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
if not thread_slug_or_id.isdigit():
cursor.execute(Thread.query_get_thread_id(thread_slug_or_id))
result = cursor.fetchone()
if not result:
connection_pool.putconn(connection)
return web.json_response(
status=404, data={"message": "Can't find user with id #41\n"})
thread_id = result['id']
else:
thread_id = thread_slug_or_id
if not title and not message:
cursor.execute(Thread.query_get_thread_by_id(thread_id))
thread = cursor.fetchone()
thread['created'] = thread['created'].astimezone().isoformat()
if thread['slug'] == 'NULL':
thread.pop('slug')
connection_pool.putconn(connection)
return web.json_response(status=200, data=thread)
try:
cursor.execute(Thread.query_update_thread(thread_id, message, title))
connection.commit()
except psycopg2.Error:
connection.rollback()
connection_pool.putconn(connection)
return web.json_response(
status=404, data={"message": "Can't find user with id #42\n"})
else:
cursor.execute(Thread.query_get_thread_by_id(thread_id))
thread = cursor.fetchone()
connection_pool.putconn(connection)
if not thread:
return web.json_response(
status=404, data={"message": "Can't find user with id #42\n"})
thread['created'] = thread['created'].astimezone().isoformat()
if thread['slug'] == 'NULL':
thread.pop('slug')
return web.json_response(status=200, data=thread)
async def handle_forum_create(request):
data = await request.json()
forum = request.match_info['slug']
forum_slug = data.get('forum', False)
if not forum_slug:
thread = Thread(**data, forum=forum)
else:
thread = Thread(**data)
connection_pool = request.app['pool']
connection = connection_pool.getconn()
cursor = connection.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
thread_id = None
try:
cursor.execute(thread.query_create_thread())
thread_id = cursor.fetchone()['id']
connection.commit()
except psycopg2.Error as e:
connection.rollback()
cursor.execute(Thread.query_get_thread_by_slug(data['slug']))
result = cursor.fetchone()
connection_pool.putconn(connection)
if result:
result['created'] = result['created'].astimezone().isoformat()
return web.json_response(status=409, data=result)
else:
return web.json_response(status=404, data={})
else:
cursor.execute(Thread.query_get_thread_by_id(thread_id))
result = cursor.fetchone()
result['created'] = result['created'].astimezone().isoformat()
if result['slug'] == 'NULL':
result.pop('slug')
connection_pool.putconn(connection)
return web.json_response(status=201, data=result)