Update sysinfo.py

sysinfo.py

@@ -1,24 +1,41 @@
 #Sysinfo.py - Get info about computer and OS
-#This code has only been tested on Ubuntu 20.04 LTS and may not work on any other distros of Linux.
+#Only works if you run this program on Linux since this program calls a bash script
+#This code has only been tested on Ubuntu 20.04 LTS and may not work on any other distros of Linux
 
 #Import all required libraries/modules
 import subprocess as sp
 import platform
 import os
+import sys
+import time
+
+#Create timer to get execution time
+start = time.time()
 
 #Ask user if they have the Linux command line tool figlet installed
 figlet_install = input("Do you have figlet installed on your linux terminal (Y/N)? ")
 
-#If they already have it installed, move on to the next part in the code
+#If they already have it installed, don't do anything and move on to the next part in the code
 if figlet_install.upper() == "Y":
     pass
 
-#If user does not have figlet installed, in the background the program runs the Linux command for installing figlet.
+#If user does not have figlet installed, in the background the program runs the Linux command for installing figlet
 if figlet_install.upper() == "N":
+    print("Installing figlet...")
     os.system("sudo apt-get install figlet")
-    
+
+#If user doesn't enter y or n, throw an error exiting the program
+if figlet_install.upper() != "Y" and "N":
+    sys.exit("Invalid input. Your input must either be Y or N (uppercase is not mandatory).")
+
 #Run figlet and get the name of the user's computer
 os.system("figlet %s" %(platform.node()))
 
 #Execute bash script sysinfo.sh
-sp.call("sysinfo.sh")
+sp.call("/mnt/e/bash/welcomewagon.sh")
+
+#End timer for findingexecution time of program
+end = time.time()
+
+#Display execution time
+print("Program execution time: %s seconds" %(end - start))