Feat: Implement exponential backoff for YouTube API calls to handle quota errors.
This commit is contained in:
20
main.py
20
main.py
@@ -45,6 +45,9 @@ def get_authenticated_service():
|
||||
API_SERVICE_NAME, API_VERSION, credentials=credentials)
|
||||
|
||||
def get_live_chat_id(youtube, video_id):
|
||||
retries = 0
|
||||
max_retries = 5
|
||||
while retries < max_retries:
|
||||
try:
|
||||
response = youtube.videos().list(
|
||||
part='liveStreamingDetails',
|
||||
@@ -57,10 +60,20 @@ def get_live_chat_id(youtube, video_id):
|
||||
return video['liveStreamingDetails']['activeLiveChatId']
|
||||
return None
|
||||
except googleapiclient.errors.HttpError as e:
|
||||
if e.resp.status == 403 and "quotaExceeded" in str(e.content):
|
||||
console.print(f"[red]Quota exceeded. Retrying in {2**retries} seconds...[/red]")
|
||||
time.sleep(2**retries)
|
||||
retries += 1
|
||||
else:
|
||||
console.print(f"[red]An HTTP error {e.resp.status} occurred: {e.content}[/red]")
|
||||
return None
|
||||
console.print(f"[red]Failed to get live chat ID after {max_retries} retries due to quota issues.[/red]")
|
||||
return None
|
||||
|
||||
def fetch_live_chat_messages(youtube, live_chat_id, page_token=None):
|
||||
retries = 0
|
||||
max_retries = 5
|
||||
while retries < max_retries:
|
||||
try:
|
||||
request = youtube.liveChatMessages().list(
|
||||
liveChatId=live_chat_id,
|
||||
@@ -70,8 +83,15 @@ def fetch_live_chat_messages(youtube, live_chat_id, page_token=None):
|
||||
response = request.execute()
|
||||
return response
|
||||
except googleapiclient.errors.HttpError as e:
|
||||
if e.resp.status == 403 and "quotaExceeded" in str(e.content):
|
||||
console.print(f"[red]Quota exceeded. Retrying in {2**retries} seconds...[/red]")
|
||||
time.sleep(2**retries)
|
||||
retries += 1
|
||||
else:
|
||||
console.print(f"[red]An HTTP error {e.resp.status} occurred: {e.content}[/red]")
|
||||
return None
|
||||
console.print(f"[red]Failed to fetch live chat messages after {max_retries} retries due to quota issues.[/red]")
|
||||
return None
|
||||
|
||||
def main():
|
||||
youtube = get_authenticated_service()
|
||||
|
||||
Reference in New Issue
Block a user