Spotify の 「バイラルトップ 50 - 日本」 のプレイリストを作成するスクリプトを書いた

SPOTIPY_CLIENT_ID=xxxx
SPOTIPY_CLIENT_SECRET=xxxx
SPOTIPY_REDIRECT_URI=http://127.0.0.1:8888/callback
SPOTIFY_REFRESH_TOKEN=xxxx
SPOTIFY_PLAYLIST_ID=プレイリストID
STATE_JSON_B64=base64化したstate.json

% python3 save_session.py
Log in manually and wait for the 'Download' button to appear.
Press Enter after login is complete...
Saved session to state.json

Copy the following and add it to your secrets as STATE_JSON_B64:

eyJ.....

from playwright.sync_api import sync_playwright
import base64

def save_login_state_and_encode():
    with sync_playwright() as p:
        browser = p.chromium.launch(headless=False)
        context = browser.new_context()
        page = context.new_page()
        page.goto("https://charts.spotify.com")

        print("Log in manually and wait for the 'Download' button to appear.")
        input("Press Enter after login is complete...")

        context.storage_state(path="state.json")
        browser.close()
        print("Saved session to state.json")

        with open("state.json", "rb") as f:
            encoded = base64.b64encode(f.read()).decode("utf-8")
            print("\nCopy the following and add it to your secrets as STATE_JSON_B64:\n")
            print(encoded)

if __name__ == "__main__":
    save_login_state_and_encode()

% python get_refresh_token.py
Open this URL to authorize: https://accounts.spotify.com/authorize?client_id=*****&response_type=code&redirect_uri=http%3A%2F%2F127.0.0.1%3A8888%2Fcallback&scope=playlist-modify-public+playlist-modify-private
Paste the redirected URL here: http://127.0.0.1:8888/callback?code=*****
/get_refresh_token.py:20: DeprecationWarning: You're using 'as_dict = True'.get_access_token will return the token string directly in future versions. Please adjust your code accordingly, or use get_cached_token instead.
  token_info = sp_oauth.get_access_token(code, as_dict=True)
Your refresh_token is: *****

import os
from spotipy.oauth2 import SpotifyOAuth
from dotenv import load_dotenv

load_dotenv()

sp_oauth = SpotifyOAuth(
    client_id=os.getenv("SPOTIPY_CLIENT_ID"),
    client_secret=os.getenv("SPOTIPY_CLIENT_SECRET"),
    redirect_uri=os.getenv("SPOTIPY_REDIRECT_URI"),
    scope="playlist-modify-public playlist-modify-private"
)

auth_url = sp_oauth.get_authorize_url()
print("Open this URL to authorize:", auth_url)

response = input("Paste the redirected URL here: ")

code = sp_oauth.parse_response_code(response)
token_info = sp_oauth.get_access_token(code)
print("Your refresh_token is:", token_info["refresh_token"])

% python3 update_playlist.py
Starting script...
state.json extracted.
Downloading CSV from Spotify Charts...
CSV downloaded: viral.csv
Authenticating with Spotify API...
playlist_id: *****
Fetching current playlist tracks...
Removed 100 track(s) from the playlist.
Number of URIs to add: 100
Added 100 track(s) to the playlist.

import csv
import os
import base64
from dotenv import load_dotenv
from spotipy import Spotify
from spotipy.oauth2 import SpotifyOAuth
from playwright.sync_api import sync_playwright

print("Starting script...")

load_dotenv()

def decode_state_json():
    encoded = os.getenv("STATE_JSON_B64")
    if not encoded:
        print("STATE_JSON_B64 is not defined.")
        return False
    decoded = base64.b64decode(encoded).decode("utf-8")
    with open("state.json", "w", encoding="utf-8") as f:
        f.write(decoded)
    print("state.json extracted.")
    return True

