Improvements to dte.py

dte.py

@@ -10,19 +10,35 @@ import sys
 
 
 try:
-    name = sys.argv[1]
-    path = sys.argv[2]
+    name: str = sys.argv[1]
+    path: str = sys.argv[2]
 except IndexError:
-    print("Name and exec path arguments are required")
-    exit()
+    print("Usage: dte NAME_OF_ENTRY EXEC_PATH")
+    exit(1)
 
 try:
     with open(f"/home/aaro/.local/share/applications/{name}.desktop", "r") as f:
         print("Desktop entry already exists");
-        exit()
+        exit(0)
 except FileNotFoundError:
     pass
 
+msg_str: str = f"""
+This will create a destop entry with the following values:
+    Name={name}
+    Exec={path}
+"""
+print(msg_str)
+
+while True:
+    sure: str = input("Are you sure? (y/n): ")
+    sure = sure.lower()
+    if sure == "y" or sure == "yes":
+        break
+    elif sure == "n" or sure == "no":
+        exit(0)
+
+
 with open(f"/home/aaro/.local/share/applications/{name}.desktop", "x") as f:
     f.writelines([
         "[Desktop Entry]\n",