twitch-autosave/daemon.py

#!/usr/bin/python3

# TODO: Сделать проверку на наличие стримера
# FIXME: не создавать папки для несуществующих стримеров
# TODO: Сделать нормальную конфигурацию

import os
from threading import Thread
import config_python
import schedule
from twitch import TwitchClient
import subprocess
import time
streamers = config_python.streamers
client_id = config_python.twitchid
ytdl_options = config_python.ytdl_options

def which(command):
    # Пиздец, почему нет нормального аналога which из bash???
    '''
    Мой аналог which из bash'а, который отдает true или false при наличии или отсутствии утилиты    
    '''
    for dirs in os.get_exec_path():
        if command in os.listdir(dirs):
            # Если что-нибудь нашли, то True
            return True
    # Если ничего не нашли во всех дирах, то выходим с False
    return False

def checkTools():
    '''
    Проверяет, установлены ли необходимые утилиты
    '''
    tools = ('youtube-dl', 'ffmpeg')
    for i in tools:
        if not which(i):
            print(i + " не установлен")
            return False
    return True

def startRecord(i):
    '''
    Функция, которая запускает в отдельном потоке запись стрима - recorder(i)
    '''
    th = Thread(target=recorder, args=(i, ))
    th.start()

def recorder(i):
    '''
    Функция, которая запускает youtube-dl, фактически записывает стрим
    '''
    path = config_python.path + "/"+ i
    print("Записываем стрим %s\n" % i)
    # cmdline для запуска youtube-dl 
    cmdline = ["youtube-dl", "-q", ytdl_options, 
        "-o", path + "/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s",
        "https://twitch.tv/"+i]
    s = subprocess.call(cmdline)
    print("Запись стрима %s закончена\n" % i)
    if (os.path.exists(path+"/pid")):
        os.system("rm "+path+"/pid")
        print("lock файл удален")

def checkAlive():
    '''
    1. Проверка на наличие стрима
    1.1 Если нет - удалить lock файл, если он есть
    1.2 Если есть - создать lock файл, запустить записывалку
    '''
    client = TwitchClient(client_id=client_id)
    for i in streamers:
        # Путь до диры со стримами
        path = config_python.path + "/"+ i
        # Создаем путь до диры со стримером, если его нет
        if not (os.path.exists(path)):
            os.makedirs(path)
        user_id=client.users.translate_usernames_to_ids(i)[0]['id'] # Получить ID по нику
        # Если стрим идет, то идем дальше
        if client.streams.get_stream_by_user(user_id):
            # Если стрим идет и лок файла нет, то записываем и ставим лок
            if (client.streams.get_stream_by_user(user_id).stream_type == 'live') and not (os.path.exists(config_python.path+"/"+i+"/pid")):
                print(i+" стримит")
                startRecord(i)
                os.system("touch "+path+"/pid")
            else:
                print(i+" Идет запись")
        else:
            # Если стрим не идет, то пишем об этом и убираем его из залоченных
            print(i+" Не стримит")
            # Если есть лок, то удаляем
            if (os.path.exists(path+"/pid")):
                os.system("rm "+path+"/pid")


def removeOldStreams():
    # https://clck.ru/WHh32
    records_path = config_python.path
    # По каждой папке со стримерами
    for i in streamers:
        # Если файлов в папке со стримами больше чем указано в конфиге
        if len(os.listdir(records_path+"/"+i)) > config_python.max_files:
            # Получаем список файлов 
            # и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге
            # Если превышает - удаляем старейший
            oldest = min(os.listdir(records_path+"/"+i), key=os.path.getctime)
            os.unlink(oldest)
            print("Удален файл: " + oldest)

if __name__ == "__main__":
    # Проверить, установлены ли нужные утилиты
    if not checkTools(): exit()
    # Проверять стримы раз в check_period
    schedule.every(config_python.check_period).seconds.do(checkAlive)
    # Каждый час удалять старые стримы
    schedule.every(1).hours.do(removeOldStreams)
    while True:
        schedule.run_pending()
        time.sleep(1)