97 lines
3.7 KiB
Python
97 lines
3.7 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()
|
|
|
|
# Define the emoji pattern and coloring function
|
|
emoji_pattern = re.compile(
|
|
r'('
|
|
r'\U0001F600-\U0001F64F' # emoticons
|
|
r'\U0001F300-\U0001F5FF' # symbols & pictographs
|
|
r'\U0001F680-\U0001F6FF' # transport & map symbols
|
|
r'\U0001F1E0-\U0001F1FF' # flags (iOS)
|
|
r'\u2702-\u27B0' # Dingbats
|
|
r'\u24C2-\u2B55' # Enclosed characters
|
|
r'\U0001F900-\U0001F9FF' # Supplemental Symbols & Pictographs
|
|
r'\u200D' # ZWJ
|
|
r'\uFE0F' # VS-16
|
|
r')+', flags=re.UNICODE)
|
|
|
|
def colour_emoji(txt):
|
|
return emoji_pattern.sub(r'[magenta]\1[/magenta]', txt)
|
|
|
|
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
|
|
|
|
# Process text-based emotes (e.g., :face-purple-sweating:)
|
|
message_text = re.sub(r'(:[a-zA-Z0-9_-]+:)', r'[blue]\1[/blue]', message_text)
|
|
|
|
# Apply emoji coloring using the new function
|
|
message_text = colour_emoji(message_text)
|
|
|
|
# Simple color for username (can be expanded to unique colors per user)
|
|
username_style = Style(color="#4CAF50", bold=True)
|
|
|
|
# Format message for terminal
|
|
formatted_message = f"[{username_style}]{author_display_name}[/]: {message_text}"
|
|
|
|
# Calculate padding to fill terminal width
|
|
# rich handles rendering, so we need to calculate the visible width of the string
|
|
# and pad it. This is a simplified approach, rich's Panel might be better for complex layouts.
|
|
rendered_length = console.measure(formatted_message).cell_len
|
|
padding = max(0, console.width - rendered_length)
|
|
padded_message = f"{formatted_message}{' ' * padding}"
|
|
|
|
# Alternate background styles
|
|
background_style = Style(bgcolor="#2B2B2B") if message_count % 2 == 0 else Style(bgcolor="#3A3A3A")
|
|
|
|
console.print(padded_message, style=background_style, overflow="crop")
|
|
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()
|