diff --git a/README.md b/README.md index c30209e..d06e3fd 100644 --- a/README.md +++ b/README.md @@ -30,17 +30,12 @@ zutui `ctrl`+`q` - Quit # Sidenotes -- config is stored @ `~/zutui/config.json` +## Config +- On Unix like is stored @ `~/.config/zutui/config.json` +- On Windows is stored @ `~/zutui/config.json` - so are `grades_cache.json` -- in plaintext..... -##### As of right now I don't see any MUCH better way. - -##### I mean we could simply encrypt and decrypt it during runtime, but... does it really make it any more secure? - -##### And we can't really store it in any form of a _hash_ as we have to log in everytime with plain text anyways... - -##### However feel free to pull request if you're smarter :) +##### Feel free to pull request if you're smarter :) # TODO: - [x] **Nothing** diff --git a/setup.py b/setup.py index 75bd356..d2802d6 100644 --- a/setup.py +++ b/setup.py @@ -8,10 +8,12 @@ setup( "requests", "beautifulsoup4", "textual", + "keyring", + "pywin32" ], entry_points={ 'console_scripts': [ 'zutui=zut_app.zutui:main', ], }, -) \ No newline at end of file +) diff --git a/zut_app/zutui.py b/zut_app/zutui.py index 6c25d68..078afa1 100644 --- a/zut_app/zutui.py +++ b/zut_app/zutui.py @@ -1,5 +1,7 @@ import json import os +import keyring +from keyring.errors import KeyringError from datetime import datetime from textual.app import App, ComposeResult from textual.containers import Container @@ -54,9 +56,20 @@ class LoginScreen(Screen): if success: try: with open(CONFIG_FILE, "w") as f: - json.dump({"username": user, "password": password}, f) + json.dump({"username": user}, f) + keyring.set_password("zutui", user, password) + except KeyringError as e: + error = e + self.app.call_from_thread( + lambda: + self.app.notify( + f"Błąd: {str(error)}", + title="Błąd zapisu hasła do systemowego menedżera haseł", + severity="warning", + timeout=5 + ) + ) except: pass - def do_switch(): self.app.switch_screen(DashboardScreen()) self.app.call_from_thread(do_switch) @@ -406,7 +419,10 @@ class ZutApp(App): try: with open(CONFIG_FILE, "r") as f: creds = json.load(f) - self.zut_client = ZUT(creds["username"], creds["password"]) + password = keyring.get_password("zutui", creds["username"]) + if password is None: + raise ValueError("Hasło nie zostało znalezione") + self.zut_client = ZUT(creds["username"], password) self.push_screen(DashboardScreen()) except: self.push_screen(LoginScreen())