Update transfer.py

transfer.py

@@ -6,15 +6,19 @@ import platform
 
 import subprocess
 
+
 def get_windows_drive_name(letter):
-    return subprocess.check_output(["cmd","/c vol " + letter]).decode().split("\r\n")[0].split(" ").pop()
+    return subprocess.check_output(["cmd", "/c vol " + letter]).decode().split("\r\n")[0].split(" ").pop()
 
 
 if platform.system().lower() == 'windows':
-    disks = [(x.mountpoint, get_windows_drive_name(x.mountpoint.strip('\\'))) for x in psutil.disk_partitions()]
-    disks = [(x, name) for x, name in disks if 'TRINKEY' in name or 'CIRCUIT' in name] 
+    disks = [(x.mountpoint, get_windows_drive_name(x.mountpoint.strip('\\')))
+             for x in psutil.disk_partitions()]
+    disks = [(x, name)
+             for x, name in disks if 'TRINKEY' in name or 'CIRCUIT' in name]
 else:
-    disks = [(x.mountpoint, x.mountpoint) for x in psutil.disk_partitions() if 'TRINKEY' in x.mountpoint.upper() or 'CIRCUIT' in x.mountpoint.upper()]
+    disks = [(x.mountpoint, x.mountpoint) for x in psutil.disk_partitions(
+    ) if 'TRINKEY' in x.mountpoint.upper() or 'CIRCUIT' in x.mountpoint.upper()]
 
 if len(disks) != 1:
     print("Could not find the trinkey.  Did you click the reset button twice and connect it?")
@@ -33,3 +37,4 @@ else:
     shutil.rmtree(os.path.join('src', '__pycache__'), True)
     shutil.copytree('src', os.path.join(disk, 'src'), dirs_exist_ok=True)
     shutil.copy('code.py', os.path.join(disk, 'code.py'))
+    print("Done! Switch back to the Serial Monitor!")