88 lines
3.6 KiB
Python
88 lines
3.6 KiB
Python
import pytchat
|
|
import sys
|
|
import os
|
|
import time
|
|
import re
|
|
from datetime import datetime
|
|
from rich.console import Console
|
|
from rich.style import Style
|
|
|
|
console = Console()
|
|
|
|
def main():
|
|
# Clear the terminal screen
|
|
os.system('clear')
|
|
|
|
video_id = input("Enter the YouTube Live Stream Video ID: ")
|
|
|
|
# Setup chat logging
|
|
log_dir = "chat_logs"
|
|
os.makedirs(log_dir, exist_ok=True)
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
log_filename = os.path.join(log_dir, f"chat_{video_id}_{timestamp}.log")
|
|
log_file = open(log_filename, "w", encoding="utf-8")
|
|
console.print(f"[green]Chat will be logged to {log_filename}[/green]")
|
|
|
|
try:
|
|
livechat = pytchat.create(video_id=video_id)
|
|
console.print(f"[green]Listening to live chat for video ID: {video_id}[/green]")
|
|
console.print(f"[green]Video ID accepted: {video_id}[/green]")
|
|
|
|
# Wait for 5 seconds, then clear the screen
|
|
time.sleep(5)
|
|
os.system('clear')
|
|
|
|
message_count = 0
|
|
while livechat.is_alive():
|
|
for c in livechat.get().sync_items():
|
|
author_display_name = c.author.name
|
|
message_text = c.message
|
|
|
|
# pytchat does not provide explicit emote data like the official API
|
|
# So we rely on regex for text-based emotes and standard emojis.
|
|
|
|
# Process text-based emotes (e.g., :face-purple-sweating:)
|
|
message_text = re.sub(r'(:[a-zA-Z0-9_-]+:)', r'[blue]\1[/blue]', message_text)
|
|
|
|
# Add coloring for standard emojis
|
|
emoji_pattern = re.compile(
|
|
r"""(
|
|
\U0001F600-\U0001F64F # emoticons
|
|
\U0001F300-\U0001F5FF # symbols & pictographs
|
|
\U0001F680-\U0001F6FF # transport & map symbols
|
|
\U0001F1E0-\U0001F1FF # flags (iOS)
|
|
\U00002702-\U000027B0 # Dingbats
|
|
\U000024C2-\U0001F251 # Enclosed characters
|
|
\U0001F900-\U0001F9FF # Supplemental Symbols and Pictographs
|
|
\U0000200D # Zero Width Joiner (for emoji sequences)
|
|
\U0000FE0F # Variation Selector-16 (for emoji presentation)
|
|
)+"""
|
|
, re.UNICODE | re.VERBOSE)
|
|
message_text = emoji_pattern.sub(r'[magenta]\1[/magenta]', message_text)
|
|
|
|
# Alternate background styles
|
|
background_style = Style(bgcolor="#2B2B2B") if message_count % 2 == 0 else Style(bgcolor="#3A3A3A")
|
|
|
|
# Simple color for username (can be expanded to unique colors per user)
|
|
username_style = Style(color="#4CAF50", bold=True)
|
|
|
|
# Format message for terminal and log file
|
|
formatted_message = f"[{username_style}]{author_display_name}[/]: {message_text}"
|
|
console.print(formatted_message, style=background_style)
|
|
log_file.write(f"{datetime.now().strftime("%H:%M:%S")} {formatted_message}\n")
|
|
message_count += 1
|
|
|
|
except Exception as e:
|
|
console.print(f"[red]An error occurred: {e}[/red]")
|
|
finally:
|
|
log_file.close()
|
|
save_log = input(f"\nDo you want to save the chat log to {log_filename}? (y/n): ").lower()
|
|
if save_log != 'y' and save_log != 'yes':
|
|
os.remove(log_filename)
|
|
console.print(f"[red]Chat log not saved. {log_filename} deleted.[/red]")
|
|
else:
|
|
console.print(f"[green]Chat log saved to {log_filename}[/green]")
|
|
|
|
if __name__ == '__main__':
|
|
main()
|