AutoZoom/functions.py

# -*- coding: utf-8 -*-

import pip
import time
import json
import os
import shutil
import gzip
from datetime import datetime
from pathlib import Path

path = Path(__file__).resolve().parent
sounds_folder = str(Path(str(path)+"/sounds/")) + os.sep
files_folder = str(Path(str(path)+"/files/")) + os.sep
logs_folder = str(Path(str(path)+"/logs/")) + os.sep


default_config = {
            "debug": False,
            "shutdown_timeout": 30,
            "shutdown_enabled": True,
            "start": "shift+f7",
            "stop": "shift+f8",
            "telegram_enabled": False,
            "use_colors": True,
            "run_fullscreen": False,
            "use_rpc": True,
            "sounds": True,
            "end_mode": "shutdown",
            "obs_exe": None,
            "obs_core": None,
            "update_check": True,
            "write_logs": True,
            "log_size": 512
    }


# Функция проверки размера файла
def checkSize():
    global logs_folder
    
    i = 0

    while i < 2:
        try:
            log = os.stat(logs_folder + 'latest.log')

            if (log.st_size / 1024) > getConfig("log_size"):
                with open(logs_folder + 'latest.log', 'rb') as f_in:
                    with gzip.open(f'{logs_folder}{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.zip', 'wb') as f_out:
                        shutil.copyfileobj(f_in, f_out)
                        
                        if getConfig("debug"):
                            print(f'Copied {logs_folder}{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.zip')
                            
                open(logs_folder + 'latest.log', 'w').close()
                
            i = 2

        except FileNotFoundError:
            if getConfig("debug"):
                print('Log file not found')
                time.sleep(2)
            
            try:
                log = open(logs_folder + 'latest.log', 'a')
                open(logs_folder + 'latest.log', 'a').close()
            except:
                try:
                    os.mkdir(logs_folder)
                    log = open(logs_folder + 'latest.log', 'a')
                    open(logs_folder + 'latest.log', 'a').close()
                except:
                    if getConfig("debug"):
                        time.sleep(2)
                        print('Log file could not be created')
            
            i += 1


# Функция добавления в лог
def appendLog(message):

    if getConfig("write_logs"):
    
        global logs_folder

        checkSize()

        try:
            log = open(logs_folder + 'latest.log', 'a')
            open(logs_folder + 'latest.log', 'a').close()
        except:
            try:
                os.mkdir(logs_folder)
                log = open(logs_folder + 'latest.log', 'a')
                open(logs_folder + 'latest.log', 'a').close()
            except:
                time.sleep(2)
                print('Log file could not be created')
                
        log.write(f'[{datetime.now().strftime("%H:%M:%S | %d.%m.%Y")}] {message}\n')
        log.close()


# Функция добавления переменных, если их нет
def repairConfig(some_dic):
    global files_folder
    global default_config
    
    for key in some_dic:
        try:
            some_dic[key]
        except NameError:
            some_dic[key] = default_config[key]
            
            saveJson(files_folder+'config.json', some_dic)
   

def getConfig(some_var):
    global files_folder
    global default_config

    if os.path.exists(files_folder):
        if not os.path.exists(files_folder+'config.json'):
            temp_config_list = default_config
            
            saveJson(files_folder+'config.json', temp_config_list)
            return temp_config_list[some_var]
        else:
            try:
                with open(f"{files_folder}config.json", encoding="utf-8") as json_file:
                    config_list = json.load(json_file)
                    json_file.close()
                    
                    try:
                        return config_list[some_var]
                    except:
                        try:
                            repairConfig(config_list)
                            config_list = json.load(json_file)
                            json_file.close()
                            return config_list[some_var]
                        except:
                            return default_config[some_var]
            except:
                return "Error"
    else:
        os.mkdir(files_folder)
        if not os.path.exists(files_folder+'config.json'):
            temp_config_list = default_config
            
            saveJson(files_folder+'config.json', temp_config_list)
            return temp_config_list[some_var]
        else:
            try:
                with open(f"{files_folder}config.json", encoding="utf-8") as json_file:
                    config_list = json.load(json_file)
                    json_file.close()
                    
                    try:
                        return config_list[some_var]
                    except:
                        try:
                            repairConfig(config_list)
                            config_list = json.load(json_file)
                            json_file.close()
                            return config_list[some_var]
                        except:
                            return default_config[some_var]
                            
            except:
                return "Error"


def saveJson(filename, value):
    with open(filename, 'w', encoding="utf-8") as f:
        json.dump(value, f, indent=4, ensure_ascii=False)
        f.close()