Merge pull request #55 from AndroidRetro/main

Don't hard-code the home directory.

Author: Tech-FZ

platformSpecific/unixSpecific.py

@@ -1,7 +1,7 @@
 import os
 import pwd
 import sqlite3
-
+homedir = os.getenv("HOME")
 def setupUnixBackend():
     try:
         userName = os.getlogin()
@@ -12,142 +12,48 @@ def setupUnixBackend():
     connection = None
 
     try:
-        if userName == "root":
-            connection = sqlite3.connect(f"/{userName}/EmuGUI/virtual_machines.sqlite")
-
-        else:
-            connection = sqlite3.connect(f"/home/{userName}/EmuGUI/virtual_machines.sqlite")
-
+        connection = sqlite3.connect(f"{homedir}/EmuGUI/virtual_machines.sqlite")
         print("Connection established.")
 
     except sqlite3.Error as e:
         print(f"The SQLite module encountered an error: {e}. Trying to create the file.")
 
         try:
             unixCreEmuGUIFolder()
-
-            if userName == "root":
-                file = open(f"/{userName}/EmuGUI/virtual_machines.sqlite", "w+")
-                file.close()
-
-            else:
-                file = open(f"/home/{userName}/EmuGUI/virtual_machines.sqlite", "w+")
-                file.close()
+            file = open(f"{homedir}/EmuGUI/virtual_machines.sqlite", "w+")
+            file.close()
 
         except:
             print("EmuGUI wasn't able to create the file.")
 
         try:
-            if userName == "root":
-                connection = sqlite3.connect(f"/{userName}/EmuGUI/virtual_machines.sqlite")
-
-            else:
-                connection = sqlite3.connect(f"/home/{userName}/EmuGUI/virtual_machines.sqlite")
+            connection = sqlite3.connect(f"{homedir}/EmuGUI/virtual_machines.sqlite")
 
         except sqlite3.Error as e:
             print(f"The SQLite module encountered an error: {e}.")
 
     return connection
 
 def unixTempVmStarterFile():
-    try:
-        userName = os.getlogin()
-    
-    except:
-        userName = pwd.getpwuid(os.getuid())[0]
-
-    if userName == "root":
-        fileName = f"/{userName}/EmuGUI/vmstart.txt"
-
-    else:
-        fileName = f"/home/{userName}/EmuGUI/vmstart.txt"
-
+    fileName = f"{homedir}/EmuGUI/vmstart.txt"
     return fileName
 
 def unixLanguageFile():
-    try:
-        userName = os.getlogin()
-    
-    except:
-        userName = pwd.getpwuid(os.getuid())[0]
-
-    if userName == "root":
-        fileName = f"/{userName}/EmuGUI/lang.txt"
-
-    else:
-        fileName = f"/home/{userName}/EmuGUI/lang.txt"
-        
+    fileName = f"{homedir}/EmuGUI/lang.txt"    
     return fileName
 
 def unixUpdateFile():
-    try:
-        userName = os.getlogin()
-    
-    except:
-        userName = pwd.getpwuid(os.getuid())[0]
-
-    if userName == "root":
-        fileName = f"/{userName}/EmuGUI/update.txt"
-
-    else:
-        fileName = f"/home/{userName}/EmuGUI/update.txt"
-
+    fileName = f"{homedir}/EmuGUI/update.txt"
     return fileName
 
 def unixExportFile():
-    try:
-        userName = os.getlogin()
-    
-    except:
-        userName = pwd.getpwuid(os.getuid())[0]
-
-    if userName == "root":
-        fileName = f"/{userName}/EmuGUI/vmdef.txt"
-
-    else:
-        fileName = f"/home/{userName}/EmuGUI/vmdef.txt"
-        
+    fileName = f"{homedir}/EmuGUI/vmdef.txt"    
     return fileName
-
 def unixErrorFile():
-    try:
-        userName = os.getlogin()
-    
-    except:
-        userName = pwd.getpwuid(os.getuid())[0]
-
-    if userName == "root":
-        fileName = f"/{userName}/EmuGUI/error.txt"
-
-    else:
-        fileName = f"/home/{userName}/EmuGUI/error.txt"
-        
+    fileName = f"{homedir}/EmuGUI/error.txt"    
     return fileName
-
 def unixLogFile(logID):
-    try:
-        userName = os.getlogin()
-    
-    except:
-        userName = pwd.getpwuid(os.getuid())[0]
-
-    if userName == "root":
-        fileName = f"/{userName}/EmuGUI/log-{logID}.txt"
-
-    else:
-        fileName = f"/home/{userName}/EmuGUI/log-{logID}.txt"
-        
+    fileName = f"{homedir}/EmuGUI/log-{logID}.txt"    
     return fileName
-
 def unixCreEmuGUIFolder():
-    try:
-        userName = os.getlogin()
-    
-    except:
-        userName = pwd.getpwuid(os.getuid())[0]
-
-    if userName == "root":
-        os.mkdir(f"/root/EmuGUI")
-
-    else:
-        os.mkdir(f"/home/{userName}/EmuGUI")
+    os.mkdir(f"{homedir}/EmuGUI")