【Python】YouTube APIで動画データを取得する方法

import argparse
from apiclient.discovery import build
import pandas as pd


YOUTUBE_API_KEY = '取得したAPIキー'
YOUTUBE_API_SERVICE_NAME = "youtube"
YOUTUBE_API_VERSION = "v3"


youtube = build(YOUTUBE_API_SERVICE_NAME, YOUTUBE_API_VERSION, developerKey=YOUTUBE_API_KEY)


# 動画の再生回数やいいね数、お気に入り数などを取得する処理
def get_statistics(id):
    statistics = youtube.videos().list(part='statistics', id=id).execute()['items'][0]['statistics']
    return statistics


# 動画データ一覧を取得する処理
def get_video_info(part, q, order, type, num):
    search_list = []
    for i in range(num):
      if i == 0:
        search_query = youtube.search().list(
          part=part,
          q=q,
          order=order,
          type=type,
          maxResults=10,# maxResults→5~50
          )
      else:
        search_query = youtube.search().list_next(
          search_query,
          api_response,
        )

      api_response = search_query.execute()

      search_response = api_response.get("items", [])
      for v in search_response:
        id = v["id"]["videoId"]
        title = v["snippet"]["title"]
        description = v["snippet"]["description"]
        channel_title = v["snippet"]["channelTitle"]
        statistics = get_statistics(id)
        if "viewCount" in statistics:
          view_count = statistics["viewCount"]
        else:
          view_count = 0
        if "likeCount" in statistics:
          like_count = statistics["likeCount"]
        else:
          like_count = 0
        if "favoriteCount" in statistics:
          favorite_count = statistics["favoriteCount"]
        else:
          favorite_count = 0
        if "commentCount" in statistics:
          comment_count = statistics["commentCount"]
        else:
          comment_count = 0
        url = 'https://www.youtube.com/watch?v=%s' % id
        published_at = v["snippet"]["publishedAt"]
        result = [id, title, description, channel_title, view_count, like_count, favorite_count, comment_count, url, published_at]
        search_list.append(result)
    return search_list



def main(args):
  data = get_video_info(part='snippet', q=args.keyword, order='viewCount', type='video', num=args.num)
  df = pd.DataFrame(data, columns=['ID', 'タイトル', '説明', 'チャンネル名', '再生回数', 'いいね数', 'お気に入り数', 'コメント数', 'URL', '公開日時'])
  df.to_csv('video_list.csv')


if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        add_help=True
    )
    subparsers = parser.add_subparsers()

    parser_convert = subparsers.add_parser('get_video_list', help='get video list')
    parser_convert.add_argument('--keyword', '-k', required=True)
    parser_convert.add_argument('--num', '-n', required=True, type=int)
    parser_convert.set_defaults(handler=main)

    args = parser.parse_args()

    if hasattr(args, 'handler'):
        args.handler(args)
    else:
        parser.print_help()