sbrandeis HF staff commited on
Commit
873436d
1 Parent(s): dad89e4

Make it work

Browse files
Files changed (1) hide show
  1. heatmap.py +3 -6
heatmap.py CHANGED
@@ -12,23 +12,20 @@ def fetch_recent_songs(client: Spotify):
12
  max_iterations = 30
13
  it = 0
14
  while it < max_iterations and cursor["cursors"] is not None:
15
- cursor = cursor["cursors"]["before"]
16
  recently_played.extend(cursor["items"])
17
 
18
  return recently_played
19
 
20
  def build_heatmap(recent_songs: List[dict]) -> np.ndarray:
21
  heatmap = np.zeros((7, 20))
22
- now = datetime.now()
23
 
24
  for track in recent_songs:
25
  played_at = parse(track["played_at"])
26
  weekday = datetime.weekday(played_at)
27
  week_offset = (now - played_at).days // 7
28
- print(weekday, week_offset)
29
-
30
- heatmap[weekday, -week_offset]
31
-
32
  return heatmap
33
 
34
 
 
12
  max_iterations = 30
13
  it = 0
14
  while it < max_iterations and cursor["cursors"] is not None:
15
+ cursor = client.current_user_recently_played(before=cursor["cursors"]["before"])
16
  recently_played.extend(cursor["items"])
17
 
18
  return recently_played
19
 
20
  def build_heatmap(recent_songs: List[dict]) -> np.ndarray:
21
  heatmap = np.zeros((7, 20))
22
+ now = datetime.now().astimezone()
23
 
24
  for track in recent_songs:
25
  played_at = parse(track["played_at"])
26
  weekday = datetime.weekday(played_at)
27
  week_offset = (now - played_at).days // 7
28
+ heatmap[weekday, -week_offset] +=1
 
 
 
29
  return heatmap
30
 
31