125 lines
3.5 KiB
Python
125 lines
3.5 KiB
Python
import json
|
|
import os
|
|
import shlex
|
|
import subprocess
|
|
import sys
|
|
from datetime import datetime
|
|
|
|
from flask import Flask, flash, redirect, render_template, request, url_for
|
|
from werkzeug.utils import secure_filename
|
|
|
|
app = Flask(__name__)
|
|
app.config["UPLOAD_FOLDER"] = "Uploads"
|
|
try:
|
|
domain = os.environ["DOMAIN"] # https://killyoy.lovinator.space/
|
|
except KeyError:
|
|
sys.exit("Environment variable 'domain' is missing!")
|
|
print(f"{domain=}")
|
|
|
|
|
|
def find_video_resolution(path_to_video):
|
|
cmd = "ffprobe -v quiet -print_format json -show_streams "
|
|
args = shlex.split(cmd)
|
|
args.append(path_to_video)
|
|
# run the ffprobe process, decode stdout into utf-8 & convert to JSON
|
|
ffprobe_output = subprocess.check_output(args).decode("utf-8")
|
|
ffprobe_output = json.loads(ffprobe_output)
|
|
|
|
# find height and width
|
|
height = ffprobe_output["streams"][0]["height"]
|
|
width = ffprobe_output["streams"][0]["width"]
|
|
|
|
return height, width
|
|
|
|
|
|
def generate_html(
|
|
video_url, video_width, video_height, video_screenshot, video_filename
|
|
):
|
|
video_html = f"""
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<!-- Generated at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} -->
|
|
<head>
|
|
<meta property="og:type" content="video.other">
|
|
<meta property="twitter:player" content="{video_url}">
|
|
<meta property="og:video:type" content="text/html">
|
|
<meta property="og:video:width" content="{video_width}">
|
|
<meta property="og:video:height" content="{video_height}">
|
|
<meta name="twitter:image" content="{video_screenshot}">
|
|
<meta http-equiv="refresh" content="0;url={video_url}">
|
|
</head>
|
|
</html>
|
|
"""
|
|
|
|
html_url = f"{domain}{video_filename}"
|
|
video_filename += ".html"
|
|
|
|
with open(f"Uploads/{video_filename}", "w") as file:
|
|
file.write(video_html)
|
|
return html_url
|
|
|
|
|
|
def get_first_frame(path_video, file_filename):
|
|
cmd = f"ffmpeg -y -i {path_video} -vframes 1 Uploads/{file_filename}.jpg"
|
|
args = shlex.split(cmd)
|
|
|
|
subprocess.check_output(args).decode("utf-8")
|
|
|
|
return f"{domain}{file_filename}.jpg"
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return render_template("index.html")
|
|
|
|
|
|
@app.route("/", methods=["GET", "POST"])
|
|
def upload_file():
|
|
if request.method == "POST":
|
|
# check if the post request has the file part
|
|
if "file" not in request.files:
|
|
flash("No file part", "error")
|
|
return redirect(request.url)
|
|
file = request.files["file"]
|
|
|
|
# if user does not select file, browser also
|
|
# submit an empty part without filename
|
|
if file.filename == "":
|
|
flash("No selected file", "error")
|
|
return redirect(request.url)
|
|
|
|
if file:
|
|
filename = secure_filename(file.filename)
|
|
print(f"{filename=}")
|
|
filepath = f"Uploads/v/{filename}"
|
|
print(f"{filepath=}")
|
|
file.save(filepath)
|
|
|
|
height, width = find_video_resolution(filepath)
|
|
print(f"{height=}")
|
|
print(f"{width=}")
|
|
|
|
screenshot_url = get_first_frame(filepath, filename)
|
|
print(f"{screenshot_url=}")
|
|
|
|
video_url = f"{domain}v/{filename}"
|
|
print(f"{video_url=}")
|
|
|
|
html_url = generate_html(
|
|
video_url,
|
|
width,
|
|
height,
|
|
screenshot_url,
|
|
filename,
|
|
)
|
|
print(f"{html_url=}")
|
|
|
|
return {"html_url": f"{html_url}"}
|
|
|
|
# If GET
|
|
return redirect(url_for("index"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=True)
|