Update dmsender.py

dmsender.py

@@ -3,6 +3,9 @@ import requests
 import os
 
 BASE_URL = "https://desk-api.channel.io/desk/channels/200605"
+MEDIA_BASE = "https://media.channel.io/cht/v1"
+
+CHANNEL_ID = "200605"
 TARGET_FORWARD_CHAT_ID = "463667"
 SKIP_PERSON_ID = 604730
 
@@ -51,12 +54,14 @@ def post_message(chat_id, body):
     r.raise_for_status()
     return r.json()
 
+
 def post_message2(chat_id, body):
     url = f"{BASE_URL}/groups/{chat_id}/messages"
     r = requests.post(url, headers=HEADERS, json=body)
     r.raise_for_status()
     return r.json()
-    
+
+
 def get_messages(chat_id):
     url = (
         f"{BASE_URL}/direct-chats/{chat_id}/messages"
@@ -74,12 +79,35 @@ def mark_messages_read(chat_id):
 
 
 def find_latest_message(messages):
-    """
-    messages 配列の中から updatedAt が最大の message を返す
-    """
     return max(messages, key=lambda m: m.get("updatedAt", 0))
 
 
+# ===== ファイル関連 =====
+def download_file(group_chat_id, file_key):
+    url = (
+        f"{MEDIA_BASE}/desk/channels/{CHANNEL_ID}"
+        f"/groups/{group_chat_id}/messages/file"
+    )
+    params = {"key": file_key}
+    r = requests.get(url, headers=HEADERS, params=params)
+    r.raise_for_status()
+    return r.content  # バイナリ
+
+
+def upload_file_to_target(file_name, file_content):
+    url = (
+        f"{MEDIA_BASE}/pri-file/{CHANNEL_ID}"
+        f"/groups/{TARGET_FORWARD_CHAT_ID}/message/{file_name}"
+    )
+
+    headers = HEADERS.copy()
+    headers.pop("Content-Type", None)
+
+    r = requests.post(url, headers=headers, data=file_content)
+    r.raise_for_status()
+    return r.json()  # files にそのまま使う
+
+
 def main_loop():
     while True:
         try:
@@ -95,41 +123,55 @@ def main_loop():
                 if not chat_id:
                     continue
 
-                # messages を取得
                 msg_data = get_messages(chat_id)
                 messages = msg_data.get("messages", [])
                 if not messages:
                     continue
 
-                # 最新メッセージ取得
                 latest = find_latest_message(messages)
 
-                # personId が 604730 の場合はスキップ
                 person_id = latest.get("personId")
                 if person_id == SKIP_PERSON_ID:
                     mark_messages_read(chat_id)
                     continue
 
-                # ① unread があるチャットに「送信しました」を送信
+                # ① 送信しました
                 post_message(chat_id, SEND_BODY_TEMPLATE)
 
                 blocks = latest.get("blocks")
                 files = latest.get("files")
+                new_files = []
+
+                # ===== ファイル転送処理 =====
+                if files:
+                    source_group_id = latest.get("chatId")
+
+                    for f in files:
+                        file_key = f.get("key")
+                        file_name = f.get("name")
+
+                        if not file_key or not file_name:
+                            continue
+
+                        file_content = download_file(source_group_id, file_key)
+                        uploaded_file_json = upload_file_to_target(
+                            file_name, file_content
+                        )
+                        new_files.append(uploaded_file_json)
 
-                # ② 固定チャットIDへ転送
+                # ===== 固定IDへ転送 =====
                 forward_body = FORWARD_BODY_TEMPLATE.copy()
                 forward_body["blocks"] = blocks
-                forward_body["files"] = files if files else None
+                forward_body["files"] = new_files if new_files else None
 
                 post_message2(TARGET_FORWARD_CHAT_ID, forward_body)
 
-                # ③ 元チャットを既読にする
+                # 既読
                 mark_messages_read(chat_id)
 
         except Exception as e:
             print("Error:", e)
 
-        # 5秒待機
         time.sleep(5)