Compare commits
No commits in common. "b31f2ebcf7290b743473cfb662d99b0edac27e8c" and "f0ad40bfdd282006e05064b52ee33faa4f8c0bef" have entirely different histories.
b31f2ebcf7
...
f0ad40bfdd
67
daemon.py
67
daemon.py
@ -1,11 +1,13 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
# TODO: Перезапускать скрипт при обнаружении новой версии
|
||||
# TODO: Сделать нормальную конфигурацию
|
||||
|
||||
import os
|
||||
import sys
|
||||
from threading import Thread
|
||||
import configparser
|
||||
from types import resolve_bases
|
||||
import config_python
|
||||
import schedule
|
||||
from twitchAPI.twitch import Twitch
|
||||
import subprocess
|
||||
@ -14,31 +16,12 @@ import logging
|
||||
from logging.handlers import TimedRotatingFileHandler
|
||||
from termcolor import colored
|
||||
|
||||
streamers = config_python.streamers
|
||||
app_id = config_python.appid
|
||||
app_secret = config_python.appsecret
|
||||
|
||||
log_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s')
|
||||
log_file = 'output.log'
|
||||
cfg_file = 'config.ini'
|
||||
|
||||
|
||||
def set_config():
|
||||
"""
|
||||
Эта функция либо читает существующий конфиг, либо создает новый.
|
||||
Возвращает объект конфига (configparser.ConfigParser())
|
||||
"""
|
||||
config = configparser.ConfigParser()
|
||||
if not config.read('cfg_file.ini'):
|
||||
config["app"] = {
|
||||
"path": "",
|
||||
"check_period": 5,
|
||||
"max_files": 3
|
||||
}
|
||||
config["twitch"] = {
|
||||
"app_id": "",
|
||||
"app_secret": "",
|
||||
"streamers": ("asdf", "qqqqq")
|
||||
}
|
||||
with open('cfg_file.ini', 'w') as cfg_file:
|
||||
config.write(cfg_file)
|
||||
return config
|
||||
|
||||
|
||||
def which(command):
|
||||
@ -71,7 +54,7 @@ def startRecord(i):
|
||||
"""
|
||||
Функция, которая запускает в отдельном потоке запись стрима - recorder(i)
|
||||
"""
|
||||
th = Thread(target=recorder, args=(i,))
|
||||
th = Thread(target=recorder, args=(i, ))
|
||||
th.start()
|
||||
|
||||
|
||||
@ -79,16 +62,16 @@ def recorder(i):
|
||||
"""
|
||||
Функция, которая запускает youtube-dl, фактически записывает стрим
|
||||
"""
|
||||
path = config['app']['path'] + "/" + i
|
||||
path = config_python.path + "/" + i
|
||||
log.info("Записываем стрим %s\n" % i)
|
||||
# cmdline для запуска youtube-dl
|
||||
cmdline = ["youtube-dl", "-q", "-o",
|
||||
path + "/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s",
|
||||
path+"/%(upload_date)s_%(title)s__%(timestamp)s_%(id)s.%(ext)s",
|
||||
"https://twitch.tv/" + i]
|
||||
subprocess.call(cmdline)
|
||||
log.info("Запись стрима %s закончена\n" % i)
|
||||
if os.path.exists(path + "/pid"):
|
||||
os.remove(path + "/pid")
|
||||
os.remove(path+"/pid")
|
||||
log.info("lock файл удален")
|
||||
|
||||
|
||||
@ -99,9 +82,9 @@ def checkAlive():
|
||||
1.1 Если нет - удалить lock файл, если он есть
|
||||
1.2 Если есть - создать lock файл, запустить записывалку
|
||||
"""
|
||||
for i in config['twitch']['streamers']:
|
||||
for i in streamers:
|
||||
# Путь до диры со стримами
|
||||
path = config['app']['path'] + "/" + i
|
||||
path = config_python.path + "/" + i
|
||||
# Получаем инфо о стримере, если не получается, выходим с ошибкой
|
||||
|
||||
# resolved_id = client.users.translate_usernames_to_ids(i)
|
||||
@ -124,11 +107,10 @@ def checkAlive():
|
||||
# Если стрим идет, то идем дальше
|
||||
if user_stream['data']:
|
||||
# Если стрим идет и лок файла нет, то записываем и ставим лок
|
||||
if (user_stream['data'][0]['type'] == 'live') and not (
|
||||
os.path.exists(config['app']['path'] + "/" + i + "/pid")):
|
||||
if (user_stream['data'][0]['type'] == 'live') and not (os.path.exists(config_python.path+"/"+i+"/pid")):
|
||||
log.info(i + " стримит")
|
||||
startRecord(i)
|
||||
open(config['app']['path'] + "/pid", 'w').close
|
||||
open(path+"/pid", 'w').close
|
||||
else:
|
||||
log.info(
|
||||
colored(
|
||||
@ -142,22 +124,22 @@ def checkAlive():
|
||||
log.info(i + " Не стримит")
|
||||
# Если есть лок, то удаляем
|
||||
if os.path.exists(path + "/pid"):
|
||||
os.remove(path + "/pid")
|
||||
os.remove(path+"/pid")
|
||||
|
||||
|
||||
def removeOldStreams():
|
||||
# https://clck.ru/WHh32
|
||||
records_path = config['app']['path']
|
||||
records_path = config_python.path
|
||||
# По каждой папке со стримерами
|
||||
for i in config['twitch']['streamers']:
|
||||
for i in streamers:
|
||||
try:
|
||||
os.chdir(records_path + "/" + i)
|
||||
os.chdir(records_path+"/"+i)
|
||||
# Если файлов в папке со стримами больше чем указано в конфиге
|
||||
if len(os.listdir(records_path + "/" + i)) > int(config['app']['max_files']):
|
||||
if len(os.listdir(records_path+"/"+i)) > config_python.max_files:
|
||||
# Получаем список файлов
|
||||
# и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге
|
||||
# Если превышает - удаляем старейший
|
||||
oldest = min(os.listdir(records_path + "/" + i),
|
||||
oldest = min(os.listdir(records_path+"/"+i),
|
||||
key=os.path.getctime)
|
||||
os.unlink(oldest)
|
||||
log.warning("Удален файл: " + oldest)
|
||||
@ -191,19 +173,16 @@ if __name__ == "__main__":
|
||||
if not checkTools():
|
||||
exit()
|
||||
|
||||
# Set config
|
||||
config = set_config()
|
||||
|
||||
# Log config
|
||||
log = get_logger("main")
|
||||
log.info("Запущен")
|
||||
|
||||
# Проверять стримы раз в check_period
|
||||
schedule.every(int(config['app']['check_period'])).seconds.do(checkAlive)
|
||||
schedule.every(config_python.check_period).seconds.do(checkAlive)
|
||||
# Каждый час удалять старые стримы
|
||||
schedule.every(1).hours.do(removeOldStreams)
|
||||
|
||||
twitch_client = Twitch(config['twitch']['app_id'], config['twitch']['app_secret'])
|
||||
twitch_client = Twitch(app_id, app_secret)
|
||||
while True:
|
||||
schedule.run_pending()
|
||||
time.sleep(1)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user