From 9a136dc3097bc5a4cd989adbdee3f6c6733aa18c Mon Sep 17 00:00:00 2001
From: Adam Blank <blank@caltech.edu>
Date: Fri, 11 Oct 2024 23:18:15 +0000
Subject: [PATCH] Update transfer.py
---
transfer.py | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/transfer.py b/transfer.py
index 358ea6a..98b4888 100644
--- a/transfer.py
+++ b/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!")
--
GitLab