file_works.py

from typing import List, Tuple, Dict
from tkinter import filedialog
from ffmpeg._run import Error
from config_works import (
    load_path,
    save_path,
    get_option,
    get_config,
)
from help_texts import (
    MANY_VIDEO,
    MANY_SUBS,
    NO_VIDEO,
    NO_AUDIO,
    NO_FOLDER,
    IN_USE,
)
from window_utils import get_subs_langs
import tkinter.messagebox
import asstosrt
import pysubs2
import tkinter
import ffmpeg
import glob
import sys
import re
import os


def resource_path(path):
    try:
        base_path = sys._MEIPASS
    except Exception:
        base_path = os.path.abspath('.')

    return os.path.join(base_path, path)


def glob_path(folder: str, extension: str) -> List[str]:
    """Функция для получения пути к файлу"""
    return glob.glob(os.path.join(folder, extension))


def get_fx_chains() -> Dict[str, str] or None:
    """Функция создания словаря из дабберов и названий их цепей эффектов"""
    fx_chains_folder = load_path('fx_chains_folder')
    if fx_chains_folder:
        fx_dict = {}
        fx_chains = glob_path(fx_chains_folder, '*.RfxChain')
        for chain in fx_chains:
            fx_chain_name = chain.split('\\')[-1]
            dubber_name = fx_chain_name.split('.')[-2].lower()
            fx_dict[dubber_name] = fx_chain_name
        return fx_dict
    return None


def path_choice(name: str) -> str or None:
    if name == 'project_path' or name == 'nrtemplate':
        defaultextension = 'rpp'
        filetypes = [('.rpp', '*.rpp')]
        initialdir = f'{os.getenv("APPDATA")}/REAPER/ProjectTemplates'
        initialfile = None
        if name == 'project_path':
            title = 'Выберите файл шаблона проекта REAPER'
        else:
            title = 'Выберите файл шаблона проекта REAPER для удаления шума'
        path = filedialog.askopenfilename(
            defaultextension=defaultextension,
            filetypes=filetypes,
            initialdir=initialdir,
            initialfile=initialfile,
            title=title,
        )
    elif name == 'fx_chains_folder' or name == 'folder':
        if name == 'fx_chains_folder':
            title = 'Выберите папку с цепями эффектов'
            initialdir = f'{os.getenv("APPDATA")}/REAPER/FXChains'
        elif name == 'folder':
            title = 'Выберите рабочую папку с эпизодом'
            initialdir = None
        path = filedialog.askdirectory(
            title=title,
            initialdir=initialdir,
        )
    if name != 'folder':
        save_path(name, path)
    else:
        return path


def subs_rename(
        folder: str,
        subs: List[str],
        number: str
        ) -> List[str] or None:
    """Функция для изменения имени субтитров"""
    try:
        filenamae = os.path.splitext(subs[0])[0].split('\\')[-2]
        new_name = f'{filenamae}/{number}.srt'
        os.rename(subs[0], new_name)
        subs = glob_path(folder, '*.srt')
    except PermissionError:
        tkinter.messagebox.showerror('Файл используется', IN_USE)
        return None
    return subs


def subs_extract(
        folder: str, video: List[str], param: str, mapping: str
        ) -> None:
    """Функция для извлечения субтитров из видео"""
    try:
        video_path = video[0].replace('\\', '/')
        input_file = ffmpeg.input(video_path)
        output_file = f'{folder}/subs.{param}'
        output = ffmpeg.output(input_file, output_file, map=mapping)
        ffmpeg.run(output)
    except Error:
        pass


def ass_sub_convert(folder: str, subs: List[str]) -> None:
    """Функция для конвертирования ass субтитров"""
    with open(subs[0], 'r', encoding='utf-8') as ass_file:
        srt_sub = asstosrt.convert(ass_file)
    with open(f'{folder}/subs.srt', 'w', encoding='utf-8') as srt_file:
        srt_file.write(srt_sub)
    os.remove(os.path.join(subs[0]))


def vtt_sub_convert(folder: str, subs: List[str]) -> None:
    """Функция для конвертирования vtt субтитров"""
    new_name = f'{folder}/subs.vtt'
    os.rename(subs[0], new_name)
    input_file = ffmpeg.input(new_name)
    output_file = f'{folder}/subs.srt'
    output = ffmpeg.output(input_file, output_file)
    ffmpeg.run(output)
    os.remove(os.path.join(new_name))


def video_rename(
        folder: str,
        video: List[str]
        ) -> Tuple[List[str] or None, str or None, str or None, str or None]:
    """Функция для изменения имени видео"""
    number = os.path.basename(folder)
    title = folder.split('/')[-2]
    ext = os.path.splitext(video[0])[-1]
    filename = os.path.splitext(video[0])[0].split('\\')[-2]
    new_name = f'{filename}/{title} {number}{ext}'
    try:
        os.rename(video[0], new_name)
        video = glob_path(folder, f'*{ext}')
        return video, title, number, ext
    except PermissionError:
        tkinter.messagebox.showerror('Файл используется', IN_USE)
        return None, None, None, None


def audio_rename(folder: str, audio: List[str], ext: str) -> List[str] or None:
    """Функция для изменения нечитаемых расширений"""
    for file in audio:
        file_ext = os.path.splitext(file)[-1]
        if file_ext == ext:
            continue
        elif '.reapeaks' not in file.lower():
            try:
                filename = os.path.splitext(file)[0]
                new_name = f'{filename}{ext}'
                copy = ''
                while os.path.exists(new_name):
                    copy += '_copy'
                    new_name = f'{filename}{copy}{ext}'
                os.rename(file, new_name)
            except PermissionError:
                tkinter.messagebox.showerror('Файл используется', IN_USE)
                return None
    fixed_audio = glob_path(folder, f'*{ext}')
    return fixed_audio


