Перевел конфиг на человеческий configparser #13

Merged
lulzette merged 1 commits from 6_lulzette_move_cfg_to_configparser into master 2023-06-01 23:07:59 +03:00
Showing only changes of commit d2fc70ae4e - Show all commits

View File

@ -1,13 +1,11 @@
#!/usr/bin/python3 #!/usr/bin/python3
# TODO: Перезапускать скрипт при обнаружении новой версии # TODO: Перезапускать скрипт при обнаружении новой версии
# TODO: Сделать нормальную конфигурацию
import os import os
import sys import sys
from threading import Thread from threading import Thread
from types import resolve_bases import configparser
import config_python
import schedule import schedule
from twitchAPI.twitch import Twitch from twitchAPI.twitch import Twitch
import subprocess import subprocess
@ -16,12 +14,31 @@ import logging
from logging.handlers import TimedRotatingFileHandler from logging.handlers import TimedRotatingFileHandler
from termcolor import colored 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_format = logging.Formatter('%(asctime)s %(levelname)s:%(message)s')
log_file = 'output.log' 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): def which(command):
@ -54,7 +71,7 @@ def startRecord(i):
""" """
Функция, которая запускает в отдельном потоке запись стрима - recorder(i) Функция, которая запускает в отдельном потоке запись стрима - recorder(i)
""" """
th = Thread(target=recorder, args=(i, )) th = Thread(target=recorder, args=(i,))
th.start() th.start()
@ -62,16 +79,16 @@ def recorder(i):
""" """
Функция, которая запускает youtube-dl, фактически записывает стрим Функция, которая запускает youtube-dl, фактически записывает стрим
""" """
path = config_python.path + "/" + i path = config['app']['path'] + "/" + i
log.info("Записываем стрим %s\n" % i) log.info("Записываем стрим %s\n" % i)
# cmdline для запуска youtube-dl # cmdline для запуска youtube-dl
cmdline = ["youtube-dl", "-q", "-o", 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] "https://twitch.tv/" + i]
subprocess.call(cmdline) subprocess.call(cmdline)
log.info("Запись стрима %s закончена\n" % i) log.info("Запись стрима %s закончена\n" % i)
if os.path.exists(path + "/pid"): if os.path.exists(path + "/pid"):
os.remove(path+"/pid") os.remove(path + "/pid")
log.info("lock файл удален") log.info("lock файл удален")
@ -82,9 +99,9 @@ def checkAlive():
1.1 Если нет - удалить lock файл, если он есть 1.1 Если нет - удалить lock файл, если он есть
1.2 Если есть - создать lock файл, запустить записывалку 1.2 Если есть - создать lock файл, запустить записывалку
""" """
for i in streamers: for i in config['twitch']['streamers']:
# Путь до диры со стримами # Путь до диры со стримами
path = config_python.path + "/" + i path = config['app']['path'] + "/" + i
# Получаем инфо о стримере, если не получается, выходим с ошибкой # Получаем инфо о стримере, если не получается, выходим с ошибкой
# resolved_id = client.users.translate_usernames_to_ids(i) # resolved_id = client.users.translate_usernames_to_ids(i)
@ -107,10 +124,11 @@ def checkAlive():
# Если стрим идет, то идем дальше # Если стрим идет, то идем дальше
if user_stream['data']: if user_stream['data']:
# Если стрим идет и лок файла нет, то записываем и ставим лок # Если стрим идет и лок файла нет, то записываем и ставим лок
if (user_stream['data'][0]['type'] == 'live') and not (os.path.exists(config_python.path+"/"+i+"/pid")): if (user_stream['data'][0]['type'] == 'live') and not (
os.path.exists(config['app']['path'] + "/" + i + "/pid")):
log.info(i + " стримит") log.info(i + " стримит")
startRecord(i) startRecord(i)
open(path+"/pid", 'w').close open(config['app']['path'] + "/pid", 'w').close
else: else:
log.info( log.info(
colored( colored(
@ -124,22 +142,22 @@ def checkAlive():
log.info(i + " Не стримит") log.info(i + " Не стримит")
# Если есть лок, то удаляем # Если есть лок, то удаляем
if os.path.exists(path + "/pid"): if os.path.exists(path + "/pid"):
os.remove(path+"/pid") os.remove(path + "/pid")
def removeOldStreams(): def removeOldStreams():
# https://clck.ru/WHh32 # https://clck.ru/WHh32
records_path = config_python.path records_path = config['app']['path']
# По каждой папке со стримерами # По каждой папке со стримерами
for i in streamers: for i in config['twitch']['streamers']:
try: try:
os.chdir(records_path+"/"+i) os.chdir(records_path + "/" + i)
# Если файлов в папке со стримами больше чем указано в конфиге # Если файлов в папке со стримами больше чем указано в конфиге
if len(os.listdir(records_path+"/"+i)) > config_python.max_files: if len(os.listdir(records_path + "/" + i)) > int(config['app']['max_files']):
# Получаем список файлов # Получаем список файлов
# и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге # и смотрим, превышает ли кол-во mp4 файлов заданное в конфиге
# Если превышает - удаляем старейший # Если превышает - удаляем старейший
oldest = min(os.listdir(records_path+"/"+i), oldest = min(os.listdir(records_path + "/" + i),
key=os.path.getctime) key=os.path.getctime)
os.unlink(oldest) os.unlink(oldest)
log.warning("Удален файл: " + oldest) log.warning("Удален файл: " + oldest)
@ -173,16 +191,19 @@ if __name__ == "__main__":
if not checkTools(): if not checkTools():
exit() exit()
# Set config
config = set_config()
# Log config # Log config
log = get_logger("main") log = get_logger("main")
log.info("Запущен") log.info("Запущен")
# Проверять стримы раз в check_period # Проверять стримы раз в check_period
schedule.every(config_python.check_period).seconds.do(checkAlive) schedule.every(int(config['app']['check_period'])).seconds.do(checkAlive)
# Каждый час удалять старые стримы # Каждый час удалять старые стримы
schedule.every(1).hours.do(removeOldStreams) schedule.every(1).hours.do(removeOldStreams)
twitch_client = Twitch(app_id, app_secret) twitch_client = Twitch(config['twitch']['app_id'], config['twitch']['app_secret'])
while True: while True:
schedule.run_pending() schedule.run_pending()
time.sleep(1) time.sleep(1)