diff --git a/bot.py b/bot.py index eabfd9e..4c0f5e2 100644 --- a/bot.py +++ b/bot.py @@ -224,7 +224,10 @@ if __name__ == "__main__": os.getenv("BACKEND_SECRETKEY") ).run(os.getenv("DISCORD_TOKEN")) - ws.SERVICE_NAME = "discord_bot" - ws.START_AFTER_CONNECT = run - wsc = ws.WS(os.getenv("BACKEND_URL"), os.getenv("BACKEND_SECRETKEY")) - wsc.run() \ No newline at end of file + if os.getenv("START_WITH_WEBSOCKET"): + ws.SERVICE_NAME = "discord_bot" + ws.START_AFTER_CONNECT = run + wsc = ws.WS(os.getenv("BACKEND_URL"), os.getenv("BACKEND_SECRETKEY")) + wsc.run() + else: + run() \ No newline at end of file