Update watcher.py

watcher.py

@@ -63,6 +63,51 @@ def extract_youtube_id(text):
             return m.group(1)
     return None
 
+def extract_youtube_and_option(text):
+    """
+    YouTube URL と画質指定オプションを抽出する
+    戻り値: (video_id, option) 例: ("abcdEFGHijk", "720")
+    """
+    # 全角スペースも対象に
+    parts = re.split(r"[ \u3000]+", text.strip())
+
+    video_id = extract_youtube_id(parts[0])
+    option = None
+    if len(parts) > 1:
+        option = parts[1].lower()
+
+    return video_id, option
+
+def normalize_quality(option):
+    """
+    ★ ユーザー入力を内部品質コードへ変換
+    """
+    if option is None:
+        return "720"
+
+    quality_map = {
+        "144": "144",
+        "240": "240",
+        "360": "360",
+        "480": "480",
+        "720": "720",
+        "1080": "1080",
+        "2k": "1440",
+        "1440": "1440",
+        "4k": "2160",
+        "2160": "2160",
+        "8k": "4320",
+        "4320": "4320",
+    }
+
+    if option in quality_map:
+        return quality_map[option]
+
+    if option in ["a", "f"]:
+        return option  # 特殊指定
+
+    return "720"  # デフォルト
+
 def get_video_title(youtube_url):
     """YouTube動画のタイトルを取得する簡易的な関数"""
     try:
@@ -138,6 +183,39 @@ def select_best_video_and_audio(items):
     
     return best_video, best_audio
 
+def select_quality_video(items, quality):
+    """
+    ★ 指定画質に最適な動画を選択（音声なし映像のみ）
+    """
+    video_candidates = []
+    for item in items:
+        if item["has_audio"] == "false":
+            m = re.match(r"(\d+)p", item["quality"])
+            if m:
+                q = int(m.group(1))
+                video_candidates.append((q, item))
+
+    if not video_candidates:
+        return None
+
+    # 最大画質
+    video_candidates.sort(key=lambda x: x[0])
+
+    if quality == "f":  # 最大画質
+        return video_candidates[-1][1]
+
+    req = int(quality)
+    best = None
+
+    # 指定画質以上で一番近いもの
+    for q, item in video_candidates:
+        if q >= req:
+            best = item
+            break
+
+    # 無ければ最大画質
+    return best if best else video_candidates[-1][1]
+
 # ===== ssyoutube サーバー側結合 =====
 def merge_video_on_server(video_url, audio_url, video_id, quality, video_title, nonce):
     """
@@ -295,39 +373,29 @@ def main():
                 time.sleep(10)
                 continue
 
-            youtube_id = extract_youtube_id(latest_msg["plainText"])
-            if not youtube_id:
+            video_id, option = extract_youtube_and_option(latest_msg["plainText"])
+            if not video_id:
                 time.sleep(10)
                 continue
 
-            youtube_url = f"https://www.youtube.com/watch?v={youtube_id}"
-            send_to_channel(f"{youtube_id}のダウンロードを開始します。")
-            # 動画タイトルを取得
+            quality_opt = normalize_quality(option)
+
+            youtube_url = f"https://www.youtube.com/watch?v={video_id}"
+            send_to_channel(f"{video_id} のダウンロードを開始します。")
+
             video_title = get_video_title(youtube_url)
-            
             items, nonce = fetch_download_links(youtube_url)
-            
-            if not nonce:
-                send_to_channel("nonceの取得に失敗しました。Izuemon(MOCA)を呼んでください。")
-                time.sleep(10)
-                continue
 
-            # すべてのダウンロードリンクを収集
-            video_items = []
-            audio_items = []
-            all_items = []
-            
-            for item in items:
-                all_items.append(item)
-                if item["quality"] == "audio":
-                    audio_items.append(item)
-                elif item["has_audio"] == "false":
-                    video_items.append(item)
-            
-            # 最高画質の動画と音声を選択
-            best_video, best_audio = select_best_video_and_audio(items)
-            
-            # メッセージを構築
+            all_items = items[:]
+            audio_items = [i for i in items if i["quality"] == "audio"]
+
+            selected_video = None
+            selected_audio = audio_items[0] if audio_items else None
+
+            if quality_opt not in ["a"]:
+                selected_video = select_quality_video(items, quality_opt)
+
+            # ダウンロードリスト送信
             message_lines = []
             message_lines.append(f"<b>{video_title}</b>\n")
             message_lines.append("<b>ダウンロードリンク</b>")
@@ -347,40 +415,30 @@ def main():
             message_lines.append("自動で音声と動画を結合しています。\n★動画ファイルと音声ファイルをダウンロードして<link type=\"url\" value=\"https://qooly.com/ja/merge-video-and-audio-online\">ここ</link>などで結合すると早くて確実です。")
             send_to_channel("\n".join(message_lines))
             
-            message_lines = []
-            # 最高画質の結合動画を作成
-            if best_video and best_audio and nonce:
-                message_lines.append("\n <b>最高画質結合動画</b>")
-                
-                # 画質情報を取得
-                quality_match = re.match(r"(\d+)p", best_video["quality"])
-                if quality_match:
-                    quality = quality_match.group(1)
-                    
-                    # サーバー側で結合
-                    print(f"サーバー側での結合を開始します: {youtube_id} {quality}p")
+            # ★ 結合フラグ
+            do_merge = quality_opt not in ["a"]
+
+            if do_merge and selected_video and selected_audio:
+                m = re.match(r"(\d+)p", selected_video["quality"])
+                if m:
+                    q = m.group(1)
                     try:
                         merged_url = merge_video_on_server(
-                            best_video["url"],
-                            best_audio["url"],
-                            youtube_id,
-                            quality,
+                            selected_video["url"],
+                            selected_audio["url"],
+                            video_id,
+                            q,
                             video_title,
                             nonce
                         )
-                        message_lines.append(f"<link type=\"url\" value=\"{merged_url}\">{quality}p 結合済み動画</link>")
+                        send_to_channel(f"<link type=\"url\" value=\"{merged_url}\">{q}p 結合済み動画</link>")
                     except Exception as e:
-                        print(f"結合処理エラー: {e}")
-                        message_lines.append(f"結合処理に失敗: {str(e)}")
-            message_lines.append(f"ダウンロードが完了しました。\nダウンロードに失敗してる場合は<link type=\"manager\" value=\"596801\">@⃠ᗰOᑕᗩっち🍫</link>を呼んでください。")
-            # メッセージを送信
-            message = "\n".join(message_lines)
-            send_to_channel(message)
-            
-            # 処理済みとしてマーク
-            processed_messages.add(latest_msg.get("id"))
-            
-            print(f"送信完了: {youtube_id}")
+                        send_to_channel(f"結合失敗: {e}")
+
+            send_to_channel("完了しました！")
+            processed_messages.add(latest_msg["id"])
+
+            print(f"送信完了: {video_id}")
 
         except Exception as e:
             print("エラー:", e)