Fixed temporary file deletion

bin/sshyp

@@ -254,12 +254,19 @@ def rename():  # renames an entry or folder TODO delete original from server to
 def edit():  # edits the contents of an entry
     print()
     system(f"cd {directory}; ls -1 *")  # TODO replace with new list
+    try:
         _entry_name = str(input('\nWhat file would you like to edit? '))
         if _entry_name.startswith('/'):
             _entry_name = _entry_name.replace('/', '', 1)
+    except KeyboardInterrupt:
+        _entry_name = None
+        print('\n')
+        s_exit()
     _shm_folder, _shm_entry = shm_gen()
+    try:
         system(f"gpg -d --output /dev/shm/{_shm_folder}/{_shm_entry} '{directory}{_entry_name}.gpg'")
         if not Path(f"/dev/shm/{_shm_folder}/{_shm_entry}").is_file():
+            rmtree(f"/dev/shm/{_shm_folder}")
             print("\nYou are trying to edit a file that does not exist!\n")
             s_exit()
         if argument == 'edit username' or argument == 'edit -u':
@@ -276,6 +283,10 @@ def edit():  # edits the contents of an entry
         remove(f"{directory}{_entry_name}.gpg")
         encrypt(_shm_folder, _shm_entry, _entry_name)
         system(f"gpg -d '{directory}{_entry_name}.gpg'")
+    except KeyboardInterrupt:
+        rmtree(f"/dev/shm/{_shm_folder}")
+        print('\n')
+        s_exit()
 
 
 def gen():  # generates a password for a new or an existing entry
@@ -430,11 +441,7 @@ elif argument == 'edit rename' or argument == 'edit relocate' or argument == 'ed
         s_exit()
 elif argument == 'edit username' or argument == 'edit -u' or argument == 'edit password' or argument == 'edit -p' or \
         argument == 'edit url' or argument == 'edit -l' or argument == 'edit note' or argument == 'edit -n':
-    try:
+    edit()  # KeyboardInterrupt exception in function for temporary file deletion
-        edit()
-    except KeyboardInterrupt:
-        print('\n')
-        s_exit()
 elif argument == 'gen' or argument == 'gen update' or argument == 'gen -u':
     try:
         gen()

share/doc/sshyp/changelog

@@ -13,7 +13,7 @@ Changes:
 - printing functions in sshyp have been combined into one function (arguments determine section to print)
 ** file not found exceptions have been re-implemented
 ** when syncing, a check is made to see if any folders are not on all devices - if the check comes back true, it is corrected
-** folders generated in /dev/shm are once again properly deleted with keyboard interrupts
+- folders generated in /dev/shm are once again properly deleted with keyboard interrupts
 - .lower() is now used to remove casing when checking inputs
 - fixed syncing of files (specifically downloading) with special characters, such as '('
 - fixed program still continuing after trying to edit an entry that doesn't exist