Update jihou.py

jihou.py

@@ -1,176 +1,115 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import os
 import time
 import requests
-import os
-import json
-from datetime import datetime, timedelta
-from zoneinfo import ZoneInfo
+from typing import List, Dict, Any
+
+# ===== 設定 =====
+X_ACCOUNT = os.getenv("dmsendertoken")
+if not X_ACCOUNT:
+    raise RuntimeError("環境変数 dmsendertoken が設定されていません")
 
-BASE_URL = "https://desk-api.channel.io/desk/channels/227312"
-TARGET_GROUP_CHAT_ID = "536135"
+BASE = "https://desk-api.channel.io/desk/channels/200605"
+GROUP_ID = 463667
+CHECK_PERSON_ID = "604730"
 
 HEADERS = {
+    "accept": "application/json",
     "accept-language": "ja",
-    "x-account": os.getenv("dmsendertoken")
+    "content-type": "application/json",
+    "x-account": X_ACCOUNT,
 }
 
-JST = ZoneInfo("Asia/Tokyo")
-
-# ---------------- 天気コード ----------------
-
-WEATHER_CODE_MAP = {
-    0: "☀️晴れ",
-    1: "☁️曇り",
-    2: "吹雪",
-    3: "砂嵐",
-    4: "🌫️霧",
-    5: "🌫️霧雨",
-    6: "🌧️雨",
-    7: "🌨️雪",
-    8: "🌦️シャワー",
-    9: "⛈️雷雨",
+GET_PARAMS = {
+    "sortOrder": "desc",
+    "limit": 34,
+    "logFolded": "false",
 }
 
+WELCOME_TEXT = "こんにちは。ITRSAにようこそ。自動でいくつかの部屋に招待しています。"
 
-def format_weather_code(code: int) -> str:
-    if code < 10:
-        return WEATHER_CODE_MAP.get(code, str(code))
-
-    main = code // 10
-    sub = code % 10
-
-    main_str = WEATHER_CODE_MAP.get(main)
-    sub_str = WEATHER_CODE_MAP.get(sub)
-
-    if main_str and sub_str:
-        return f"{main_str}、{sub_str}"
+INVITE_GROUPS = [519217, 536194, 534868]
 
-    return str(code)
 
-# ---------------- Channel 投稿 ----------------
-
-
-def post_group_message(chat_id, body):
-    url = f"{BASE_URL}/groups/{chat_id}/messages"
-    r = requests.post(url, headers=HEADERS, json=body)
+# ===== API =====
+def get_messages() -> List[Dict[str, Any]]:
+    url = f"{BASE}/groups/{GROUP_ID}/messages"
+    r = requests.get(url, headers=HEADERS, params=GET_PARAMS, timeout=20)
     r.raise_for_status()
-    return r.json()
-
-# ---------------- 時報 ----------------
-
+    return r.json().get("messages", [])
 
-def create_time_signal_body():
-    now = datetime.now(JST)
-    hour = now.hour
 
-    if hour < 12:
-        period = "午前"
-        display_hour = hour if hour != 0 else 12
-    else:
-        period = "午後"
-        display_hour = hour - 12 if hour > 12 else 12
-
-    message = f"<b>時報</b>\n{period}{display_hour}時をお知らせします。"
-
-    return {
-        "requestId": f"desk-web-time-signal-{int(time.time())}",
-        "blocks": [{"type": "text", "value": message}],
+def post_welcome_message() -> None:
+    url = f"{BASE}/groups/{GROUP_ID}/messages"
+    payload = {
+        "requestId": f"desk-web-{int(time.time() * 1000)}",
+        "blocks": [
+            {"type": "text", "value": WELCOME_TEXT}
+        ]
     }
-
-# ---------------- 天気予報 ----------------
+    requests.post(url, headers=HEADERS, json=payload, timeout=20).raise_for_status()
+    print("[INFO] 歓迎メッセージを送信しました")
 
 
-def load_todoufuken():
-    with open("todoufuken.json", encoding="utf-8") as f:
-        return json.load(f)
+def invite_person(group_id: int, person_id: str) -> None:
+    url = f"{BASE}/groups/{group_id}/invite"
+    params = {"managerIds": person_id}
+    requests.post(url, headers=HEADERS, params=params, timeout=20).raise_for_status()
+    print(f"[INFO] personId={person_id} を group {group_id} に招待しました")
 
 
-def fetch_weather(todoufuken):
-    lats = ",".join(str(p["lat"]) for p in todoufuken)
-    lons = ",".join(str(p["lon"]) for p in todoufuken)
+# ===== ロジック =====
+def process():
+    messages = get_messages()
 
-    url = (
-        "https://api.open-meteo.com/v1/forecast"
-        f"?latitude={lats}"
-        f"&longitude={lons}"
-        "&hourly=relative_humidity_2m,apparent_temperature,temperature_2m,"
-        "precipitation_probability,weather_code"
-        "&forecast_days=1"
+    # personId=599642 の最新発言時刻
+    latest_599642 = max(
+        (int(m.get("createdAt", 0))
+         for m in messages
+         if str(m.get("personId")) == CHECK_PERSON_ID),
+        default=0
     )
 
-    r = requests.get(url)
-    r.raise_for_status()
-    return r.json()
-
-
-def create_weather_body():
-    now = datetime.now(JST)
-    today_7am = now.replace(hour=7, minute=0, second=0, microsecond=0)
-
-    todoufuken = load_todoufuken()
-    data = fetch_weather(todoufuken)
+    # 条件を満たす join メッセージを抽出
+    join_targets = []
+    for m in messages:
+        log = m.get("log") or {}
+        if log.get("action") != "join":
+            continue
 
-    lines = ["<b>☀️ 本日の天気予報（7時以降・1時間ごと）</b>"]
+        created_at = int(m.get("createdAt", 0))
+        person_id = str(m.get("personId", ""))
 
-    for idx, pref in enumerate(todoufuken):
-        hourly = data[idx]["hourly"]
-        times = hourly["time"]
+        # join 以降に 599642 の発言が無い
+        if created_at > latest_599642:
+            join_targets.append(person_id)
 
-        lines.append(f"\n<b>{pref['name']}：</b>")
+    if not join_targets:
+        return
 
-        for i, t in enumerate(times):
-            t_jst = (
-                datetime.fromisoformat(t)
-                .replace(tzinfo=ZoneInfo("UTC"))
-                .astimezone(JST)
-            )
+    # 重複排除
+    join_targets = list(set(join_targets))
 
-            if t_jst < today_7am:
-                continue
+    # 歓迎メッセージは1回だけ
+    post_welcome_message()
 
-            wc = hourly["weather_code"][i]
-            at = hourly["apparent_temperature"][i]
-            temp = hourly["temperature_2m"][i]
-            pop = hourly["precipitation_probability"][i]
-            rh = hourly["relative_humidity_2m"][i]
+    # 全員を各グループへ招待
+    for pid in join_targets:
+        for gid in INVITE_GROUPS:
+            invite_person(gid, pid)
 
-            wc_str = format_weather_code(wc)
-
-            line = (
-                f"　{t_jst.strftime('%H時')}："
-                f"{wc_str} "
-                f"{at}℃ {temp}℃ {pop}% {rh}%"
-            )
-            lines.append(line)
-
-    return {
-        "requestId": f"desk-web-weather-{int(time.time())}",
-        "blocks": [{"type": "text", "value": "\n".join(lines)}],
-    }
 
-# ---------------- メインループ ----------------
-
-
-def main_loop():
+def main():
+    print("[INFO] Bot 起動（10秒間隔）")
     while True:
         try:
-            now = datetime.now(JST)
-
-            # 毎正時の時報
-            if now.minute == 0:
-                body = create_time_signal_body()
-                post_group_message(TARGET_GROUP_CHAT_ID, body)
-
-            # 午後2時9分に天気予報
-            if now.hour == 14 and now.minute == 20:
-                body = create_weather_body()
-                post_group_message(TARGET_GROUP_CHAT_ID, body)
-
-            time.sleep(60)
-
+            process()
         except Exception as e:
-            print("Error:", e)
-            time.sleep(10)
+            print(f"[ERROR] {e}")
+        time.sleep(10)
 
 
 if __name__ == "__main__":
-    main_loop()
+    main()