ReviveMii/ReviveTube
2
0
Fork 0
mirror of https://github.com/ReviveMii/revivetube synced 2025-09-05 21:10:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update revivetube.py

Browse Source 
Added Meta Infos
This commit is contained in:
TheErrorExe 2024-12-05 19:03:54 +01:00 committed by GitHub
parent 4281abde3f
commit fac789f778
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 77 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

106
revivetube.py
View File

@ -85,13 +85,34 @@ INDEX_TEMPLATE = """
</html> </html>
""" """
WATCH_STANDARD_TEMPLATE = """
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>{{ title }}</title>
</head>
<body style="text-align: center; font-family: Arial, sans-serif;">
<h1>{{ title }}</h1>
<h3>Uploaded by: {{ uploader }}</h3>
<video width="640" height="360" controls>
<source src="{{ video_mp4 }}" type="video/mp4">
Your browser does not support the video tag.
</video>
<h3>Description:</h3>
<p>{{ description }}</p>
</body>
</html>
"""
WATCH_WII_TEMPLATE = """ WATCH_WII_TEMPLATE = """
<!DOCTYPE html> <!DOCTYPE html>
<html lang="en"> <html lang="en">
<head> <head>
<meta charset="UTF-8"> <meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0"> <meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>ReviveTube Video</title> <title>{{ title }}</title>
</head> </head>
<body style="text-align: center; font-family: Arial, sans-serif;"> <body style="text-align: center; font-family: Arial, sans-serif;">
<div style="width: 100%; background-color: #000; text-align: center;"> <div style="width: 100%; background-color: #000; text-align: center;">
@ -102,29 +123,14 @@ WATCH_WII_TEMPLATE = """
</object> </object>
</div> </div>
<p>If the video does not play smoothly, restart the Internet Channel by pressing the Home button and then Reset. It's a bug. It happens if you visit too many Sites</p> <p>If the video does not play smoothly, restart the Internet Channel by pressing the Home button and then Reset. It's a bug. It happens if you visit too many Sites</p>
<h1>{{ title }}</h1>
<h3>Uploaded by: {{ uploader }}</h3>
<h3>Description:</h3>
<p>{{ description }}</p>
</body> </body>
</html> </html>
""" """
WATCH_STANDARD_TEMPLATE = """
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>ReviveTube Video</title>
</head>
<body style="text-align: center; font-family: Arial, sans-serif;">
<h1>ReviveTube Video</h1>
<video width="640" height="360" controls>
<source src="{{ video_mp4 }}" type="video/mp4">
Your browser does not support the video tag.
</video>
</body>
</html>
"""
# Routen
@app.route("/", methods=["GET"]) @app.route("/", methods=["GET"])
def index(): def index():
query = request.args.get("query") query = request.args.get("query")
@ -149,12 +155,34 @@ def index():
"id": item["id"]["videoId"], "id": item["id"]["videoId"],
"title": item["snippet"]["title"], "title": item["snippet"]["title"],
"uploader": item["snippet"]["channelTitle"], "uploader": item["snippet"]["channelTitle"],
"thumbnail": item["snippet"]["thumbnails"]["high"]["url"], "thumbnail": f"/thumbnail/{item['id']['videoId']}",
} }
for item in data.get("items", []) for item in data.get("items", [])
] ]
return render_template_string(INDEX_TEMPLATE, results=results) return render_template_string(INDEX_TEMPLATE, results=results)
@app.route("/video_metadata/<video_id>")
def video_metadata(video_id):
# Metadaten aus der YouTube-API abrufen
params = {
"part": "snippet",
"id": video_id,
"key": YOUTUBE_API_KEY,
}
response = requests.get(YOUTUBE_SEARCH_URL.replace("search", "videos"), params=params, timeout=3)
if response.status_code != 200:
return {"error": "Failed to fetch metadata"}, 500
data = response.json()
if "items" not in data or not data["items"]:
return {"error": "No metadata found"}, 404
video_data = data["items"][0]["snippet"]
return {
"title": video_data["title"],
"uploader": video_data["channelTitle"],
"description": video_data["description"],
}
@app.route("/watch", methods=["GET"]) @app.route("/watch", methods=["GET"])
def watch(): def watch():
@ -162,6 +190,13 @@ def watch():
if not video_id: if not video_id:
return "Missing video ID.", 400 return "Missing video ID.", 400
# Metadaten abrufen
metadata_response = requests.get(f"http://127.0.0.1:5000/video_metadata/{video_id}")
if metadata_response.status_code != 200:
return f"Failed to fetch video metadata: {metadata_response.text}", 500
metadata = metadata_response.json()
# User-Agent prüfen # User-Agent prüfen
user_agent = request.headers.get("User-Agent", "").lower() user_agent = request.headers.get("User-Agent", "").lower()
is_wii = "wii" in user_agent and "wiiu" not in user_agent is_wii = "wii" in user_agent and "wiiu" not in user_agent
@ -192,12 +227,6 @@ def watch():
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
return f"An error occurred while downloading the video: {str(e)}", 500 return f"An error occurred while downloading the video: {str(e)}", 500
# Check folder size
if get_folder_size(VIDEO_FOLDER) > MAX_FOLDER_SIZE:
shutil.rmtree(VIDEO_FOLDER)
os.makedirs(VIDEO_FOLDER, exist_ok=True) # Recreate the folder after deletion
return "The video folder exceeded 5 GB and was deleted.", 400
# Für Wii in FLV umwandeln # Für Wii in FLV umwandeln
if is_wii and not os.path.exists(video_flv_path): if is_wii and not os.path.exists(video_flv_path):
try: try:
@ -219,15 +248,32 @@ def watch():
# Passendes Template rendern # Passendes Template rendern
if is_wii: if is_wii:
return render_template_string(WATCH_WII_TEMPLATE, video_flv=f"/videos/{video_id}.flv") return render_template_string(WATCH_WII_TEMPLATE, video_flv=f"/videos/{video_id}.flv", **metadata)
else: else:
return render_template_string(WATCH_STANDARD_TEMPLATE, video_mp4=f"/videos/{video_id}.mp4") return render_template_string(WATCH_STANDARD_TEMPLATE, video_mp4=f"/videos/{video_id}.mp4", **metadata)
@app.route("/<path:filename>") @app.route("/<path:filename>")
def serve_static(filename): def serve_static(filename):
return send_file(os.path.join(filename)) return send_file(os.path.join(filename))
@app.route("/thumbnail/<video_id>")
def get_thumbnail(video_id):
thumbnail_url = f"https://img.youtube.com/vi/{video_id}/hqdefault.jpg"
try:
# Thumbnail von YouTube abrufen
response = requests.get(thumbnail_url, stream=True, timeout=5)
if response.status_code == 200:
# Content-Type weiterleiten
return send_file(
response.raw,
mimetype=response.headers.get("Content-Type", "image/jpeg"),
as_attachment=False,
)
else:
return f"Failed to fetch thumbnail. Status: {response.status_code}", 500
except requests.exceptions.RequestException as e:
return f"Error fetching thumbnail: {str(e)}", 500
if __name__ == "__main__": if __name__ == "__main__":
app.run(debug=True) app.run(debug=True)