GUI development

modules/auth.py

@@ -2,7 +2,7 @@ import json
 import modules.vars as horsy_vars
 
 
-def get_auth():
+def get_auth(is_gui=False, login_ui=None):
     with open(horsy_vars.horsypath + 'config.cfg') as f:
         config = json.load(f)
 
@@ -12,16 +12,29 @@ def get_auth():
         else:
             raise Exception('No auth found')
     except:
-        print('[!] No auth found, please login first')
-        print('email')
-        email = input('> ')
-        print('password')
-        password = input('> ')
-        config['auth'] = {'email': email, 'password': password}
+        if not is_gui:
+            print('[!] No auth found, please login first')
+            print('email')
+            email = input('> ')
+            print('password')
+            password = input('> ')
+            config['auth'] = {'email': email, 'password': password}
+            with open(horsy_vars.horsypath + 'config.cfg', 'w') as f:
+                json.dump(config, f)
+            print('[OK] Auth created')
+            return config['auth']
+        else:
+            login_ui.show()
+            login_ui.login_button.clicked.connect(lambda: get_auth(login_ui=login_ui))
+
+
+def get_gui_auth(login_ui):
+    with open(horsy_vars.horsypath + 'config.cfg') as f:
+        config = json.load(f)
+    if login_ui.email_box.text() != '' and login_ui.password_box.text() != '':
         with open(horsy_vars.horsypath + 'config.cfg', 'w') as f:
+            config['auth'] = {'email': login_ui.email_box.text(), 'password': login_ui.password_box.text()}
             json.dump(config, f)
-        print('[OK] Auth created')
-        return config['auth']
 
 
 def del_auth():