qBittorrent search engine plugins
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

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)