-
Notifications
You must be signed in to change notification settings - Fork 1
/
update.py
66 lines (52 loc) · 2.43 KB
/
update.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
"""Updates the category playlist cache."""
import argparse
import json
import tekore
from cached import CachedPlaylist, CachedPlaylistGroup
from categories import CATEGORIES
from utils import get_spotify_object
def update_cached_playlists(spotify):
user = spotify.current_user()
playlist_items = spotify.all_items(spotify.followed_playlists())
playlists_by_name = {item.name: item for item in playlist_items if item.owner.id == user.id}
missing_found = 0
for name, playlist_names in CATEGORIES.items():
group = CachedPlaylistGroup()
for playlist_name in playlist_names:
try:
playlist = playlists_by_name[playlist_name]
except KeyError:
if args.create_missing:
playlist = spotify.playlist_create(
user.id, playlist_name,
description="Automatically created by a script.")
print(f"\033[1;32mCreated playlist: {playlist_name}\033[0m")
else:
print(f"\033[0;33mWarning: no playlist called '{playlist_name}' found\033[0m")
missing_found += 1
continue
print(f"Updating cache for [{playlist.id}] {playlist.name}...")
obj = CachedPlaylist.from_tekore_playlist(playlist, spotify)
group.add_playlist(obj)
fp = open(name, 'w')
json.dump(group.serialize(), fp, indent=2)
fp.close()
if missing_found:
if missing_found == 1:
print("\033[1;33m1 playlist wasn't found.\n"
"Rerun this script with --create-missing to create it.\033[0m")
else:
print(f"\033[1;33m{missing_found} playlists weren't found.\n"
"Rerun this script with --create-missing to create them.\033[0m")
if __name__ == "__main__":
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--tekore-cfg", '-T', type=str, default='tekore.cfg',
help="file to use to store Tekore (Spotify) user token")
parser.add_argument("--create-missing", action="store_true", default=False,
help="create playlists that don't already exist")
args = parser.parse_args()
scope = tekore.scope.playlist_read_private
if args.create_missing:
scope += tekore.scope.playlist_modify_private
spotify = get_spotify_object(args.tekore_cfg, scope=scope)
update_cached_playlists(spotify)