Add analytics script to analyze decade, allow fetching specific playlist

Author: khrj

README.md

@@ -40,6 +40,19 @@ To automatically backup (in a script, on a schedule...):
 poetry run python src/main.py --backup
 ```
 
+To get only a specific playlist
+
+```bash
+poetry run python src/main.py --backup --playlist <name of playlist>
+```
+
+To get decade-distribution for any file
+
+```bash
+poetry run python src/main.py --file <path to file>
+```
+
+
 ## Time taken
 
 > **Quick restore causes liked songs to lose order. Playlists are always ordered

poetry.lock

@@ -0,0 +1,34 @@
+from json import load
+
+
+def analyze(file_path):
+    raw_data = load(open(file_path, "r"))
+    if "tracks" in raw_data:
+        raw_data = raw_data["tracks"]
+
+    tracks = sorted(
+        [
+            int(x["release_date"].split("-")[0])
+            for x in raw_data
+            if x["release_date"] != "0000"
+        ]
+    )
+
+    by_decade = {}
+
+    for t in tracks:
+        decade = str(t)[:-1] + "0s"
+        if decade in by_decade:
+            by_decade[decade] = by_decade[decade] + 1
+        else:
+            by_decade[decade] = 1
+
+        if decade == "0s":
+            print(t)
+
+    print(by_decade)
+
+
+# import matplotlib.pyplot as plt
+# plt.pie(data.values(), labels=data.keys())
+# plt.show()

src/analyze.py

@@ -161,81 +161,87 @@ def write(data, path):
     dump(data, open(path, "w"), indent="\t")
 
 
-def backup(sp: Spotify):
+def backup(sp: Spotify, playlist_name=None):
     print("Backing up... This might take a while")
 
     backup = Path("backup")
     backup.mkdir(parents=True, exist_ok=True)
 
-    git_status = subprocess.run(
-        ["git", "status", "--porcelain"], cwd=backup, capture_output=True, text=True
-    )
+    if not playlist_name:
+        git_status = subprocess.run(
+            ["git", "status", "--porcelain"], cwd=backup, capture_output=True, text=True
+        )
 
-    if git_status.returncode != 0:
-        subprocess.run(["git", "init"], cwd=backup).check_returncode()
+        if git_status.returncode != 0:
+            subprocess.run(["git", "init"], cwd=backup).check_returncode()
 
-    if git_status.stdout:
-        print("You have uncommitted changes in backup/, exiting")
-        exit(1)
+        if git_status.stdout:
+            print("You have uncommitted changes in backup/, exiting")
+            exit(1)
 
-    subprocess.run(["git", "pull"], cwd=backup).check_returncode()
+        subprocess.run(["git", "pull"], cwd=backup).check_returncode()
 
-    rmtree("backup/playlists", ignore_errors=True)
-    Path("backup/liked-songs.json").unlink(missing_ok=True)
-    Path("backup/saved-albums.json").unlink(missing_ok=True)
-    Path("backup/followed-artists.json").unlink(missing_ok=True)
-    Path("backup/blend-names.json").unlink(missing_ok=True)
+        rmtree("backup/playlists", ignore_errors=True)
+        Path("backup/liked-songs.json").unlink(missing_ok=True)
+        Path("backup/saved-albums.json").unlink(missing_ok=True)
+        Path("backup/followed-artists.json").unlink(missing_ok=True)
+        Path("backup/blend-names.json").unlink(missing_ok=True)
 
     Path("backup/playlists/owned").mkdir(parents=True, exist_ok=True)
     Path("backup/playlists/collaborative").mkdir(parents=True, exist_ok=True)
     Path("backup/playlists/followed").mkdir(parents=True, exist_ok=True)
 
-    print("Backing up liked songs...")
-    songs = get_liked_songs(sp)
-    write(songs, "backup/liked-songs.json")
+    if not playlist_name:
+        print("Backing up liked songs...")
+        songs = get_liked_songs(sp)
+        write(songs, "backup/liked-songs.json")
 
-    print("Backing up albums...")
-    albums = get_albums(sp)
-    write(albums, "backup/saved-albums.json")
+        print("Backing up albums...")
+        albums = get_albums(sp)
+        write(albums, "backup/saved-albums.json")
 
-    print("Backing up followed artists...")
-    followed = get_followed_artists(sp)
-    write(followed, "backup/followed-artists.json")
+        print("Backing up followed artists...")
+        followed = get_followed_artists(sp)
+        write(followed, "backup/followed-artists.json")
 
     print("Backing up playlists...")
     playlists, blends = get_playlists(sp)
     for playlist in playlists:
+        if playlist_name and playlist["name"] != playlist_name:
+            continue
+
         playlist["tracks"] = get_playlist_tracks(sp, playlist)
         write(
             playlist,
             f"backup/playlists/{playlist['type']}/{slugify(playlist['name'])}-{slugify(playlist['id'])}.json",
         )
 
-    write(blends, "backup/blend-names.json")
+    if not playlist_name:
+        write(blends, "backup/blend-names.json")
 
-    print("Commiting and pushing changes...")
+        print("Commiting and pushing changes...")
 
-    subprocess.run(["git", "add", "."], cwd=backup).check_returncode()
+        subprocess.run(["git", "add", "."], cwd=backup).check_returncode()
 
-    subprocess.run(
-        "git diff-index --quiet HEAD || git commit -m 'Automated update'",
-        cwd=backup,
-        shell=True,
-    ).check_returncode()
+        subprocess.run(
+            "git diff-index --quiet HEAD || git commit -m 'Automated update'",
+            cwd=backup,
+            shell=True,
+        ).check_returncode()
 
-    subprocess.run(["git", "push"], cwd=backup)
+        subprocess.run(["git", "push"], cwd=backup)
 
-    print("Backup complete!")
-    print("* Your liked songs were backed up")
-    print("* Your followed artists were backed up")
-    print("* Your saved albums were backed up")
-    print("* Names of people in blends were backed up (excluding large blends)")
-    print("* The following playlists were backed up:")
+        print("Backup complete!")
+        print("* Your liked songs were backed up")
+        print("* Your followed artists were backed up")
+        print("* Your saved albums were backed up")
+        print("* Names of people in blends were backed up (excluding large blends)")
+        print("* The following playlists were backed up:")
 
-    print(
-        tabulate(
-            [[x["name"], x["type"]] for x in playlists],
-            headers=["Name", "Type"],
-            showindex=range(1, len(playlists) + 1),
+        print(
+            tabulate(
+                [[x["name"], x["type"]] for x in playlists],
+                headers=["Name", "Type"],
+                showindex=range(1, len(playlists) + 1),
+            )
         )
-    )

src/backup.py

@@ -6,6 +6,7 @@
 from backup import backup
 from clean_library import clean_library
 from restore import restore
+from analyze import analyze
 
 client_id = "d93f79db5bbb41999a52734b9c95585a"
 redirect_uri = "http://localhost:3000/authed"
@@ -29,15 +30,25 @@
 )
 
 if not auth_manager.validate_token(auth_manager.get_cached_token()):
-    print("If your browser doesn't open automatically, open", auth_manager.get_authorize_url())
+    print(
+        "If your browser doesn't open automatically, open",
+        auth_manager.get_authorize_url(),
+    )
 
 sp = spotipy.Spotify(auth_manager=auth_manager)
 
 parser = argparse.ArgumentParser(description="Backup and restore spotify library")
-parser.add_argument('--backup', action='store_true')
+parser.add_argument("--backup", action="store_true")
+parser.add_argument("--playlist")
+parser.add_argument("--file")
 args = parser.parse_args()
 
-choice = "y" if args.backup else input(f"Logged in as {sp.me()['display_name']}. Continue? [y/n/logout] ")
+choice = (
+    "y"
+    if args.backup or args.file
+    else input(f"Logged in as {sp.me()['display_name']}. Continue? [y/n/logout] ")
+)
+
 if choice == "y":
     pass
 elif choice == "logout":
@@ -47,20 +58,30 @@
     quit()
 
 
-choice = "1" if args.backup else input(
-    """What would you like to do? Available:
+choice = (
+    "1"
+    if args.backup
+    else "5"
+    if args.file
+    else input(
+        """What would you like to do? Available:
 1. Backup
 2. Quick restore (doesn't preserve order of liked songs)
 3. Restore (preserves order of liked songs)
 4. [DANGEROUS] Clean library
+5. Analyze playlist
 
 Note that while quick restore loses order for liked songs, playlists are always ordered correctly.
 
 [1/2/3/4/q]: """
+    )
 )
 
 if choice == "1":
-    backup(sp)
+    if args.playlist:
+        backup(sp, playlist_name=args.playlist)
+    else:
+        backup(sp)
 elif choice == "2":
     restore(sp, True)
 elif choice == "3":
@@ -73,5 +94,11 @@
         clean_library(sp)
     else:
         quit()
+elif choice == "5":
+    if args.file:
+        analyze(args.file)
+    else:
+        file_path = input("Input file path: ")
+        analyze(file_path)
 else:
     quit()