code before moving database code to its own file.

This commit is contained in:
2026-07-17 14:59:58 -04:00
parent 228cf37eba
commit b573ca05aa
4 changed files with 132 additions and 57 deletions
+55 -44
View File
@@ -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: