TheLovinator/discord-embed
1
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

Use environment variable for domain

Browse Source
This commit is contained in:
Joakim Hellsén
2021-05-25 23:32:01 +02:00
parent 1897ca13be
commit cea8d60099
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
main.py
View File

@ -1,6 +1,8 @@
import json import json
import os
import shlex import shlex
import subprocess import subprocess
import sys
from datetime import datetime from datetime import datetime
from flask import Flask, flash, redirect, render_template, request, url_for from flask import Flask, flash, redirect, render_template, request, url_for
@ -8,9 +10,13 @@ from werkzeug.utils import secure_filename
app = Flask(__name__) app = Flask(__name__)
app.config["UPLOAD_FOLDER"] = "Uploads" app.config["UPLOAD_FOLDER"] = "Uploads"
try:
domain = os.environ["DOMAIN"] # https://killyoy.lovinator.space/
except KeyError:
sys.exit("Environment variable 'domanin' is missing!")
print(f"{domain=}")
# function to find the resolution of the input video file
def find_video_resolution(path_to_video): def find_video_resolution(path_to_video):
cmd = "ffprobe -v quiet -print_format json -show_streams " cmd = "ffprobe -v quiet -print_format json -show_streams "
args = shlex.split(cmd) args = shlex.split(cmd)
@ -31,7 +37,7 @@ def generate_html(
): ):
video_html = f""" video_html = f"""
<!DOCTYPE html> <!DOCTYPE html>
<html>
<!-- Generated at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} --> <!-- Generated at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} -->
<head> <head>
<meta property="og:type" content="video.other"> <meta property="og:type" content="video.other">
@ -42,9 +48,10 @@ def generate_html(
<meta name="twitter:image" content="{video_screenshot}"> <meta name="twitter:image" content="{video_screenshot}">
<meta http-equiv="refresh" content="0;url={video_url}"> <meta http-equiv="refresh" content="0;url={video_url}">
</head> </head>
</html>
""" """
html_url = f"https://killyoy.lovinator.space/{video_filename}" html_url = f"{domain}{video_filename}"
video_filename += ".html" video_filename += ".html"
with open(f"Uploads/{video_filename}", "w") as file: with open(f"Uploads/{video_filename}", "w") as file:
@ -58,7 +65,7 @@ def get_first_frame(path_video, file_filename):
subprocess.check_output(args).decode("utf-8") subprocess.check_output(args).decode("utf-8")
return f"https://killyoy.lovinator.space/{file_filename}.jpg" return f"{domain}{file_filename}.jpg"
@app.route("/") @app.route("/")
@ -95,7 +102,7 @@ def upload_file():
screenshot_url = get_first_frame(filepath, filename) screenshot_url = get_first_frame(filepath, filename)
print(f"{screenshot_url=}") print(f"{screenshot_url=}")
video_url = f"https://killyoy.lovinator.space/v/{filename}" video_url = f"{domain}v/{filename}"
print(f"{video_url=}") print(f"{video_url=}")
html_url = generate_html( html_url = generate_html(