diff options
author | Mango0x45 <thomasvoss@live.com> | 2020-09-12 13:13:29 +0000 |
---|---|---|
committer | Mango0x45 <thomasvoss@live.com> | 2020-09-12 13:13:29 +0000 |
commit | 0baa36497519711953603db640bf099db19af826 (patch) | |
tree | a5e196ee48d6552894c683db2272de16f8d77c41 /cogs/webserver.py | |
parent | fed0c6b1767f3eddbabed9cc28dd0f5615889811 (diff) | |
download | steve-bot-0baa36497519711953603db640bf099db19af826.tar steve-bot-0baa36497519711953603db640bf099db19af826.tar.gz steve-bot-0baa36497519711953603db640bf099db19af826.tar.bz2 steve-bot-0baa36497519711953603db640bf099db19af826.tar.lz steve-bot-0baa36497519711953603db640bf099db19af826.tar.xz steve-bot-0baa36497519711953603db640bf099db19af826.tar.zst steve-bot-0baa36497519711953603db640bf099db19af826.zip |
Black is neat
Diffstat (limited to 'cogs/webserver.py')
-rwxr-xr-x | cogs/webserver.py | 87 |
1 files changed, 44 insertions, 43 deletions
diff --git a/cogs/webserver.py b/cogs/webserver.py index d9c92cf..757c8c5 100755 --- a/cogs/webserver.py +++ b/cogs/webserver.py @@ -11,48 +11,49 @@ routes = web.RouteTableDef() class Webserver(commands.Cog): - def __init__(self, bot): - self.bot = bot - self.web_server.start() - - @routes.get('/') - async def welcome(request): - return web.Response(text="Hello, world") - - @routes.get('/keys') - async def get_keys(request): - with open('./api_keys.json') as f: - keys = json.load(f) - return web.json_response(keys) - - @routes.post('/keys') - async def post_keys(request): - data = await request.post() - try: - discord_id = data['discord_id'] - src_id = data['src_id'] - except KeyError: - return 400 - with open('./api_keys.json', 'r') as f: - keys = json.load(f) - keys[discord_id] = src_id - with open('./api_keys.json', 'w') as f: - json.dump(keys, f, indent=4) - return web.json_response(keys) - - self.webserver_port = os.environ.get('PORT', 5000) - app.add_routes(routes) - - @tasks.loop() - async def web_server(self): - runner = web.AppRunner(app) - await runner.setup() - site = web.TCPSite(runner, host='0.0.0.0', port=self.webserver_port) - await site.start() - - @web_server.before_loop - async def web_server_before_loop(self): - await self.bot.wait_until_ready() + def __init__(self, bot): + self.bot = bot + self.web_server.start() + + @routes.get("/") + async def welcome(request): + return web.Response(text="Hello, world") + + @routes.get("/keys") + async def get_keys(request): + with open("./api_keys.json") as f: + keys = json.load(f) + return web.json_response(keys) + + @routes.post("/keys") + async def post_keys(request): + data = await request.post() + try: + discord_id = data["discord_id"] + src_id = data["src_id"] + except KeyError: + return 400 + with open("./api_keys.json", "r") as f: + keys = json.load(f) + keys[discord_id] = src_id + with open("./api_keys.json", "w") as f: + json.dump(keys, f, indent=4) + return web.json_response(keys) + + self.webserver_port = os.environ.get("PORT", 5000) + app.add_routes(routes) + + @tasks.loop() + async def web_server(self): + runner = web.AppRunner(app) + await runner.setup() + site = web.TCPSite(runner, host="0.0.0.0", port=self.webserver_port) + await site.start() + + @web_server.before_loop + async def web_server_before_loop(self): + await self.bot.wait_until_ready() + def setup(bot): - bot.add_cog(Webserver(bot))
\ No newline at end of file + bot.add_cog(Webserver(bot)) |