Update ngwords.py

ngwords.py

@@ -4,7 +4,10 @@ import requests
 from datetime import datetime
 
 BASE_URL = "https://desk-api.channel.io/desk/channels/200605"
-SOURCE_GROUP_ID = "536194"
+
+# 監視対象グループID（配列で指定可能）
+SOURCE_GROUP_IDS = ["536194", "463667"]
+
 DEST_GROUP_ID = "536281"
 
 TOKEN = os.getenv("dmsendertoken")
@@ -14,7 +17,6 @@ HEADERS = {
     "Content-Type": "application/json"
 }
 
-GET_URL = f"{BASE_URL}/groups/{SOURCE_GROUP_ID}/messages"
 POST_URL = f"{BASE_URL}/groups/{DEST_GROUP_ID}/messages"
 
 
@@ -27,75 +29,81 @@ def contains_ng_word(text, ng_words):
     return any(word in text for word in ng_words)
 
 
+def process_group(source_group_id, ng_words):
+    get_url = f"{BASE_URL}/groups/{source_group_id}/messages"
+
+    resp = requests.get(
+        get_url,
+        headers=HEADERS,
+        params={
+            "sortOrder": "desc",
+            "limit": 34,
+            "logFolded": "false"
+        }
+    )
+    resp.raise_for_status()
+    data = resp.json()
+
+    messages = data.get("messages", [])
+
+    for msg in messages:
+        plain_text = msg.get("plainText", "")
+        if not plain_text:
+            continue
+
+        if not contains_ng_word(plain_text, ng_words):
+            continue
+
+        person_id = msg.get("personId")
+        created_at = msg.get("createdAt")
+        blocks = msg.get("blocks") or []
+        files = msg.get("files")
+
+        info_block = {
+            "type": "text",
+            "value": f"送信者：{person_id}、送信エポックミリ秒：{created_at}"
+        }
+
+        new_blocks = [info_block] + blocks
+
+        post_body = {
+            "requestId": f"desk-web-{int(time.time() * 1000)}",
+            "blocks": new_blocks,
+            "buttons": None,
+            "form": None,
+            "webPage": None,
+            "files": files if files else None,
+            "customPayload": None
+        }
+
+        # POST
+        post_resp = requests.post(
+            POST_URL,
+            headers=HEADERS,
+            json=post_body
+        )
+        post_resp.raise_for_status()
+
+        # DELETE 元メッセージ
+        msg_id = msg.get("id")
+        delete_url = f"{BASE_URL}/groups/{source_group_id}/messages/{msg_id}"
+
+        del_resp = requests.delete(
+            delete_url,
+            headers=HEADERS
+        )
+        del_resp.raise_for_status()
+
+        print(f"処理完了: group_id={source_group_id}, message_id={msg_id}")
+
+
 def main():
     ng_words = load_ng_words()
 
     while True:
         try:
-            resp = requests.get(
-                GET_URL,
-                headers=HEADERS,
-                params={
-                    "sortOrder": "desc",
-                    "limit": 34,
-                    "logFolded": "false"
-                }
-            )
-            resp.raise_for_status()
-            data = resp.json()
-
-            messages = data.get("messages", [])
-
-            for i, msg in enumerate(messages):
-                plain_text = msg.get("plainText", "")
-                if not plain_text:
-                    continue
-
-                if not contains_ng_word(plain_text, ng_words):
-                    continue
-
-                person_id = msg.get("personId")
-                created_at = msg.get("createdAt")
-                blocks = msg.get("blocks") or []
-                files = msg.get("files")
-
-                # 先頭に送信者情報ブロックを追加
-                info_block = {
-                    "type": "text",
-                    "value": f"送信者：{person_id}、送信エポックミリ秒：{created_at}"
-                }
-
-                new_blocks = [info_block] + blocks
-
-                post_body = {
-                    "requestId": f"desk-web-{int(time.time() * 1000)}",
-                    "blocks": new_blocks,
-                    "buttons": None,
-                    "form": None,
-                    "webPage": None,
-                    "files": files if files else None,
-                    "customPayload": None
-                }
-
-                # POST
-                post_resp = requests.post(
-                    POST_URL,
-                    headers=HEADERS,
-                    json=post_body
-                )
-                post_resp.raise_for_status()
-
-                # DELETE 元メッセージ
-                msg_id = msg.get("id")
-                delete_url = f"{BASE_URL}/groups/{SOURCE_GROUP_ID}/messages/{msg_id}"
-
-                del_resp = requests.delete(
-                    delete_url,
-                    headers=HEADERS
-                )
-                del_resp.raise_for_status()
-
-                print(f"処理完了: message_id={msg_id}")
+            for group_id in SOURCE_GROUP_IDS:
+                process_group(group_id, ng_words)
 
         except Exception as e:
             print("エラー:", e)