Spotipy教程：如何通过艺人ID精确获取并添加指定艺人歌曲

import spotipy
from spotipy.oauth2 import SpotifyClientCredentials
import collections

# 假设您已经设置了Spotify客户端凭据
# client_id = 'YOUR_CLIENT_ID'
# client_secret = 'YOUR_CLIENT_SECRET'
# sp = spotipy.Spotify(auth_manager=SpotifyClientCredentials(client_id=client_id,
#                                                           client_secret=client_secret))

# 假设sp是已经认证的Spotipy客户端实例
sp = spotipy.Spotify(client_credentials_manager=SpotifyClientCredentials()) # 仅用于示例，实际应用请使用您的凭据

def get_artist_id(artist_name):
    """
    通过艺人名称搜索并返回艺人ID。
    """
    try:
        response = sp.search(q=artist_name, type="artist", limit=1)
        artists = response['artists']['items']
        if artists:
            artist = artists[0]
            print(f"找到艺人: {artist['name']} (ID: {artist['id']})")
            return artist['id']
        else:
            print(f"未找到艺人: {artist_name}")
            return None
    except spotipy.exceptions.SpotifyException as e:
        print(f"搜索艺人时发生错误: {e}")
        return None
    except IndexError:
        print(f"未能在搜索结果中找到艺人: {artist_name}")
        return None

# 示例：获取Adele的艺人ID
adele_artist_id = get_artist_id("Adele")
if adele_artist_id:
    print(f"Adele的艺人ID是: {adele_artist_id}")
else:
    print("无法获取Adele的艺人ID。")

def get_artist_top_tracks_uris(artist_id, country='US'):
    """
    获取指定艺人在特定国家/地区的前10首热门歌曲的URI。
    """
    songs_uris = []
    try:
        response = sp.artist_top_tracks(artist_id, country)
        tracks = response['tracks']
        if tracks:
            for track in tracks:
                songs_uris.append(track['uri'])
                print(f"添加热门歌曲: {track['name']} - {track['artists'][0]['name']}")
        else:
            print(f"未找到艺人ID {artist_id} 在国家 {country} 的热门歌曲。")
    except spotipy.exceptions.SpotifyException as e:
        print(f"获取艺人热门歌曲时发生错误: {e}")
    return songs_uris

# 示例：获取Adele在美国的前10首热门歌曲
if adele_artist_id:
    adele_top_songs_uris = get_artist_top_tracks_uris(adele_artist_id, country='US')
    print(f"\nAdele的前10首热门歌曲URI: {adele_top_songs_uris}")

def get_filtered_artist_tracks(artist_id, artist_name, max_results=50):
    """
    通过艺人ID过滤通用搜索结果，获取指定艺人的热门歌曲URI。
    """
    songs_uris = []
    artist_tracks_data = {} # 存储 {track_name: {'uri': uri, 'popularity': popularity}}

    offset = 0
    limit = 50 # 每次API请求获取50条结果
    search_count = 0 # 追踪已处理的搜索结果数量

    print(f"\n正在为艺人 '{artist_name}' (ID: {artist_id}) 搜索并过滤歌曲...")

    while search_count < max_results * 2: # 搜索结果数量的两倍作为上限，防止无限循环
        try:
            # 执行宽泛的歌曲搜索，指定艺人名称
            response = sp.search(q=f"artist:{artist_name}", type="track", 
                                 limit=limit, offset=offset)
        except spotipy.exceptions.SpotifyException as e:
            print(f"搜索歌曲时发生错误: {e}")
            break

        tracks_items = response['tracks']['items']
        if not tracks_items:
            break # 没有更多结果

        for track in tracks_items:
            # 关键过滤步骤：检查歌曲的艺人列表中是否包含目标艺人ID
            is_target_artist = any(artist_id == artist['id'] for artist in track['artists'])

            if is_target_artist:
                track_name = track['name']
                track_uri = track['uri']
                track_popularity = track['popularity']

                # 如果歌曲名称已存在，则保留人气更高的版本
                if track_name not in artist_tracks_data or \
                   track_popularity > artist_tracks_data[track_name]['popularity']:
                    artist_tracks_data[track_name] = {
                        'uri': track_uri,
                        'popularity': track_popularity,
                        'name': track_name # 方便后续排序打印
                    }

            search_count += 1
            if len(artist_tracks_data) >= max_results and search_count >= max_results:
                # 已经收集到足够多的唯一歌曲，且处理了一定数量的搜索结果，可以提前结束
                break

        offset += limit # 准备获取下一页结果
        if len(artist_tracks_data) >= max_results and search_count >= max_results:
            break

    # 根据人气值降序排序，然后按名称升序排序（次要排序）
    sorted_tracks = sorted(
        artist_tracks_data.values(),
        key=lambda x: (-x['popularity'], x['name'])
    )

    # 提取URI，并限制到所需的数量
    for track_info in sorted_tracks[:max_results]:
        songs_uris.append(track_info['uri'])
        print(f"添加过滤歌曲: {track_info['name']} (人气: {track_info['popularity']})")

    return songs_uris

# 示例：获取Adele的最多20首热门歌曲（通过过滤）
if adele_artist_id:
    adele_filtered_songs_uris = get_filtered_artist_tracks(adele_artist_id, "Adele", max_results=20)
    print(f"\nAdele的过滤后歌曲URI ({len(adele_filtered_songs_uris)}首): {adele_filtered_songs_uris}")