Remove temporary file after upload and download.

diff --git a/gdrive_knife.py b/gdrive_knife.py
index 81e3076a7cc7d377244387be74742288445279fc..605ea7e2518da0a2a4203e685a9bb7c891a8aa93 100644 (file)
--- a/gdrive_knife.py
+++ b/gdrive_knife.py
@@ -146,6 +146,10 @@ def download(args):
         shutil.move(path_in_tmp, destination)
         print(f'Moved to {destination}.')
 
+    if os.path.exists(path_in_tmp):
+        os.remove(path_in_tmp)
+        print(f'Removed {path_in_tmp}.')
+
 def upload(args):
     drive = get_drive_client(args.token)
 
@@ -178,6 +182,9 @@ def upload(args):
         drive.files().create(body=body, media_body=media).execute()
         print(f'Created {path} on drive.')
 
+    os.remove(path_in_tmp)
+    print(f'Removed {path_in_tmp}.')
+
 def delete(args):
     drive = get_drive_client(args.token)