def download_spotify_csv():
    print("Downloading CSV from Spotify Charts...")
    with sync_playwright() as p:
        browser = p.chromium.launch(headless=True)
        context = browser.new_context(
            storage_state="state.json",
            accept_downloads=True,
            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
            locale="ja-JP",
            timezone_id="Asia/Tokyo",
            geolocation={"longitude": 139.6917, "latitude": 35.6895},
            permissions=["geolocation"]
        )
        page = context.new_page()
        page.add_init_script("Object.defineProperty(navigator, 'webdriver', {get: () => undefined})")

        try:
            page.goto("https://charts.spotify.com/charts/view/viral-jp-daily/latest", timeout=30000)
            page.wait_for_load_state("domcontentloaded")
            page.evaluate("document.getElementById('onetrust-banner-sdk')?.remove()")
            page.locator('button[data-encore-id="buttonTertiary"]').first.wait_for(timeout=15000)
            with page.expect_download(timeout=15000) as download_info:
                page.locator('button[data-encore-id="buttonTertiary"]').first.click()
            download = download_info.value
            download.save_as("viral.csv")
            print("CSV downloaded: viral.csv")
        except Exception as e:
            print("CSV download failed:", e)
            try:
                page.screenshot(path="debug.png", full_page=True)
            except:
                pass
        finally:
            browser.close()

def get_spotify_client():
    refresh_token = os.getenv("SPOTIFY_REFRESH_TOKEN")
    if not refresh_token:
        print("SPOTIFY_REFRESH_TOKEN is not set.")
        exit(1)

    auth_manager = SpotifyOAuth(
        client_id=os.getenv("SPOTIPY_CLIENT_ID"),
        client_secret=os.getenv("SPOTIPY_CLIENT_SECRET"),
        redirect_uri=os.getenv("SPOTIPY_REDIRECT_URI"),
    )

    try:
        token_info = auth_manager.refresh_access_token(refresh_token)
        return Spotify(auth=token_info["access_token"])
    except Exception as e:
        print("Failed to get access token:", e)
        exit(1)

def update_playlist():
    print("Authenticating with Spotify API...")
    sp = get_spotify_client()

    playlist_id = os.getenv("SPOTIFY_PLAYLIST_ID")
    print("playlist_id:", playlist_id)

    if not os.path.exists("viral.csv"):
        print("viral.csv not found.")
        return

    print("Fetching current playlist tracks...")
    results = sp.playlist_items(playlist_id)
    track_uris = [item["track"]["uri"] for item in results["items"] if item["track"]]
    if track_uris:
        sp.playlist_remove_all_occurrences_of_items(playlist_id, track_uris)
        print(f"Removed {len(track_uris)} track(s) from the playlist.")

    with open("viral.csv", newline="", encoding="utf-8") as csvfile:
        reader = csv.DictReader(csvfile)
        uris = [row["uri"] for row in reader if row["uri"].startswith("spotify:track:")]
    print("Number of URIs to add:", len(uris))
    if uris:
        sp.playlist_add_items(playlist_id, uris)
        print(f"Added {len(uris)} track(s) to the playlist.")

if __name__ == "__main__":
    if decode_state_json():
        download_spotify_csv()
        update_playlist()

FROM mcr.microsoft.com/playwright/python:v1.42.0-jammy

WORKDIR /app

COPY . /app

RUN pip install --no-cache-dir -r requirements.txt
RUN playwright install --with-deps chromium

CMD ["python", "update_playlist.py"]

name: Run Spotify Updater (via Docker)

on:
  schedule:
    - cron: '0 0 * * *'  # 毎日9:00 JST
  workflow_dispatch:

jobs:
  run-updater:
    runs-on: ubuntu-latest

    steps:
      - uses: actions/checkout@v4

      - name: Build Docker image
        run: docker build -t spotify-updater .

      - name: Run updater container
        run: |
          echo "$STATE_JSON_B64" | base64 -d > state.json
          docker run --rm \
            -e SPOTIPY_CLIENT_ID=${{ secrets.SPOTIPY_CLIENT_ID }} \
            -e SPOTIPY_CLIENT_SECRET=${{ secrets.SPOTIPY_CLIENT_SECRET }} \
            -e SPOTIPY_REDIRECT_URI=${{ secrets.SPOTIPY_REDIRECT_URI }} \
            -e SPOTIFY_REFRESH_TOKEN=${{ secrets.SPOTIFY_REFRESH_TOKEN }} \
            -e SPOTIFY_PLAYLIST_ID=${{ secrets.SPOTIFY_PLAYLIST_ID }} \
            -e STATE_JSON_B64="${{ secrets.STATE_JSON_B64 }}" \
            spotify-updater