Try test for config.env

This commit is contained in:
yasirarism 2023-09-04 11:36:28 +07:00 committed by GitHub
parent 4d5ecc7ad4
commit cd934330da
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 26 additions and 18 deletions

View file

@ -9,25 +9,33 @@ from os import environ
LOGGER = getLogger("MissKaty") LOGGER = getLogger("MissKaty")
def getConfig(name: str):
try:
return environ[name]
except:
return ""
# Required ENV # Required ENV
try: TELEGRAM_API = environ.get("TELEGRAM_API", "")
BOT_TOKEN = getConfig("BOT_TOKEN") if not TELEGRAM_API:
API_ID = getConfig("API_ID") LOGGER.error("TELEGRAM_API variable is missing! Exiting now")
API_HASH = getConfig("API_HASH")
# MongoDB information
DATABASE_URI = getConfig("DATABASE_URI")
LOG_CHANNEL = int(environ.get("LOG_CHANNEL"))
except Exception as e:
LOGGER.error(f"One or more env variables missing! Exiting now.\n{e}")
sys.exit(1) sys.exit(1)
else:
TELEGRAM_API = int(TELEGRAM_API)
TELEGRAM_HASH = environ.get("TELEGRAM_HASH", "")
if not TELEGRAM_HASH:
LOGGER.error("TELEGRAM_HASH variable is missing! Exiting now")
sys.exit(1)
BOT_TOKEN = environ.get("BOT_TOKEN", "")
if not BOT_TOKEN:
LOGGER.error("BOT_TOKEN variable is missing! Exiting now")
sys.exit(1)
DATABASE_URI = environ.get("DATABASE_URI", "")
if not DATABASE_URI:
LOGGER.error("DATABASE_URI variable is missing! Exiting now")
sys.exit(1)
LOG_CHANNEL = environ.get("LOG_CHANNEL", "")
if not LOG_CHANNEL:
LOGGER.error("LOG_CHANNEL variable is missing! Exiting now")
sys.exit(1)
else:
LOG_CHANNEL = int(LOG_CHANNEL)
# Optional ENV
USER_SESSION = environ.get("USER_SESSION") USER_SESSION = environ.get("USER_SESSION")
DATABASE_NAME = environ.get("DATABASE_NAME", "MissKatyDB") DATABASE_NAME = environ.get("DATABASE_NAME", "MissKatyDB")
TZ = environ.get("TZ", "Asia/Jakarta") TZ = environ.get("TZ", "Asia/Jakarta")
@ -43,7 +51,7 @@ SUDO = list(
) )
SUPPORT_CHAT = environ.get("SUPPORT_CHAT", "YasirPediaChannel") SUPPORT_CHAT = environ.get("SUPPORT_CHAT", "YasirPediaChannel")
AUTO_RESTART = environ.get("AUTO_RESTART", False) AUTO_RESTART = environ.get("AUTO_RESTART", False)
OPENAI_API = getConfig("OPENAI_API") OPENAI_API = environ.get("OPENAI_API")
## Config For AUtoForwarder ## Config For AUtoForwarder
# Forward From Chat ID # Forward From Chat ID

View file

@ -29,7 +29,7 @@ try:
try: try:
res = requests.get(ENV_URL) res = requests.get(ENV_URL)
if res.status_code == 200: if res.status_code == 200:
with open(".env", "wb+") as f: with open("config.env", "wb+") as f:
f.write(res.content) f.write(res.content)
else: else:
LOGGER.error(f"config.env err: {res.status_code}") LOGGER.error(f"config.env err: {res.status_code}")