code before moving database code to its own file.
This commit is contained in:
+55
-44
@@ -2,7 +2,8 @@
|
||||
import os
|
||||
import json
|
||||
import argparse
|
||||
from enum import Enum, auto, unique
|
||||
from enum import Enum
|
||||
from datetime import datetime
|
||||
from google.oauth2.credentials import Credentials
|
||||
from google_auth_oauthlib.flow import InstalledAppFlow
|
||||
from google.auth.transport.requests import Request
|
||||
@@ -10,40 +11,24 @@ from googleapiclient.discovery import build
|
||||
from googleapiclient.http import MediaFileUpload
|
||||
from googleapiclient.errors import HttpError
|
||||
|
||||
@unique
|
||||
class CategoryId(Enum):
|
||||
FILM_ANIMATION = 1
|
||||
AUTOS_VEHICLES = 2
|
||||
MUSIC = 10
|
||||
PETS_ANIMALS = 15
|
||||
SPORTS = 17
|
||||
SHORT_MOVIES = 18
|
||||
TRAVEL_EVENTS = 19
|
||||
GAMING = 20
|
||||
VIDEOBLOGGING = 21
|
||||
PEOPLE_BLOGS = 22
|
||||
COMEDY = 23
|
||||
ENTERTAINMENT = 24
|
||||
NEWS_POLITICS = 25
|
||||
HOWTO_STYLE = 26
|
||||
EDUCATION = 27
|
||||
SCIENCE_TECHNOLOGY = 28
|
||||
NONPROFITS_ACTIVISM = 29
|
||||
MOVIES = 30
|
||||
ANIME_ANIMATION = 31
|
||||
ACTION_ADVENTURE = 32
|
||||
CLASSICS = 33
|
||||
COMEDY_MOVIE = 34 # Renamed to avoid name collision
|
||||
DOCUMENTARY = 35
|
||||
DRAMA = 36
|
||||
FAMILY = 37
|
||||
FOREIGN = 38
|
||||
HORROR = 39
|
||||
SCI_FI_FANTASY = 40
|
||||
THRILLER = 41
|
||||
SHORTS = 42
|
||||
SHOWS = 43
|
||||
TRAILERS = 44
|
||||
"""Official YouTube Category IDs for API Uploads."""
|
||||
|
||||
FILM_AND_ANIMATION = "1"
|
||||
AUTOS_AND_VEHICLES = "2"
|
||||
MUSIC = "10"
|
||||
PETS_AND_ANIMALS = "15"
|
||||
SPORTS = "17"
|
||||
TRAVEL_AND_EVENTS = "19"
|
||||
GAMING = "20"
|
||||
PEOPLE_AND_BLOGS = "22"
|
||||
COMEDY = "23"
|
||||
ENTERTAINMENT = "24"
|
||||
NEWS_AND_POLITICS = "25"
|
||||
HOWTO_AND_STYLE = "26"
|
||||
EDUCATION = "27"
|
||||
SCIENCE_AND_TECHNOLOGY = "28"
|
||||
NONPROFITS_AND_ACTIVISM = "29"
|
||||
|
||||
def load_credentials():
|
||||
"""Load credentials from secrets.json"""
|
||||
@@ -72,18 +57,24 @@ def load_credentials():
|
||||
print(f"Error loading credentials: {str(e)}")
|
||||
return None
|
||||
|
||||
def upload_video(file_path: str, title: str, description: str, category: CategoryId, privacyStatus: str = 'private', tags: list = []):
|
||||
def upload_video(file_path: str, title: str, category: CategoryId, description: str = "", privacyStatus: str = 'private', tags: list = None, release_time: datetime = None):
|
||||
"""
|
||||
Upload a video to YouTube
|
||||
|
||||
Args:
|
||||
file_path (str): Path to the video file
|
||||
title (str): Title of the Video
|
||||
description (str): Video description
|
||||
categoryId (str): categoryId of the video
|
||||
description (str): Video description
|
||||
privacyStatus (str): privacyStatus of the video
|
||||
tags (str): tags to be use on the video
|
||||
release_time (datetime): Optional timezone-aware UTC datetime object for timed release
|
||||
|
||||
schedule_date = datetime.now(timezone.utc) + timedelta(days=2)
|
||||
"""
|
||||
if tags is None:
|
||||
tags = []
|
||||
|
||||
try:
|
||||
# Check if file exists
|
||||
if not os.path.exists(file_path):
|
||||
@@ -95,11 +86,27 @@ def upload_video(file_path: str, title: str, description: str, category: Categor
|
||||
if not credentials:
|
||||
return False
|
||||
|
||||
# Format release time to ISO 8601 UTC format (Required by YouTube)
|
||||
# Example format generated: "2026-07-20T15:00:00Z"
|
||||
publish_at_iso = release_time.strftime('%Y-%m-%dT%H:%M:%S.000Z')
|
||||
|
||||
# Create YouTube API client
|
||||
youtube = build('youtube', 'v3', credentials=credentials)
|
||||
|
||||
# Get the filename without extension as default title
|
||||
title = os.path.splitext(os.path.basename(file_path))[0]
|
||||
# Configure the status object dynamically
|
||||
status_body = {
|
||||
'selfDeclaredMadeForKids': False
|
||||
}
|
||||
|
||||
if release_time:
|
||||
# If release_time is passed, YouTube forces privacyStatus to 'private'
|
||||
status_body['privacyStatus'] = 'private'
|
||||
status_body['publishAt'] = release_time.strftime('%Y-%m-%dT%H:%M:%S.000Z')
|
||||
print(f"Configuring timed release for: {status_body['publishAt']}")
|
||||
else:
|
||||
# Standard immediate upload
|
||||
status_body['privacyStatus'] = privacyStatus
|
||||
print(f"Configuring immediate upload with status: {privacyStatus}")
|
||||
|
||||
# Prepare the video upload request
|
||||
body = {
|
||||
@@ -107,12 +114,9 @@ def upload_video(file_path: str, title: str, description: str, category: Categor
|
||||
'title': title,
|
||||
'description': description,
|
||||
'tags': tags,
|
||||
'categoryId': str(category.value) # Default to 'People & Blogs' category
|
||||
'categoryId': category.value
|
||||
},
|
||||
'status': {
|
||||
'privacyStatus': privacyStatus, # Default to private
|
||||
'selfDeclaredMadeForKids': False
|
||||
}
|
||||
'status': status_body
|
||||
}
|
||||
|
||||
|
||||
@@ -130,7 +134,7 @@ def upload_video(file_path: str, title: str, description: str, category: Categor
|
||||
media_body=media
|
||||
)
|
||||
|
||||
print("Starting upload...")
|
||||
print(f"Starting upload for '{title}'...")
|
||||
response = None
|
||||
while response is None:
|
||||
status, response = insert_request.next_chunk()
|
||||
@@ -141,6 +145,13 @@ def upload_video(file_path: str, title: str, description: str, category: Categor
|
||||
print(f"Video ID: {response['id']}")
|
||||
print(f"Title: {response['snippet']['title']}")
|
||||
print(f"URL: https://youtu.be/{response['id']}")
|
||||
|
||||
# Output confirmation based on what was chosen
|
||||
if 'publishAt' in response['status']:
|
||||
print(f"Scheduled Release Time: {response['status']['publishAt']}")
|
||||
else:
|
||||
print(f"Current Privacy Status: {response['status']['privacyStatus']}")
|
||||
|
||||
return True
|
||||
|
||||
except HttpError as e:
|
||||
|
||||
Reference in New Issue
Block a user