diff options
author | ziro <palembani@gmail.com> | 2020-08-06 07:37:19 +0000 |
---|---|---|
committer | ziro <palembani@gmail.com> | 2020-08-06 07:37:19 +0000 |
commit | 13447ea06137e9a2d856283037692837836e9b42 (patch) | |
tree | eab42aa5e4e3615162a7581c8b44f4c962654ea3 /cogs/webserver.py | |
parent | f797397950a1757a978962e912edae1a7263d3cb (diff) | |
download | steve-bot-13447ea06137e9a2d856283037692837836e9b42.tar steve-bot-13447ea06137e9a2d856283037692837836e9b42.tar.gz steve-bot-13447ea06137e9a2d856283037692837836e9b42.tar.bz2 steve-bot-13447ea06137e9a2d856283037692837836e9b42.tar.lz steve-bot-13447ea06137e9a2d856283037692837836e9b42.tar.xz steve-bot-13447ea06137e9a2d856283037692837836e9b42.tar.zst steve-bot-13447ea06137e9a2d856283037692837836e9b42.zip |
+ Steve Branch
Diffstat (limited to 'cogs/webserver.py')
-rwxr-xr-x | cogs/webserver.py | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/cogs/webserver.py b/cogs/webserver.py new file mode 100755 index 0000000..3061f8f --- /dev/null +++ b/cogs/webserver.py @@ -0,0 +1,57 @@ +from aiohttp import web +from discord.ext import commands, tasks +import discord +import os +import aiohttp +import json + +app = web.Application() +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 setup(bot): + bot.add_cog(Webserver(bot))
\ No newline at end of file |