from flask import Flask, render_template, request, jsonify
import sys
from io import StringIO
import contextlib
import subprocess
import importlib
import threading
import time
import ast
import reapp = Flask(__name__)RESTRICTED_PACKAGES = {'tkinter': '抱歉，在線編譯器不支持 tkinter，因為它需要圖形界面環境。請在本地運行需要GUI的代碼。','tk': '抱歉，在線編譯器不支持 tk/tkinter，因為它需要圖形界面環境。請在本地運行需要GUI的代碼。','pygame': 'pygame將被轉換為Web版本運行' # 不再限制pygame，而是轉換它
}def convert_tkinter_to_web(code):"""將tkinter代碼轉換為Web等效實現"""# 解析Python代碼tree = ast.parse(code)# 提取窗口屬性window_props = {'title': 'Python GUI','width': '700','height': '500','buttons': [],'labels': [],'entries': [],'layout': []}# 用于存儲函數定義functions = {}# 首先收集所有函數定義for node in ast.walk(tree):if isinstance(node, ast.FunctionDef):functions[node.name] = ast.unparse(node)# 分析代碼中的tkinter組件for node in ast.walk(tree):if isinstance(node, ast.Assign):if isinstance(node.value, ast.Call):# 提取窗口標題if hasattr(node.value.func, 'attr') and node.value.func.attr == 'Tk':for subnode in ast.walk(tree):if isinstance(subnode, ast.Call) and hasattr(subnode.func, 'attr'):if subnode.func.attr == 'title' and len(subnode.args) > 0:window_props['title'] = ast.literal_eval(subnode.args[0])elif subnode.func.attr == 'geometry' and len(subnode.args) > 0:geom = ast.literal_eval(subnode.args[0])match = re.match(r'(\d+)x(\d+)', geom)if match:window_props['width'] = match.group(1)window_props['height'] = match.group(2)# 提取按鈕elif hasattr(node.value.func, 'attr') and node.value.func.attr == 'Button':button = {'text': 'Button', 'command': None}for kw in node.value.keywords:if kw.arg == 'text':button['text'] = ast.literal_eval(kw.value)elif kw.arg == 'command':# 處理不同類型的commandif isinstance(kw.value, ast.Name):# 簡單的函數名button['command'] = kw.value.idelif isinstance(kw.value, ast.Lambda):# Lambda表達式button['command'] = f"lambda_{len(window_props['buttons'])}"functions[button['command']] = ast.unparse(kw.value)else:# 其他情況，嘗試轉換為字符串try:button['command'] = ast.unparse(kw.value)except:button['command'] = 'unknown_command'window_props['buttons'].append(button)# 提取標簽elif hasattr(node.value.func, 'attr') and node.value.func.attr == 'Label':label = {'text': ''}for kw in node.value.keywords:if kw.arg == 'text':try:label['text'] = ast.literal_eval(kw.value)except:# 如果不是字面量，嘗試直接轉換為字符串label['text'] = ast.unparse(kw.value)window_props['labels'].append(label)# 提取輸入框elif hasattr(node.value.func, 'attr') and node.value.func.attr == 'Entry':try:entry_id = node.targets[0].idexcept:entry_id = f"entry_{len(window_props['entries'])}"window_props['entries'].append({'id': entry_id})# 生成Web等效代碼web_code = f"""
<!DOCTYPE html>
<div class="tk-window" style="width: {window_props['width']}px; height: {window_props['height']}px;"><div class="tk-title-bar">{window_props['title']}</div><div class="tk-content">
"""# 添加標簽for label in window_props['labels']:web_code += f' <div class="tk-label">{label["text"]}</div>\n'# 添加輸入框for entry in window_props['entries']:web_code += f' <input type="text" class="tk-entry" id="{entry["id"]}">\n'# 添加按鈕for button in window_props['buttons']:command = button['command'] if button['command'] else ''web_code += f' <button class="tk-button" onclick="tkButtonClick(\'{command}\')">{button["text"]}</button>\n'web_code += """ </div>
</div>
<script>
window.pythonFunctions = {
"""# 添加Python函數定義for func_name, func_code in functions.items():web_code += f" '{func_name}': {func_code},\n"web_code += """};
</script>
"""return web_codedef convert_pygame_to_web(code):"""將pygame代碼轉換為Web Canvas實現"""web_code = """
<canvas id="pygame-canvas" style="border: 1px solid #000;"></canvas>
<script>
const canvas = document.getElementById('pygame-canvas');
const ctx = canvas.getContext('2d');// 設置畫布大小
canvas.width = 800;
canvas.height = 600;// 模擬 pygame 的基本功能
const pygame = {display: {set_mode: (size) => {canvas.width = size[0];canvas.height = size[1];return canvas;},update: () => {// Canvas 自動更新},flip: () => {// Canvas 自動更新}},draw: {rect: (surface, color, rect) => {ctx.fillStyle = `rgb(${color[0]},${color[1]},${color[2]})`;ctx.fillRect(rect[0], rect[1], rect[2], rect[3]);},circle: (surface, color, pos, radius) => {ctx.beginPath();ctx.fillStyle = `rgb(${color[0]},${color[1]},${color[2]})`;ctx.arc(pos[0], pos[1], radius, 0, Math.PI * 2);ctx.fill();}},event: {get: () => [], // 簡化的事件處理pump: () => {}},init: () => {},quit: () => {},time: {Clock: function() {return {tick: (fps) => 1000/fps};}}
};// 轉換后的Python代碼
function runGame() {try {// 這里將插入轉換后的游戲代碼%PYTHON_CODE%} catch (error) {console.error('Game error:', error);}
}// 啟動游戲循環
runGame();
</script>
"""# 處理 Python 代碼try:tree = ast.parse(code)# 轉換 Python 代碼為 JavaScriptjs_code = convert_pygame_code_to_js(tree)web_code = web_code.replace('%PYTHON_CODE%', js_code)return web_codeexcept Exception as e:return f"<div class='error'>轉換錯誤: {str(e)}</div>"def convert_pygame_code_to_js(tree):"""將 Python AST 轉換為 JavaScript 代碼"""js_code = []for node in ast.walk(tree):if isinstance(node, ast.Import):continue # 跳過導入語句elif isinstance(node, ast.Assign):# 轉換賦值語句if hasattr(node.value, 'func') and isinstance(node.value.func, ast.Attribute):if node.value.func.attr == 'set_mode':js_code.append(f"const screen = pygame.display.set_mode([{node.value.args[0].elts[0].n}, {node.value.args[0].elts[1].n}]);")elif isinstance(node, ast.While):# 轉換游戲主循環js_code.append("function gameLoop() {")# ... 處理循環體js_code.append(" requestAnimationFrame(gameLoop);")js_code.append("}")js_code.append("gameLoop();")return "\n".join(js_code)def install_package(package):"""自動安裝缺失的包"""# 檢查是否是受限制的包if package.lower() in RESTRICTED_PACKAGES:raise ImportError(RESTRICTED_PACKAGES[package.lower()])try:importlib.import_module(package)except ImportError:try:# 嘗試使用 pip 安裝包subprocess.check_call([sys.executable, "-m", "pip", "install", package])except subprocess.CalledProcessError as e:raise Exception(f"安裝包 {package} 失敗: {str(e)}")def timeout_handler():"""強制終止超時的代碼執行"""raise TimeoutError("代碼執行超時（最大執行時間：5秒）")@app.route('/')
def index():return render_template('index.html')@app.route('/execute', methods=['POST'])
def execute_code():code = request.json.get('code', '')try:# 檢測是否包含pygame代碼if 'pygame' in code:web_code = convert_pygame_to_web(code)return jsonify({'status': 'success','output': '','gui': web_code})# 檢測是否包含tkinter代碼elif 'tkinter' in code or 'tk' in code:web_code = convert_tkinter_to_web(code)return jsonify({'status': 'success','output': '','gui': web_code})# 非GUI代碼正常執行output_buffer = StringIO()with contextlib.redirect_stdout(output_buffer):exec(code, globals(), {})output = output_buffer.getvalue()return jsonify({'status': 'success','output': output if output else '程序執行完成，沒有輸出'})except Exception as e:return jsonify({'status': 'error','output': f'錯誤: {str(e)}'})if __name__ == '__main__':app.run(debug=True)
)
