1.77

DoobyDouglas · Jun 29, 2023 · 91085a7 · 91085a7
1 parent e4c0255
commit 91085a7
Show file tree

Hide file tree

Showing 9 changed files with 474 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -8,6 +8,8 @@ __pycache__/
 
 # Distribution / packaging
 .Python
+HARDSUBBER.ini
+subs_langs.json
 config.ini
 some_files/
 develop-eggs/

diff --git a/HardSubber.exe b/HardSubber.exe
diff --git a/HardSubber.py b/HardSubber.py
@@ -0,0 +1,229 @@
+from PIL import Image, ImageTk
+from typing import Dict
+import subprocess
+import tkinter
+import os
+import pysubs2
+import tkinter.messagebox
+from threading import Thread
+import sys
+import tkinter.ttk as ttk
+from ttkbootstrap import Style
+import json
+from configs import (
+    get_config,
+    get_value,
+    save_options,
+    SUBS_LANGS_DICT,
+    RESOLUTIONS_LIST,
+)
+from fileworks import choice_files
+from tktooltip import ToolTip
+from helptexts import HELP_DICT
+# pyinstaller --noconfirm --onefile --noconsole --add-data 'background.png;.' --add-data 'ico.ico;.' --icon=ico.ico HardSubber.py
+
+
+def hardsubber(master: tkinter.Tk) -> None:
+    config = get_config()
+    master.nametowidget('start').config(state='disabled')
+    video_ext, subs_ext, folder, video, subs = choice_files(master)
+    if video_ext and subs_ext and folder and video and subs:
+        master.nametowidget('pb').start()
+        disk = folder.split(':')[0].lower()
+        folder_path = folder.split(':')[1]
+        if subs_ext == '.ass':
+            param = 'ass'
+        else:
+            param = 'subtitles'
+        if get_value('audio'):
+            sound_param = ' '
+        else:
+            sound_param = ' -an '
+        if config['OPTIONS']['bitrate']:
+            bitrate = f" -b {config['OPTIONS']['bitrate']}k "
+        else:
+            bitrate = ' '
+        if config['OPTIONS']['resolution'] != 'ORIGINAL':
+            resolution = f" -s {config['OPTIONS']['resolution']} "
+        else:
+            resolution = ' '
+        if get_value('convert_to_mp4'):
+            video_ext = '.mp4'
+        outputname = 'HRDSBBD_' + os.path.splitext(video)[0]
+        command = (
+            f'{disk}: && cd {folder_path} && ffmpeg -i {video} -vf '
+            f'{param}={subs} -vcodec libx264'
+            f'{bitrate}{resolution}'
+            f'-acodec copy{sound_param}{outputname}{video_ext}'
+        )
+        subprocess.call(command, shell=True)
+        master.nametowidget('pb').stop()
+    master.nametowidget('start').config(state='normal')
+
+
+def main(checkboxes: Dict, master: tkinter.Tk) -> None:
+    save_options(checkboxes, master)
+    hardsubber(master)
+
+
+def on_save_click(checkboxes: Dict, master: tkinter.Tk) -> None:
+    thread = Thread(target=main, args=(checkboxes, master))
+    thread.start()
+
+
+def resource_path(path):
+    try:
+        base_path = sys._MEIPASS
+    except Exception:
+        base_path = os.path.abspath('.')
+
+    return os.path.join(base_path, path)
+
+
+def get_subs_langs():
+    if not os.path.exists('subs_langs.json'):
+        with open('subs_langs.json', 'w+') as json_file:
+            data = {}
+            for key, value in SUBS_LANGS_DICT.items():
+                data[key] = value
+            json.dump(data, json_file)
+    with open('subs_langs.json', 'r') as json_file:
+        data = json.load(json_file)
+    return data
+
+
+def menu_setup(menu, key):
+    if key == 'subs_lang':
+        LIST = list(get_subs_langs().keys())
+    else:
+        LIST = RESOLUTIONS_LIST
+    try:
+        if config['OPTIONS'][key] in LIST:
+            menu.set(config['OPTIONS'][key])
+        else:
+            menu.set(LIST[0])
+    except KeyError:
+        menu.set(LIST[0])
+
+
+def set_geometry(master: tkinter.Tk):
+    width = 300
+    height = 236
+    upper = master.winfo_screenheight() // 5
+    x = (master.winfo_screenwidth() - width) // 2
+    y = (master.winfo_screenheight() - height) // 2
+    return f'{width}x{height}+{x}+{y - upper}'
+
+
+config = get_config()
+master = tkinter.Tk(className='HARDSUBBER.main')
+master.geometry(set_geometry(master))
+master.resizable(False, False)
+master.title('HARDSUBBER v1.77')
+master.iconbitmap(default=resource_path('ico.ico'))
+img = Image.open(resource_path('background.png'))
+raw_img = ImageTk.PhotoImage(img)
+background_label = tkinter.Label(master, image=raw_img)
+background_label.place(x=0, y=0, relwidth=1, relheight=1)
+
+style = Style(theme='pulse')
+style.configure('TCheckbutton', background='#ffc0cb')
+style.configure('TButton', background='#ffc0cb', foreground='black')
+style.configure('TLabel', background='#ffc0cb')
+style.configure('Horizontal.TProgressbar', thickness=25)
+
+subs_extract_ = ttk.Label(master, text='Select subtitles to extract:')
+subs_extract_.grid(row=0, column=0, sticky=tkinter.W, padx=6, pady=6)
+select_resolution = ttk.Label(master, text='Select resolution:')
+select_resolution.grid(row=1, column=0, sticky=tkinter.W, padx=6, pady=6)
+enter_bitrate = ttk.Label(master, text='Enter bitrate:')
+enter_bitrate.grid(row=2, column=0, sticky=tkinter.W, padx=6, pady=6)
+
+subs_menu = ttk.Combobox(
+    master,
+    values=list(get_subs_langs().keys()),
+    name='subs_menu',
+    state='readonly',
+    width=12,
+)
+subs_menu.place(relx=0.5, rely=1.0, anchor="s", x=95, y=-204)
+menu_setup(subs_menu, 'subs_lang')
+ToolTip(subs_menu, HELP_DICT['subs_menu'], 1)
+
+resolution_menu = ttk.Combobox(
+    master,
+    values=RESOLUTIONS_LIST,
+    name='resolution_menu',
+    state='readonly',
+    width=12,
+)
+resolution_menu.place(relx=0.5, rely=1.0, anchor="s", x=95, y=-173)
+menu_setup(resolution_menu, 'resolution')
+ToolTip(resolution_menu, HELP_DICT['resolution_menu'], 1)
+
+bitrate_entry = ttk.Entry(
+    master,
+    name='bitrate_entry',
+    width=14,
+)
+bitrate_entry.place(relx=0.5, rely=1.0, anchor="s", x=95, y=-142)
+try:
+    if config['OPTIONS']['bitrate']:
+        bitrate_entry.insert(1, config['OPTIONS']['bitrate'])
+except KeyError:
+    pass
+ToolTip(bitrate_entry, HELP_DICT['bitrate_entry'], 1)
+
+OPTIONS = [
+    'audio',
+    'subs_extract',
+    'subs_cleaner',
+    'convert_to_mp4',
+]
+
+checkboxes = {}
+if 'OPTIONS' not in config:
+    config['OPTIONS'] = {}
+
+for i, option in enumerate(OPTIONS):
+    var = tkinter.BooleanVar()
+    if option in config['OPTIONS']:
+        var.set(config['OPTIONS'].getboolean(option))
+    else:
+        var.set(False)
+    checkbox = ttk.Checkbutton(
+        master,
+        text=option,
+        variable=var,
+        padding=6,
+    )
+    checkbox.grid(
+        row=i + 3,
+        column=0,
+        sticky=tkinter.W
+    )
+    checkboxes[option] = var
+    ToolTip(checkbox, HELP_DICT[option], 1)
+
+save_button = ttk.Button(
+    master,
+    name='start',
+    text='START',
+    width=7,
+    padding=5,
+    command=lambda: on_save_click(checkboxes, master),
+)
+save_button.place(relx=0.5, rely=1.0, anchor="center", y=-21, x=116)
+ToolTip(save_button, HELP_DICT['start'], 1)
+
+progressbar = ttk.Progressbar(
+    master,
+    mode='determinate',
+    name='pb',
+    length=224,
+    style='Horizontal.TProgressbar',
+)
+progressbar.place(relx=0.5, rely=1.0, anchor="center", y=-21, x=-31)
+
+if __name__ == '__main__':
+    master.mainloop()
diff --git a/background.png b/background.png
diff --git a/configs.py b/configs.py
@@ -0,0 +1,58 @@
+import configparser
+import tkinter
+
+DEFAULT_BITRATE = 1500
+SUBS_LANGS_DICT = {
+    'Russia': 'rus',
+    'US': 'eng',
+    'Saudi Arabia': 'ara',
+    'Germany': 'ger',
+    'Latin America': 'spa',
+    'France': 'fre',
+    'Italy': 'ita',
+    'Brasil': 'por'
+}
+RESOLUTIONS_LIST = [
+    'ORIGINAL',
+    '1280x720',
+    '1920x1080',
+    '3840x2160',
+]
+
+
+def get_config() -> configparser.ConfigParser:
+    config = configparser.ConfigParser()
+    config.read('HARDSUBBER.ini', encoding='utf-8')
+    if 'OPTIONS' not in config:
+        config['OPTIONS'] = {}
+    return config
+
+
+def get_value(name: str) -> bool:
+    return get_config()['OPTIONS'].getboolean(name)
+
+
+def save_options(checkboxes: dict, master: tkinter.Tk) -> None:
+    config = get_config()
+    for option, var in checkboxes.items():
+        config['OPTIONS'][option] = str(var.get())
+    subs_lang = master.nametowidget('subs_menu').get()
+    config['OPTIONS']['subs_lang'] = subs_lang
+    resolution = master.nametowidget('resolution_menu').get()
+    config['OPTIONS']['resolution'] = resolution
+    bitrate = master.nametowidget('bitrate_entry').get()
+    if bitrate:
+        if len(bitrate) < 4:
+            for _ in range(4 - len(bitrate)):
+                bitrate += '0'
+        try:
+            bitrate = int(bitrate)
+        except ValueError:
+            bitrate = DEFAULT_BITRATE
+    config['OPTIONS']['bitrate'] = str(bitrate)
+    with open('HARDSUBBER.ini', 'w', encoding='utf-8') as config_file:
+        config.write(config_file)
+
+
+if __name__ == '__main__':
+    pass