Rename source dir

src/bonsai/scripts/get_wheels.py

@@ -0,0 +1,32 @@
+import os
+import sys
+
+
+def find_whl_files(directory):
+    whl_files = []
+    for root, dirs, files in os.walk(directory):
+        for file in files:
+            if file.endswith(".whl"):
+                whl_files.append(os.path.relpath(os.path.join(root, file), directory))
+    return whl_files
+
+
+def update_pyproject_toml(pyproject_path, whl_files):
+    with open(pyproject_path, "a") as f:
+        f.write("\nwheels = [\n")
+        for whl in whl_files:
+            f.write(f'    "./wheels/{whl}",\n')
+        f.write("]\n")
+
+
+if __name__ == "__main__":
+    if len(sys.argv) != 3:
+        print("Usage: python script.py <folder_path> <pyproject.toml_path>")
+        sys.exit(1)
+
+    folder_path = sys.argv[1]
+    pyproject_toml_path = sys.argv[2]
+
+    whl_files = find_whl_files(folder_path)
+    update_pyproject_toml(pyproject_toml_path, whl_files)
+    print("pyproject.toml has been updated with the wheel files.")