From e9ba63b9d0da14c5f125f852baeebc548bc00dfc Mon Sep 17 00:00:00 2001
From: yiannisha <hadjiyiannis.3@gmail.com>
Date: Sat, 23 Oct 2021 17:15:41 +0300
Subject: [PATCH] Added argument parsing in search function

Added a block for parsing arguments passed with options
-l -limit
-t -type
in the search() function. Arguments are passed inside search_term.
They work as follows:

* -l -limit <number> : sets the limit of results to that number
  and raises a ValueError if that number exceeds 50. Default is 10.

* -t -type <album/track/playlist> : sets the type that is requested
  from the API about the search query. Raises a ValueError if an arguments
  passed is different than track, album, playlist. Default is all three.

Example:
Enter search or URL: <query> -l 30 -t track album

This will result with 30 tracks and 30 albums associated with query.
Options can be passed in any order but the query must be first.
---
 zspotify.py | 51 ++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 48 insertions(+), 3 deletions(-)

diff --git a/zspotify.py b/zspotify.py
index d031ea9c..a58790f8 100755
--- a/zspotify.py
+++ b/zspotify.py
@@ -354,10 +354,55 @@ def search(search_term):
     params = {
         "limit" : "10",
         "q" : search_term,
-        "type" : ["track","album","playlist"]
+        "type" : set(),
     }
 
-    # ADD BLOCK FOR READING OPTIONS AND CLEAN SEARCH_TERM
+    # Block for parsing passed arguments
+    splits = search_term.split()
+    for split in splits:
+        index = splits.index(split)
+
+        if split[0] == "-" and len(split) > 1:
+            if len(splits)-1 == index:
+                raise IndexError("No parameters passed after option: {}\n".
+                                 format(split))
+
+        if split == "-l" or split == "-limit":
+            try:
+                int(splits[index+1])
+            except ValueError:
+                raise ValueError("Paramater passed after {} option must be an integer.\n".
+                                 format(split))
+            if int(splits[index+1]) > 50:
+                raise ValueError("Invalid limit passed. Max is 50.\n")
+            params["limit"] = splits[index+1]
+
+
+        if split == "-t" or split == "-type":
+
+            allowed_types = ["track", "playlist", "album"]
+            for i in range(index+1, len(splits)):
+                if splits[i][0] == "-":
+                    break
+
+                if splits[i] not in allowed_types:
+                    raise ValueError("Parameters passed after {} option must be from this list:\n{}".
+                                     format(split, '\n'.join(allowed_types)))
+
+                params["type"].add(splits[i])
+
+    if len(params["type"]) == 0:
+        params["type"] = {"track", "album", "playlist"}
+
+    # Clean search term
+    search_term_list = []
+    for split in splits:
+        if split[0] == "-":
+            break
+        search_term_list.append(split)
+    if not search_term_list:
+        raise ValueError("Invalid query.")
+    params["q"] = ' '.join(search_term_list)
 
     resp = requests.get(
         "https://api.spotify.com/v1/search",
@@ -437,8 +482,8 @@ def search(search_term):
 
     if "playlist" in params["type"]:
         playlists = resp.json()["playlists"]["items"]
-        print("###  PLAYLISTS  ###")
         if len(playlists) > 0:
+            print("###  PLAYLISTS  ###")
             for playlist in playlists:
                 # collect needed data
                 dic = {