mirror of https://github.com/YGGverse/qBt_SE.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
117 lines
4.7 KiB
117 lines
4.7 KiB
import json |
|
import tkinter as tk |
|
from pathlib import Path |
|
from tkinter import ttk, messagebox |
|
|
|
|
|
class EngineSettingsGUI: |
|
def __init__(self, engine_name: str): |
|
self.config = dict() |
|
self.cfg_file = Path(engine_name + ".json") |
|
if self.cfg_file.exists(): |
|
self.config = json.loads(self.cfg_file.read_text()) |
|
|
|
self.window = tk.Tk() |
|
self.window.title(engine_name.capitalize() + " Settings") |
|
|
|
mainframe = ttk.Frame(self.window, padding="10") |
|
mainframe.grid(column=0, row=0, sticky=tk.N) |
|
self.window.columnconfigure(0, weight=1) |
|
self.window.rowconfigure(0, weight=1) |
|
|
|
self.username = tk.StringVar(value=self.config.get("username", "")) |
|
self.password = tk.StringVar(value=self.config.get("password", "")) |
|
self.proxy_http = tk.StringVar( |
|
value=self.config.get("proxies", dict()).get("http", "") |
|
) |
|
self.proxy_https = tk.StringVar( |
|
value=self.config.get("proxies", dict()).get("https", "") |
|
) |
|
|
|
self.date = tk.BooleanVar(value=self.config.get("torrentDate", True)) |
|
self.magnet = tk.BooleanVar(value=self.config.get("magnet", False)) |
|
self.proxy = tk.BooleanVar(value=self.config.get("proxy", False)) |
|
|
|
ttk.Label(mainframe, text="Username:").grid( |
|
column=0, row=0, sticky=tk.W) |
|
ttk.Label(mainframe, text="Password:").grid( |
|
column=0, row=1, sticky=tk.W, rowspan=2) |
|
|
|
ttk.Entry(mainframe, width=25, textvariable=self.username, state=( |
|
("!" if self.config.get("username") else "") + tk.DISABLED) |
|
).grid(column=1, row=0, sticky=tk.EW, padx=(0, 5)) |
|
ttk.Entry(mainframe, width=25, textvariable=self.password, state=( |
|
("!" if self.config.get("password") else "") + tk.DISABLED) |
|
).grid(column=1, row=1, rowspan=2, sticky=tk.EW, padx=(0, 5)) |
|
|
|
ttk.Checkbutton( |
|
mainframe, text="Date before torrent", variable=self.date, |
|
onvalue=True |
|
).grid(column=2, row=0, sticky=tk.W) |
|
ttk.Checkbutton( |
|
mainframe, text="Use magnet link", variable=self.magnet, |
|
onvalue=True, state=( |
|
("!" if self.config.get("magnet") else "") + tk.DISABLED) |
|
).grid(column=2, row=1, sticky=tk.W) |
|
ttk.Checkbutton( |
|
mainframe, text="Proxy", variable=self.proxy, onvalue=True, |
|
command=self.proxy_action |
|
).grid(column=2, row=2, sticky=tk.W) |
|
|
|
ttk.Label(mainframe, text="HTTP:").grid(column=0, row=3, sticky=tk.W) |
|
ttk.Label(mainframe, text="HTTPS:").grid(column=0, row=4, sticky=tk.W) |
|
|
|
proxy_state = tk.NORMAL if self.proxy.get() else tk.DISABLED |
|
self.http_entry = ttk.Entry( |
|
mainframe, textvariable=self.proxy_http, state=proxy_state |
|
) |
|
self.http_entry.grid(column=1, row=3, sticky=tk.EW, |
|
padx=(0, 5), pady=(0, 5)) |
|
self.https_entry = ttk.Entry( |
|
mainframe, textvariable=self.proxy_https, state=proxy_state |
|
) |
|
self.https_entry.grid(column=1, row=4, sticky=tk.EW, padx=(0, 5)) |
|
|
|
ttk.Button( |
|
mainframe, text="Save", command=self.close |
|
).grid(column=2, row=3, rowspan=2) |
|
|
|
self.window.mainloop() |
|
|
|
def proxy_action(self) -> None: |
|
state = ("!" if self.proxy.get() else "") + tk.DISABLED |
|
self.http_entry.state([state]) |
|
self.https_entry.state([state]) |
|
|
|
def close(self) -> None: |
|
if self.config.get("username") and self.config.get("password"): |
|
if not (self.username.get() or self.password.get()): |
|
messagebox.showinfo("Error", "Some fields is empty!") |
|
return None |
|
|
|
if self.proxy.get() and not (self.http_entry.get() |
|
or self.https_entry.get()): |
|
messagebox.showinfo("Error", "Some fields is empty!") |
|
return None |
|
|
|
if self.config.get("username") and self.config.get("password"): |
|
self.config["username"] = self.username.get() |
|
self.config["password"] = self.password.get() |
|
self.config["proxy"] = self.proxy.get() |
|
if self.config["proxy"]: |
|
self.config["proxies"] = { |
|
"http": self.http_entry.get(), |
|
"https": self.https_entry.get() |
|
} |
|
self.config["torrentDate"] = self.date.get() |
|
if self.config.get("magnet"): |
|
self.config["magnet"] = self.magnet.get() |
|
self.cfg_file.write_text( |
|
json.dumps(self.config, indent=4, sort_keys=False) |
|
) |
|
self.window.destroy() |
|
|
|
|
|
if __name__ == "__main__": |
|
settings = EngineSettingsGUI("engines/kinozal") |
|
print(settings.config)
|
|
|