From 3acee4a04bf09b7659dd6b04c60c2a0cd5b1eef4 Mon Sep 17 00:00:00 2001
From: Cassandra de la Cruz-Munoz <me@cass-dlcm.dev>
Date: Sun, 22 Oct 2023 21:01:33 +0200
Subject: [PATCH] Update rotateMusic.py

---
 rotateMusic.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/rotateMusic.py b/rotateMusic.py
index 8dc30c3..a394108 100644
--- a/rotateMusic.py
+++ b/rotateMusic.py
@@ -1,7 +1,7 @@
 import os
 import random
 import shutil
-import sys
+import argparse
 
 
 def get_dir_size(path):
@@ -16,11 +16,14 @@ def get_dir_size(path):
 
 
 def main():
-    if len(sys.argv) != 4:
-        return
-    library = sys.argv[1]
-    player = sys.argv[2]
-    amount = int(sys.argv[3])
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--library")
+    parser.add_argument("--playerStorage")
+    parser.add_argument("--maxBytes")
+    args = parser.parse_args()
+    library = args["library"]
+    player = args["playerStorage"]
+    amount = int(args["maxBytes"])
     libraryArtists = os.listdir(library)
     playerArtists = os.listdir(player)
     librarySpace = {}