munyal-client/config.py

124 lines
3.3 KiB
Python

import os
import pathlib
import pickle
from hashlib import sha256
from tkinter import Button, Entry, Label, StringVar, Tk, filedialog
from PIL import Image, ImageTk
from misc import alert, get_os
def __get_config_folder():
_os = get_os()
if _os == "linux":
folder = os.path.join(os.getenv("HOME"), ".munyal")
else:
folder = os.path.join(os.path.expandvars("%APPDATA%"), "Munyal")
return folder
def get_config():
while True:
config = __get_config()
if not config:
gui_config()
else:
return config
def __get_config():
folder = __get_config_folder()
config_file = os.path.join(folder, "config")
if os.path.exists(config_file):
config_bytes = open(config_file, "rb")
config = pickle.load(config_bytes)
config_bytes.close()
return config
def __validate_values(user, password):
return True
def _set_config(root, user, password, folder):
passwd_hash = sha256(password.get().encode("utf-8")).hexdigest()
if not user:
alert(root, "Introduce tu usuario de la red Munyal")
elif not password:
alert(root, "Introduce tu contraseña")
elif not folder:
alert(root, "Elige la carpeta que quieres sincronizar")
elif not __validate_values(user, passwd_hash):
alert(
root,
"Tu usuario o contraseña son incorrectos, favor de reintentarlo")
else:
password.set("")
folder_config = __get_config_folder()
path = pathlib.Path(folder_config)
path.mkdir(parents=True, exist_ok=True)
with open(os.path.join(folder_config, "config"), "wb") as config_file:
config_file.write(
pickle.dumps({
"login": {
"user": user,
"password": passwd_hash
},
"folder": folder
}))
root.destroy()
def gui_config():
root = Tk()
root.geometry("250x420")
img_logo = Image.open("img/logo.png")
img_tk = ImageTk.PhotoImage(img_logo.resize((200, 150), Image.ANTIALIAS))
host = StringVar()
host.set("localhost")
host_field = Entry(root, textvariable=host)
passwd = StringVar()
passwd_field = Entry(root, textvariable=passwd, show="*")
folder = StringVar()
folder.set(os.path.join(os.getenv("HOME"), "Munyal"))
folder_field = Entry(root, textvariable=folder)
btn_folder = Button(root,
text="Examinar",
command=lambda: search_folder(folder))
connect = Button(
root,
text="Conectar",
command=lambda: _set_config(root, host.get(), passwd, folder.get()))
Label(root, text="MUNYAL").pack()
Label(root, image=img_tk).pack()
Label(root, text="").pack()
Label(root, text="Ruta del servidor").pack()
host_field.pack()
Label(root, text="").pack()
Label(root, text="Contraseña").pack()
passwd_field.pack()
Label(root, text="").pack()
Label(root, text="Carpeta a sincronizar").pack()
folder_field.pack()
btn_folder.pack()
Label(root, text="").pack()
connect.pack()
root.mainloop()
def search_folder(field):
path = filedialog.askdirectory()
field.set(path)
if __name__ == "__main__":
print(get_config())