Python で Chatwork の履歴を保存

import requests

API_TOKEN = 'YOUR_API_TOKEN'
ROOM_ID = '123'
MESSAGE = 'Hello, Chatwork!'

url = f'https://api.chatwork.com/v2/rooms/{ROOM_ID}/messages'
headers = {'X-ChatWorkToken': API_TOKEN}
params = {'body': MESSAGE}
response = requests.post(url, headers=headers, params=params)

print(response.json())

# ----------------------------------------------------------------------------
# Author: tomo
# Creation Date: 2023-04-02
#
# This script is free to use and modify at your own discretion.
# No warranty is provided for the script's functionality.
# Use this script at your own risk.
# ----------------------------------------------------------------------------
import requests
import json
import os
from datetime import datetime

# APIトークンを設定
api_token = "YOUR_API_TOKEN"

headers = {
    "X-ChatWorkToken": api_token
}

def get_rooms():
    url = "https://api.chatwork.com/v2/rooms"
    response = requests.get(url, headers=headers)
    return response.json()

def get_messages(room_id, last_message_id):
    url = f"https://api.chatwork.com/v2/rooms/{room_id}/messages"
    params = {"force": 1}

    if last_message_id:
        params["end_id"] = last_message_id

    response = requests.get(url, headers=headers, params=params)

    if response.status_code != 200:
        print(f"Error: HTTP status code {response.status_code} for room {room_id}")
        return []

    try:
        messages = response.json()
    except json.JSONDecodeError:
        print(f"Error decoding JSON for room {room_id}: {response.text}")
        return []

    return messages

def save_messages_to_file(room_name, messages):
    filename = f"{room_name}.txt"

    # ファイルが存在する場合は追記モード（'a'）、存在しない場合は新規作成モード（'w'）
    mode = 'a' if os.path.exists(filename) else 'w'

    with open(filename, mode, encoding="utf-8") as f:
        for message in messages:
            f.write(f"{message['account']['name']} ({message['account']['account_id']}): {message['body']}\n")

        # 最後のメッセージIDをファイルに保存
        if messages:
            f.write(f"LAST_MESSAGE_ID:{messages[-1]['message_id']}\n")

def get_last_message_id(filename):
    if not os.path.exists(filename):
        return None

    last_message_id = None

    with open(filename, 'r', encoding='utf-8') as f:
        for line in f:
            if line.startswith("LAST_MESSAGE_ID:"):
                last_message_id = line.strip().split(':')[1]

    return last_message_id

def main():
    rooms = get_rooms()
    for room in rooms:
        room_id = room["room_id"]
        room_name = room["name"]
        filename = f"{room_name}.txt"
        last_message_id = get_last_message_id(filename)
        messages = get_messages(room_id, last_message_id)
        
        # 最後のメッセージIDから新しいメッセージだけをフィルタリング
        if last_message_id:
            messages = [msg for msg in messages if int(msg["message_id"]) > int(last_message_id)]

        save_messages_to_file(room_name, messages)

if __name__ == "__main__":
    main()