44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""Read settings from environment variables."""
|
|
import os
|
|
import pathlib
|
|
import sys
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
DESCRIPTION = (
|
|
"Discord will only create embeds for videos and images if they are "
|
|
"smaller than 8 mb. We can 'abuse' this by creating a .html that "
|
|
"contains the 'twitter:player' HTML meta tag linking to the video."
|
|
)
|
|
# Load environment variables
|
|
load_dotenv()
|
|
|
|
# Check if user has added a domain to the environment.
|
|
try:
|
|
serve_domain = os.environ["SERVE_DOMAIN"]
|
|
except KeyError:
|
|
sys.exit("discord-embed: Environment variable 'SERVE_DOMAIN' is missing!")
|
|
|
|
# Remove trailing slash from domain
|
|
if serve_domain.endswith("/"):
|
|
serve_domain = serve_domain[:-1]
|
|
|
|
# Check if we have a folder for uploads.
|
|
try:
|
|
upload_folder = os.environ["UPLOAD_FOLDER"]
|
|
except KeyError:
|
|
sys.exit("Environment variable 'UPLOAD_FOLDER' is missing!")
|
|
|
|
# Create upload_folder if it doesn't exist.
|
|
pathlib.Path(upload_folder).mkdir(parents=True, exist_ok=True)
|
|
|
|
# Remove trailing slash from upload_folder
|
|
if upload_folder.endswith("/"):
|
|
upload_folder = upload_folder[:-1]
|
|
|
|
# Discord webhook URL
|
|
try:
|
|
webhook_url = os.environ["WEBHOOK_URL"]
|
|
except KeyError:
|
|
sys.exit("Environment variable 'WEBHOOK_URL' is missing!")
|