From 1963ba53d96659c571ef2b04e1aac64149edc86b Mon Sep 17 00:00:00 2001
From: MMaker <mmaker@mmaker.moe>
Date: Thu, 27 Feb 2025 11:12:42 -0500
Subject: [PATCH] Cache video CDN status

---
 app.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/app.py b/app.py
index 10f99c2..d4625ce 100644
--- a/app.py
+++ b/app.py
@@ -109,6 +109,8 @@ def download_and_upload_video(video_id, url, video_quality):
                     )
                     
                     logger.info(f"Successfully uploaded video {video_id} to CDN")
+                    if cache:
+                        cache.set(f"{video_id}_uploaded", True, expire=CACHE_EXPIRATION_SECONDS)
                     
                     # Clear cache for this video to ensure next view gets updated HTML
                     if cache:
@@ -160,6 +162,9 @@ worker_thread.start()
 
 def is_video_in_cdn(video_id):
     """Check if video exists in CDN"""
+    if cache and cache.get(f"{video_id}_uploaded"):
+        return True
+
     if not s3_client:
         return False