Spaces:
Running
Running
| """Refresh Spotify popularity snapshots for existing Spotify tracks.""" | |
| import os | |
| import sys | |
| from datetime import datetime | |
| from dotenv import load_dotenv | |
| sys.path.insert(0, os.path.dirname(__file__)) | |
| from db import ( # noqa: E402 | |
| get_connection, | |
| generate_cycle_id, | |
| get_previous_snapshot, | |
| init_db, | |
| insert_snapshot, | |
| insert_viral_alert, | |
| ) | |
| from collect_spotify import ( # noqa: E402 | |
| SPOTIFY_VIRAL_POPULARITY_JUMP, | |
| normalize_release_date, | |
| ) | |
| from spotify_client import SpotifyClient # noqa: E402 | |
| load_dotenv(os.path.join(os.path.dirname(__file__), "..", ".env")) | |
| def enrich_spotify_data(): | |
| """Refresh Spotify track popularity and insert new snapshots.""" | |
| print(f"\n{'='*64}") | |
| print(f"Spotify Enrichment — {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") | |
| print(f"Viral jump threshold: +{SPOTIFY_VIRAL_POPULARITY_JUMP}") | |
| print(f"{'='*64}") | |
| init_db() | |
| conn = get_connection() | |
| client = SpotifyClient() | |
| cycle_id = generate_cycle_id("spotify") | |
| print(f"Cycle ID: {cycle_id}") | |
| songs = conn.execute( | |
| """ | |
| SELECT s.id, s.platform_id, s.artist_id, a.name AS artist_name | |
| FROM songs s | |
| JOIN artists a ON s.artist_id = a.id | |
| WHERE s.platform = 'spotify' | |
| AND s.platform_id IS NOT NULL | |
| ORDER BY a.name, s.title | |
| """ | |
| ).fetchall() | |
| if not songs: | |
| conn.close() | |
| print("\nNo Spotify songs found to enrich.\n") | |
| return {"songs_updated": 0, "viral_alerts": 0} | |
| total_updated = 0 | |
| total_viral = 0 | |
| for i in range(0, len(songs), 50): | |
| batch = songs[i : i + 50] | |
| track_ids = [row["platform_id"] for row in batch if row["platform_id"]] | |
| try: | |
| tracks = client.get_tracks(track_ids) | |
| except Exception as exc: | |
| print(f" ❌ failed batch {i // 50 + 1}: {exc}") | |
| continue | |
| tracks_by_id = {track.get("id"): track for track in tracks if track and track.get("id")} | |
| for row in batch: | |
| track = tracks_by_id.get(row["platform_id"]) | |
| if not track: | |
| continue | |
| popularity = int(track.get("popularity") or 0) | |
| prev = get_previous_snapshot(conn, row["id"], "spotify") | |
| insert_snapshot(conn, row["id"], popularity, 0, 0, "spotify", cycle_id) | |
| album = track.get("album") or {} | |
| images = album.get("images") or [] | |
| conn.execute( | |
| """ | |
| UPDATE songs | |
| SET title = ?, | |
| album_name = ?, | |
| release_date = ?, | |
| thumbnail_url = ? | |
| WHERE id = ? | |
| """, | |
| ( | |
| track.get("name", "Unknown"), | |
| album.get("name"), | |
| normalize_release_date(album.get("release_date", "")), | |
| images[0].get("url", "") if images else "", | |
| row["id"], | |
| ), | |
| ) | |
| total_updated += 1 | |
| if prev and prev.get("play_count") is not None: | |
| delta = popularity - int(prev["play_count"]) | |
| if delta >= SPOTIFY_VIRAL_POPULARITY_JUMP: | |
| insert_viral_alert( | |
| conn, | |
| { | |
| "song_id": row["id"], | |
| "previous_count": int(prev["play_count"]), | |
| "current_count": popularity, | |
| "growth_factor": float(delta), | |
| "platform": "spotify", | |
| }, | |
| ) | |
| total_viral += 1 | |
| conn.commit() | |
| print( | |
| f" ✅ batch {i // 50 + 1}: {len(batch)} tracks processed " | |
| f"(running total: {total_updated})" | |
| ) | |
| conn.close() | |
| print(f"\n{'='*64}") | |
| print("Spotify enrichment complete") | |
| print(f" Songs updated: {total_updated}") | |
| print(f" Viral alerts: {total_viral}") | |
| print(f"{'='*64}\n") | |
| return {"songs_updated": total_updated, "viral_alerts": total_viral} | |
| if __name__ == "__main__": | |
| try: | |
| enrich_spotify_data() | |
| except Exception as exc: | |
| print(f"Spotify enrichment failed: {exc}") | |
| sys.exit(1) | |