62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
|
import asyncio
|
||
|
|
||
|
from aiohttp import web
|
||
|
from aiohttp_session import setup as setup_session
|
||
|
from aiohttp_session.redis_storage import RedisStorage
|
||
|
from aiohttp_security import setup as setup_security
|
||
|
from aiohttp_security import SessionIdentityPolicy
|
||
|
from aiopg.sa import create_engine
|
||
|
from aioredis import create_pool
|
||
|
|
||
|
|
||
|
from demo.db_auth import DBAuthorizationPolicy
|
||
|
from demo.handlers import Web
|
||
|
|
||
|
|
||
|
@asyncio.coroutine
|
||
|
def init(loop):
|
||
|
redis_pool = yield from create_pool(('localhost', 6379))
|
||
|
db_engine = yield from create_engine(user='aiohttp_security',
|
||
|
password='aiohttp_security',
|
||
|
database='aiohttp_security',
|
||
|
host='127.0.0.1')
|
||
|
app = web.Application(loop=loop)
|
||
|
app.db_engine = db_engine
|
||
|
setup_session(app, RedisStorage(redis_pool))
|
||
|
setup_security(app,
|
||
|
SessionIdentityPolicy(),
|
||
|
DBAuthorizationPolicy(db_engine))
|
||
|
|
||
|
web_handlers = Web()
|
||
|
web_handlers.configure(app)
|
||
|
|
||
|
handler = app.make_handler()
|
||
|
srv = yield from loop.create_server(handler, '127.0.0.1', 8080)
|
||
|
print('Server started at http://127.0.0.1:8080')
|
||
|
return srv, app, handler
|
||
|
|
||
|
|
||
|
@asyncio.coroutine
|
||
|
def finalize(srv, app, handler):
|
||
|
sock = srv.sockets[0]
|
||
|
app.loop.remove_reader(sock.fileno())
|
||
|
sock.close()
|
||
|
|
||
|
yield from handler.finish_connections(1.0)
|
||
|
srv.close()
|
||
|
yield from srv.wait_closed()
|
||
|
yield from app.finish()
|
||
|
|
||
|
|
||
|
def main():
|
||
|
loop = asyncio.get_event_loop()
|
||
|
srv, app, handler = loop.run_until_complete(init(loop))
|
||
|
try:
|
||
|
loop.run_forever()
|
||
|
except KeyboardInterrupt:
|
||
|
loop.run_until_complete((finalize(srv, app, handler)))
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
main()
|