Difyで飲食店紹介botを制作するに当たり飲食店のデータを抽出する方法を探していたところ、ホットペッパーグルメに連携できるリクルートAPIというものを見つけました。実際にpythonを使用してデータ抽出をしてみたのでその過程を記録します。
リクルートAPIの取得
リクルートWEBサービスから新規登録を行います。メールアドレスを登録して、認証を行うことでAPIを取得できます。1分くらいで結構簡単に取得することができました。
リクルートWEBサービス：https://webservice.recruit.co.jp/
Pythonを使用しデータ抽出
今回は、pythonを使用し、取得したデータをCSVファイルに保存しました。
簡単にまとめると以下のような手順で進めました。
『準備』と『コード記入』では以下のwebサイトを参考にしました。
https://zenn.dev/shintaro/articles/053fe2ca8b3430
コード追加：｢ジャンル｣｢予算｣｢営業時間｣を追加
参考にしたwebサイトでは店名と住所のデータのみ抽出しており、加えてジャンル・予算・営業時間も知りたかったので、webサイトのコードを参考にしてコードを追加しました。追加したコードは以下のようになりました。
shops = data['results'].get('shop', [])
for shop in shops:
writer.writerow({
'店名': shop['name'],
'住所': shop['address'],
'ジャンル': shop['genre']['name'],
'予算': shop['budget']['average'] if shop['budget'] else '不明'
'営業時間': shop['open']
})
※writer.writerowはデータをCSVファイルに書き込むためのメソッドです。
コード追加：愛知県内の全てのデータを取得できるようにする
参考にしたwebサイトは50件分のデータを抽出するコードだったので、chatGPTに相談して愛知県内の全てのデータを取得するコードを生成してもらいました。1回あたりに100件ずつデータの抽出を行い、それをループして、愛知県内の全てのデータ数を上回ったら終わるという流れです。実際に生成してもらったコードは以下のようになっています。
import requests
import time
API_KEY = "あなたのAPIキー" # APIキーをここに記入
BASE_URL = "https://webservice.recruit.co.jp/hotpepper/gourmet/v1/"
LARGE_AREA = "Z033" # 愛知県
COUNT = 100 # 一度に取得する件数（最大100件）
def fetch_all_stores():
start = 1 # データ取得の開始位置
all_stores = [] # すべての店舗情報を格納するリスト
while True:
# APIリクエストを送信
response = requests.get(BASE_URL, params={
"key": API_KEY,
"large_area": LARGE_AREA,
"count": COUNT,
"start": start,
"format": "json"
})
# レスポンスの確認
if response.status_code != 200:
print(f"リクエスト失敗: {response.status_code}")
break
# JSONデータを解析
data = response.json()
shops = data["results"].get("shop", [])
all_stores.extend(shops)
print(f"{start}件目から{start + len(shops) - 1}件目の店舗情報を取得しました。")
# 取得データが100件未満なら終了
if len(shops) < COUNT:
break
# 次の開始位置を更新
start += COUNT
# API制限を考慮して少し待機
time.sleep(1)
return all_stores
# 店舗情報を取得して表示
if __name__ == "__main__":
stores = fetch_all_stores()
print(f"取得した店舗数: {len(stores)}")
# 必要ならデータを保存
# 例: JSONファイルに保存
import json
with open("aichi_stores.json", "w", encoding="utf-8") as f:
json.dump(stores, f, ensure_ascii=False, indent=4)
print("データをaichi_stores.jsonに保存しました。")
コード作成
APIをリクエストするコード、CSVファイルに保存するコード、追加で作成したコードを結合させて最終的なコードを作成しました。
今後の課題
- chatGPTにつくってもらったコードをしっかり理解できていないので、解析して理解するとともにより簡単なコードにすることができないか考える
