Update rotateMusic.py

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 = {}