Merge pull request #6 from PyBotDevs/replit-deployment

Get YAToDB ready for replit deployment

Author: notsniped

config/startup.json

@@ -3,7 +3,7 @@
         "owner_id": ""
     },
     "auth": {
-        "token": "",
+        "token": null,
         "public_key": ""
     }
 }

main.py

@@ -4,6 +4,7 @@
 import time
 import os, os.path
 import utils.auth
+import utils.ping
 from random import choice
 from discord import ApplicationContext, option
 from discord.ext import commands
@@ -24,6 +25,8 @@
 async def on_ready():
     print(f"[client] Bot client successfully signed into API. ({round(time.time()) - start_time}ms)")
     print(f"[client] Logged in as \"{client.user.name}\".")
+    print("[utils/ping] Starting web pinging server...")
+    utils.ping.host()
 
 # Functions
 @client.slash_command(

utils/auth.py

@@ -15,6 +15,7 @@
 # Functions
 def get_token() -> str:
     """Returns the bot token, if provided in the `startup.json` file."""
+    if config["auth"]["token"] == None: return os.environ["TOKEN"]
     return config["auth"]["token"]
 
 def get_public_key() -> str:

utils/ping.py

@@ -0,0 +1,20 @@
+from flask import Flask
+from threading import Thread
+
+app = Flask('')
+
+@app.route('/')
+def main():
+  return """
+YAToDB is online.
+""", 200
+
+def run():
+    app.run(host="0.0.0.0", port="8080")
+
+def host(*, no_thread: bool = False):
+    if no_thread: run()
+    else:
+        server = Thread(target=run)
+        server.daemon = True
+        server.start()