def comparator(sub: str) -> bool:
    """Функция для проверки субтитра"""
    if (
        'text' in sub
        or 'sign' in sub
        or 'надпись' in sub
        or 'caption' in sub
        or ('title' in sub and 'subtitle' not in sub)
        or 'song' in sub
        or 'screen' in sub
        or 'typedigital' in sub
        or 'phonetics' in sub
    ):
        return True
    return False


def subs_edit(subs: List[str], flag: str) -> None:
    """Функция для удаления из субтитров надписей и песен"""
    subtitles = pysubs2.load(subs[0])
    if flag == 'srt':
        pattern_1 = r'\((.*?)\)'
        pattern_2 = r'\[.*?]$'
        pattern_3 = '♫'
        pattern_4 = '♪'
        to_delete = [
            i for i, line in enumerate(subtitles) if (
                re.match(pattern_1, line.text)
                or re.match(pattern_2, line.text)
                or pattern_3 in line.text
                or pattern_4 in line.text
            )
        ]
    elif flag == 'ass':
        to_delete = []
        search_char = '{'
        if subtitles.events[0].name:
            for i, sub in enumerate(subtitles.events):
                if comparator(sub.name.lower()) and search_char in sub.text:
                    to_delete.append(i)
        if subtitles.events[0].style:
            for i, sub in enumerate(subtitles.events):
                if comparator(sub.style.lower()) and search_char in sub.text:
                    if i not in to_delete:
                        to_delete.append(i)
        else:
            return
    to_delete.sort()
    for i in reversed(to_delete):
        del subtitles[i]
    subtitles.save(subs[0])


def file_works(folder: str) -> (
        Tuple[List[str] or None,
              List[str] or None,
              List[str] or None,
              str or None,
              str or None,
              str or None]
        ):
    """Функция для подготовки файлов к работе"""
    config = get_config()
    if not folder:
        tkinter.messagebox.showerror('Ошибка', NO_FOLDER)
        return None, None, None, None, None, None
    mkv_video = glob_path(folder, '*.mkv')
    mp4_video = glob_path(folder, '*.mp4')
    if not mkv_video and not mp4_video:
        tkinter.messagebox.showerror('Нет видеофайлов', NO_VIDEO)
        return None, None, None, None, None, None
    if (mkv_video and mp4_video) or len(mkv_video) > 1 or len(mp4_video) > 1:
        tkinter.messagebox.showerror('Много видеофайлов', MANY_VIDEO)
        return None, None, None, None, None, None
    if mkv_video:
        video, title, number, ext = video_rename(folder, mkv_video)
    elif mp4_video:
        video, title, number, ext = video_rename(folder, mp4_video)
    flac_audio = glob_path(folder, '*.flac*')
    wav_audio = glob_path(folder, '*.wav*')
    if not flac_audio and not wav_audio:
        tkinter.messagebox.showerror('Нет аудиофайлов', NO_AUDIO)
        return None, None, None, None, None, None
    if flac_audio:
        flac_audio = audio_rename(folder, flac_audio, '.flac')
    if wav_audio:
        wav_audio = audio_rename(folder, wav_audio, '.wav')
    audio = list(flac_audio + wav_audio)
    subs = glob_path(folder, '*.srt')
    if len(subs) > 1:
        tkinter.messagebox.showerror('Много файлов субтитров', MANY_SUBS)
        return None, None, None, None, None, None
    if subs:
        subs = subs_rename(folder, subs, number)
        if get_option('subs_cleaner'):
            subs_edit(subs, 'srt')
    else:
        lang = get_subs_langs()[config['SUBS']['subs_lang']]
        try_sub = f'0:s:m:language:{lang}'
        eng_sub = '0:s:m:language:eng'
        any_sub = '0:s:m:language:?'
        ass_subs = glob_path(folder, '*.ass')
        vtt_subs = glob_path(folder, '*.vtt')
        if (ass_subs and vtt_subs) or len(ass_subs) > 1 or len(vtt_subs) > 1:
            tkinter.messagebox.showerror('Много файлов субтитров', MANY_SUBS)
            return None, None, None, None, None, None
        if vtt_subs:
            vtt_sub_convert(folder, vtt_subs)
        elif ass_subs:
            if get_option('subs_cleaner'):
                subs_edit(ass_subs, 'ass')
            ass_sub_convert(folder, ass_subs)
        elif not ass_subs and video:
            subs_extract(folder, video, 'ass', try_sub)
            ass_subs = glob_path(folder, '*.ass')
            if not ass_subs:
                subs_extract(folder, video, 'ass', eng_sub)
                ass_subs = glob_path(folder, '*.ass')
            if not ass_subs:
                subs_extract(folder, video, 'ass', any_sub)
                ass_subs = glob_path(folder, '*.ass')
            if ass_subs:
                if get_option('subs_cleaner'):
                    subs_edit(ass_subs, 'ass')
                ass_sub_convert(folder, ass_subs)
        srt_subs = glob_path(folder, '*.srt')
        if srt_subs:
            subs = subs_rename(folder, srt_subs, number)
            if get_option('subs_cleaner'):
                subs_edit(subs, 'srt')
        else:
            try:
                subs_extract(folder, video, 'srt', '0:s:m:language:?')
                subs = glob_path(folder, '*.srt')
                subs = subs_rename(folder, subs, number)
                if get_option('subs_cleaner'):
                    subs_edit(subs, 'srt')
            except IndexError:
                pass
            except TypeError:
                pass
    return subs, audio, video, title, number, ext


if __name__ == '__main__':
    pass