klemek/miniscord
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed dotenv loading

Browse Source
This commit is contained in:
klemek
2020-11-06 21:02:07 +01:00
parent fe4b600bb8
commit 3eb2b4f9af
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
miniscord/_bot.py
+5 -3
View File
@@ -7,8 +7,9 @@ import re
import asyncio import asyncio
import random import random
import os import os
from os import path
from datetime import datetime from datetime import datetime
from dotenv import load_dotenv, find_dotenv from dotenv import load_dotenv
from ._utils import sanitize_input, parse_arguments from ._utils import sanitize_input, parse_arguments
@@ -210,10 +211,11 @@ class Bot(object):
def start(self): def start(self):
logging.info(f"Current PID: {os.getpid()}") logging.info(f"Current PID: {os.getpid()}")
env_file_found = load_dotenv(find_dotenv()) env_file = path.join(os.getcwd(), ".env")
env_file_found = load_dotenv(env_file)
self.__token = os.getenv(self.token_env_var) self.__token = os.getenv(self.token_env_var)
if self.__token is None: if self.__token is None:
if env_file_found: if path.exists(env_file) and env_file_found:
raise EnvironmentError(f"No token was loaded, please verify your .env file has '{self.token_env_var}'") raise EnvironmentError(f"No token was loaded, please verify your .env file has '{self.token_env_var}'")
else: else:
raise EnvironmentError(f"No environment variable '{self.token_env_var}' found") raise EnvironmentError(f"No environment variable '{self.token_env_var}' found")