73 lines
2.8 KiB
Python
73 lines
2.8 KiB
Python
import logging
|
|
import os
|
|
import time
|
|
import sys
|
|
from datetime import datetime
|
|
|
|
from clipper.api import TwitchApi, TwitchStreamStatus
|
|
from clipper.chat import TwitchChatRecorder
|
|
from clipper.video import TwitchVideoRecorder
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RecorderConfig:
|
|
def __init__(self, tw_client, tw_secret, tw_streamer, tw_quality, output_folder):
|
|
self.output_folder = output_folder
|
|
self.tw_quality = tw_quality
|
|
self.tw_streamer = tw_streamer
|
|
self.tw_secret = tw_secret
|
|
self.tw_client = tw_client
|
|
|
|
|
|
class Recorder:
|
|
def __init__(self, config):
|
|
self.config = config
|
|
self.api = TwitchApi(config.tw_client, config.tw_secret)
|
|
self.streamer_folder = os.path.join(self.config.output_folder, self.config.tw_streamer)
|
|
self.video_recorder = TwitchVideoRecorder()
|
|
self.chat_recorder = TwitchChatRecorder(self.api, debug=False)
|
|
|
|
def run(self):
|
|
logger.info("Start watching streamer %s", self.config.tw_streamer)
|
|
status = self.api.get_user_status(self.config.tw_streamer)
|
|
|
|
while True:
|
|
if self.video_recorder.is_running or self.chat_recorder.is_running:
|
|
if not (self.video_recorder.is_running and self.chat_recorder.is_running):
|
|
self.video_recorder.stop()
|
|
self.chat_recorder.stop()
|
|
time.sleep(1)
|
|
continue
|
|
|
|
logger.info("Recording in progress. Wait 1m")
|
|
time.sleep(60)
|
|
|
|
elif status == TwitchStreamStatus.ONLINE:
|
|
logger.info("Streamer %s is online. Start recording", self.config.tw_streamer)
|
|
|
|
start_time = datetime.now()
|
|
record_folder_name = start_time.strftime("%d-%m-%Y_%H-%M-%S")
|
|
record_folder = os.path.join(self.streamer_folder, record_folder_name)
|
|
os.makedirs(record_folder)
|
|
|
|
chat_file = os.path.join(record_folder, "chat.txt")
|
|
video_file = os.path.join(record_folder, "video.mp4")
|
|
|
|
self.chat_recorder.run(self.config.tw_streamer, chat_file)
|
|
self.video_recorder.run(self.config.tw_streamer, video_file, quality="160p")
|
|
|
|
time.sleep(5)
|
|
|
|
elif status == TwitchStreamStatus.OFFLINE:
|
|
logger.info("Streamer %s is offline. Waiting for 300 sec", self.config.tw_streamer)
|
|
time.sleep(300)
|
|
|
|
if status == TwitchStreamStatus.ERROR:
|
|
logger.critical("Error occurred. Exit", self.config.tw_streamer)
|
|
sys.exit(1)
|
|
|
|
elif status == TwitchStreamStatus.NOT_FOUND:
|
|
logger.critical(f"Streamer %s not found, invalid streamer_name or typo", self.config.tw_streamer)
|
|
sys.exit(1)
|