Simplify timestamp format in output folder name to YYYY-MM-DD_HHMMSS

src/app.py

@@ -2,13 +2,28 @@ import os
 import shutil
 from pathlib import Path
 from mutagen.id3 import ID3
+from datetime import datetime
+from collections import defaultdict
 
-CD_SIZE = 700 * 1024 * 1024  # 700 MB
+CD_SIZE = 695 * 1024 * 1024  # 695 MB
 GROUP_SIZE = 5
 
-dir_a = Path(r"C:\Users\Edgar\Desktop\Peak Time BOF (ed1337x)\Already where in MP3")
+# Store the current date
-dir_b = Path(r"C:\Users\Edgar\Desktop\Peak Time BOF (ed1337x)\CD")
+run_date = datetime.now().strftime("%Y-%m-%d_%H%M%S")
-dir_b.mkdir(parents=True, exist_ok=True)
+
+# Ask for source path, strip quotes if pasted
+source_input = input("Drag and drop your music folder here, then press Enter: ").strip().strip('"')
+source_media_path = Path(source_input)
+
+if not source_media_path.exists() or not source_media_path.is_dir():
+    print(f"Error: {source_media_path} is not a valid directory.")
+    exit(1)
+
+# Build destination path
+parent = source_media_path.parent
+folder_name = source_media_path.name
+dest_media_path = parent / f"[CDs-{run_date}]{folder_name}"
+dest_media_path.mkdir(parents=True, exist_ok=True)
 
 def get_bpm(file_path):
     try:
@@ -20,35 +35,38 @@ def get_bpm(file_path):
         print(f"Skipping {file_path.name} (no BPM): {e}")
     return None
 
-# Collect all tracks
+# Collect all tracks recursively, grouping unknown BPM separately
 all_tracks = []
-for file in dir_a.glob("*.mp3"):
+for file in source_media_path.rglob("*.mp3"):  # <-- recursive glob
     bpm = get_bpm(file)
-    if bpm is None:
-        continue
     size = file.stat().st_size
-    bpm_range = f"{(bpm // GROUP_SIZE) * GROUP_SIZE}-to-{((bpm // GROUP_SIZE) * GROUP_SIZE) + GROUP_SIZE - 1}"
+    if bpm is None:
+        bpm_range = "[Unknown BPM]"
+    else:
+        bpm_range = f"{(bpm // GROUP_SIZE) * GROUP_SIZE}-to-{((bpm // GROUP_SIZE) * GROUP_SIZE) + GROUP_SIZE - 1}"
     all_tracks.append({"file": file, "size": size, "bpm_range": bpm_range})
 
+if not all_tracks:
+    print("No MP3 files found.")
+    exit(0)
+
 # Calculate total size and number of CDs needed
 total_size = sum(t["size"] for t in all_tracks)
-num_cds = max(1, (total_size + CD_SIZE - 1) // CD_SIZE)  # ceiling division
+num_cds = max(1, (total_size + CD_SIZE - 1) // CD_SIZE)
 
 print(f"Total size: {total_size / 1024**2:.2f} MB, splitting into {num_cds} CDs")
 
 # Group tracks by BPM range
-from collections import defaultdict
 bpm_groups = defaultdict(list)
 for track in all_tracks:
     bpm_groups[track["bpm_range"]].append(track)
 
-# Sort each bpm group by filename (or size if you want)
+# Sort each bpm group
 for bpm_range in bpm_groups:
     bpm_groups[bpm_range].sort(key=lambda x: x["file"].name)
 
 # Split each BPM group evenly into num_cds parts
 def split_evenly(lst, n):
-    """Split list lst into n chunks as evenly as possible by count."""
     k, m = divmod(len(lst), n)
     return [lst[i*k + min(i, m):(i+1)*k + min(i+1, m)] for i in range(n)]
 
@@ -59,14 +77,14 @@ for bpm_range, tracks in bpm_groups.items():
 # Prepare CDs
 cd_contents = [[] for _ in range(num_cds)]
 
-# Fill CDs with chunks from each BPM range
+# Fill CDs with balanced BPM chunks
 for bpm_range in bpm_chunks:
     for i, chunk in enumerate(bpm_chunks[bpm_range]):
         cd_contents[i].extend(chunk)
 
 # Write CDs with BPM subfolders
 for i, tracks in enumerate(cd_contents, start=1):
-    cd_folder = dir_b / f"CD-{i:02}"
+    cd_folder = dest_media_path / f"CD-{i:02}"
     cd_folder.mkdir(parents=True, exist_ok=True)
     size_accum = 0
     for track in tracks:
@@ -76,3 +94,5 @@ for i, tracks in enumerate(cd_contents, start=1):
         size_accum += track["size"]
     print(f"[WRITE] CD-{i:02}: {len(tracks)} tracks, approx {size_accum/1024**2:.2f} MB")
 
+print("\n✅ Done!")
+