Upload mirror.py

mirror.py

@@ -0,0 +1,89 @@
+import sys
+import os
+import subprocess
+import pickle
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaFileUpload
+from google.auth.transport.requests import Request
+
+TOKEN_FILE = '/app/token.pickle'
+VIDEO_PATH = '/app/mirror_video.mp4'
+
+def get_credentials():
+    creds = None
+    if os.path.exists(TOKEN_FILE):
+        with open(TOKEN_FILE, 'rb') as f:
+            creds = pickle.load(f)
+    if creds and creds.expired and creds.refresh_token:
+        creds.refresh(Request())
+        with open(TOKEN_FILE, 'wb') as f:
+            pickle.dump(creds, f)
+    return creds
+
+def download_from_youtube(video_id, output_path):
+    url = f"https://www.youtube.com/watch?v={video_id}"
+    print(f"Downloading: {url}")
+
+    if os.path.exists(output_path):
+        os.remove(output_path)
+
+    result = subprocess.run([
+        'yt-dlp',
+        '-f', 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
+        '--merge-output-format', 'mp4',
+        '-o', output_path,
+        '--no-playlist',
+        url
+    ], capture_output=True, text=True)
+
+    if result.returncode != 0:
+        raise Exception(f"yt-dlp failed: {result.stderr}")
+
+    print("Download complete!")
+
+def upload_to_channel_b(title, description):
+    creds = get_credentials()
+    youtube = build('youtube', 'v3', credentials=creds)
+
+    body = {
+        'snippet': {
+            'title': title,
+            'description': description,
+            'categoryId': '22',
+            'tags': ['shorts', 'viral', 'trending']
+        },
+        'status': {
+            'privacyStatus': 'public',
+            'selfDeclaredMadeForKids': False
+        }
+    }
+
+    media = MediaFileUpload(VIDEO_PATH, mimetype='video/mp4', resumable=True)
+    request = youtube.videos().insert(part='snippet,status', body=body, media_body=media)
+
+    response = None
+    while response is None:
+        status, response = request.next_chunk()
+        if status:
+            print(f"Uploading... {int(status.progress() * 100)}%")
+
+    new_id = response['id']
+    print(f"Uploaded! New Video ID: {new_id}")
+
+    if os.path.exists(VIDEO_PATH):
+        os.remove(VIDEO_PATH)
+
+    return new_id
+
+if __name__ == '__main__':
+    video_id    = sys.argv[1]
+    title       = sys.argv[2] if len(sys.argv) > 2 else 'Mirrored Video'
+    description = sys.argv[3] if len(sys.argv) > 3 else ''
+
+    try:
+        download_from_youtube(video_id, VIDEO_PATH)
+        new_video_id = upload_to_channel_b(title, description)
+        print(f"SUCCESS:{new_video_id}")
+    except Exception as e:
+        print(f"FAILED: {e}")
+        sys.exit(1)