feat: token from env.

mckomaken · Aug 12, 2024 · da629d3 · da629d3
1 parent 05cc8a9
commit da629d3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/CommandLab.py b/CommandLab.py
@@ -3,6 +3,7 @@
 import logging.config
 from datetime import datetime
 from os import listdir
+import os
 
 import aiofiles
 import discord
@@ -187,6 +188,8 @@ async def close(self) -> None:
 
 if config.token == "FILE":
     config.token = open("..\\CMTK.txt", mode="r").read()
+if config.token == "ENV":
+    config.token = os.getenv("TOKEN")
 
 if __name__ == "__main__":
     asyncio.run(CommandLabBot.start(token=config.token))