minor fixes

pranjal-joshi · Apr 8, 2021 · b032868 · b032868
1 parent 6ae053b
commit b032868
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/screenipy.py b/screenipy.py
@@ -311,6 +311,8 @@ def setConfig(parser):
         parser.write(fp)
         fp.close()
         print(colorText.BOLD + colorText.GREEN +'[+] User configuration saved.' + colorText.END)
+        print(colorText.BOLD + colorText.GREEN +'[+] Restart the program now.' + colorText.END)
+        input('')
     except:
         print(colorText.BOLD + colorText.FAIL +'[+] Failed to save user config. Aborting..' + colorText.END)
         sys.exit(1)
@@ -426,4 +428,5 @@ def showConfigFile():
         saveResults.to_excel('screenipy-result.xlsx')
         print(colorText.BOLD + colorText.GREEN + "[+] Results saved to screenipy-result.xlsx" + colorText.END)
         print(colorText.BOLD + colorText.GREEN + "[+] Screening Completed! Happy Trading! :)" + colorText.END)
+        input('')
         sys.exit(0)