Creating shortcuts in installer

horsy_installer.py

@@ -6,6 +6,9 @@ import urllib.request
 import os
 import threading
 import ctypes
+import winshell
+from win32com.client import Dispatch
+import pythoncom
 
 
 class Ui_MainWindow(object):
@@ -209,6 +212,20 @@ def install():
         for thread in threads:
             thread.join()
         ui.logs_box.append("Downloading finished")
+        if ui.install_gui_check.isChecked():
+            ui.logs_box.append("Creating shortcuts")
+            desktop = winshell.desktop()
+            path = os.path.join(desktop, "horsy GUI.lnk")
+            target = os.path.join(path_to_install) + '/horsygui.exe'
+            wDir = os.path.join(path_to_install)
+            icon = os.path.join(path_to_install) + '/horsygui.exe'
+            pythoncom.CoInitializeEx(0)
+            shell = Dispatch('WScript.Shell')
+            shortcut = shell.CreateShortCut(path)
+            shortcut.Targetpath = target
+            shortcut.WorkingDirectory = wDir
+            shortcut.IconLocation = icon
+            shortcut.save()
         ui.logs_box.append("Installation complete")
 
     threading.Thread(target=wait_for_finish).start()

modules/change_password.py

@@ -1,7 +1,7 @@
 import requests
 import modules.gui as gui
 import threading
-from modules.auth import del_auth
+from modules.auth import del_auth, get_auth
 import modules.vars as horsy_vars
 import json
 from PyQt5 import QtWidgets
@@ -40,6 +40,9 @@ def change(oldpass, newpass):
 
             elif r == '':
                 gui.popup('Success', 'Success, your password has been changed')
+                config['auth'] = {'email': config['auth']['email'], 'password': newpass}
+                with open(horsy_vars.horsypath + 'config.cfg', 'w') as f:
+                    json.dump(config, f)
 
             else:
                 print('[red]Unknown error, please try again[/red]')

requirements.txt

@@ -6,3 +6,4 @@ tqdm
 algoliasearch
 PyQt5
 easygui
+winshell