Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -6,45 +6,29 @@ from io import BytesIO
 import scratchcommunication
 import traceback
 
-# --- Scratchクラウド接続 ---
+# --- Turbowarpクラウド接続 ---
 PROJECT_ID = 1293416663
 tw = scratchcommunication.TwCloudConnection(
     project_id=PROJECT_ID,
     username="server",
-    contact_info="contact"
+    cloud_host="wss://clouddata.turbowarp.org/",
+    accept_strs=False,  # 数値を扱いやすくする
+    reconnect=True
 )
 
-def get_var(name):
-    try:
-        return tw.get_variable(name=name, name_literal=False)
-    except scratchcommunication.exceptions.NotSupported as e:
-        print(f"[WARN] NotSupported error while getting '{name}': {e}")
-        return None
-    except KeyError as e:
-        print(f"[ERROR] KeyError while getting '{name}': {e}")
-        return None
-    except Exception as e:
-        print(f"[ERROR] Unexpected error while getting '{name}': {e}")
-        traceback.print_exc()
-        return None
-
-def set_var(name, value):
-    try:
-        return tw.set_variable(name=name, value=value, name_literal=False)
-    except Exception as e:
-        print(f"[ERROR] Failed to set '{name}' to '{value}': {e}")
-        traceback.print_exc()
-        return None
+# Quickaccess有効化（辞書のように扱える）
+tw.enable_quickaccess()
 
 # --- n-chars.txt の読み込み ---
 try:
     with open("turbowarp-server/n-chars.txt", "r", encoding="utf-8") as f:
         n_chars = [line.strip() for line in f]
 except Exception as e:
-    print(f"[FATAL] Failed to load n-chars.txt: {e}")
+    print(f"[FATAL] n-chars.txt 読み込み失敗: {e}")
     traceback.print_exc()
     n_chars = []
 
+# --- ユーティリティ関数 ---
 def decode_prompt(encoded_str):
     chars = []
     for i in range(0, len(encoded_str), 2):
@@ -53,7 +37,7 @@ def decode_prompt(encoded_str):
             if idx < len(n_chars):
                 chars.append(n_chars[idx])
         except Exception as e:
-            print(f"[WARN] Failed to decode chunk '{encoded_str[i:i+2]}': {e}")
+            print(f"[WARN] decode失敗: {encoded_str[i:i+2]} -> {e}")
     return "".join(chars)
 
 def rgb_to_scratch_number(rgb):
@@ -81,7 +65,7 @@ def generate_image(prompt):
         resp.raise_for_status()
         return Image.open(BytesIO(resp.content))
     except Exception as e:
-        print(f"[ERROR] Image generation failed for prompt '{prompt}': {e}")
+        print(f"[ERROR] 画像生成失敗: '{prompt}' -> {e}")
         traceback.print_exc()
         return None
 
@@ -97,39 +81,45 @@ def split_packets(data, max_length=9998):
     packets = []
     idx = 0
     while idx < len(data):
-        chunk = data[idx:idx + max_length]
-        packets.append(chunk)
+        packets.append(data[idx:idx + max_length])
         idx += max_length
     return packets
 
 # --- メインループ ---
 while True:
     try:
-        n1 = get_var("n1")
-        if n1 and len(n1) >= 3 and n1[0] == "0":
-            if get_var("n0") == "0":
-                set_var("n0", "1")
-                user_id = n1[1:3]
-                encoded_prompt = n1[3:]
-                prompt = decode_prompt(encoded_prompt)
-                
-                img = generate_image(prompt)
-                if img is None:
-                    set_var("n0", "0")
-                    time.sleep(1)
-                    continue
-                
-                scratch_data = resize_and_encode(img)
-                packets = split_packets(scratch_data)
-                
-                for pkt in packets:
-                    set_var("n1", pkt)
-                    time.sleep(0.2)
-                    while get_var("n1") != "11":
-                        time.sleep(0.1)
-                
-                set_var("n0", "0")
+        n1 = tw.get("n1", None)
+        n0 = tw.get("n0", None)
+
+        if n1 and len(n1) >= 3 and n1[0] == "0" and n0 == 0:
+            tw["n0"] = 1  # 処理中フラグ
+
+            user_id = n1[1:3]
+            encoded_prompt = n1[3:]
+            prompt = decode_prompt(encoded_prompt)
+
+            # 画像生成
+            img = generate_image(prompt)
+            if img is None:
+                tw["n0"] = 0
+                time.sleep(1)
+                continue
+
+            scratch_data = resize_and_encode(img)
+            packets = split_packets(scratch_data)
+
+            # パケット送信
+            for pkt in packets:
+                tw["n1"] = pkt
+                time.sleep(0.2)
+                # 次パケット送信待ち
+                while tw.get("n1") != 11:
+                    time.sleep(0.1)
+
+            # 完了
+            tw["n0"] = 0
+
     except Exception as e:
-        print(f"[CRITICAL] Main loop error: {e}")
+        print(f"[CRITICAL] メインループ例外: {e}")
         traceback.print_exc()
     time.sleep(0.2)