File size: 2,385 Bytes
787689e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | import sys
import os
import requests
from googleapiclient.discovery import build
from googleapiclient.http import MediaFileUpload
from google.auth.transport.requests import Request
import pickle
TOKEN_FILE = '/app/token.pickle'
VIDEO_PATH = '/app/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_video(url, path):
print(f"Downloading video...")
session = requests.Session()
for attempt in range(3):
try:
r = session.get(url, stream=True, timeout=60, headers={
'User-Agent': 'Mozilla/5.0',
'Referer': 'https://www.pexels.com/'
})
with open(path, 'wb') as f:
for chunk in r.iter_content(chunk_size=65536):
if chunk:
f.write(chunk)
print("Download complete!")
return
except Exception as e:
print(f"Attempt {attempt+1} failed: {e}")
if attempt == 2:
raise
def upload_video(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)}%")
print(f"Uploaded! Video ID: {response['id']}")
return response['id']
if __name__ == '__main__':
video_url = sys.argv[1]
title = sys.argv[2]
description = sys.argv[3]
download_video(video_url, VIDEO_PATH)
video_id = upload_video(title, description)
print(f"SUCCESS:{video_id}") |