You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
RedditVideoMakerBot/utils/envUpdate.py

48 lines
1.3 KiB

import os
import subprocess
import tempfile
import logging
from os import path
from sys import platform, stderr
log = logging.getLogger(__name__)
def envUpdate():
if path.exists(".env.template"):
if platform == "win32" or platform == "cygwin":
runPS('utils\envUpdateWin.ps1')
f = open("envVars.txt", "rb")
envTemplate = f.read()
elif platform == "darwin" or platform == "linux":
envTemplate = subprocess.check_output(
"awk -F '=' 'NF {print $1}' .env.template | grep --regexp=^[a-zA-Z]",
shell=True,
)
return
tempEnv = tempfile.TemporaryFile()
tempEnv.write(envTemplate)
tempEnv.seek(0)
envVars = tempEnv.readlines()
missing = []
isMissingEnvs = False
for env in envVars:
try:
env = env.decode("utf-8").strip()
except AttributeError:
env = env.strip()
if env not in os.environ:
isMissingEnvs = True
missing.append(env)
if isMissingEnvs:
log.error(
f"[ERROR] The following environment variables are missing: {missing}.)"
)
exit(-1)
def runPS(cmd):
completed = subprocess.run(["powershell", "-Command", cmd], capture_output=True)
return completed