| import asyncio |
| import websockets |
| import io |
| import sys |
| import os as ps |
| import shutil |
|
|
|
|
| def execute_code(code): |
| try: |
| stdout_capture = io.StringIO() |
| sys.stdout = stdout_capture |
| exec(code, globals()) |
| stdout_output = stdout_capture.getvalue() |
| sys.stdout = sys.__stdout__ |
| return stdout_output |
| except Exception as e: |
| return str(e) |
|
|
|
|
| def move_files(): |
| assets_path = "/app/pyassets" |
| current_dir = ps.getcwd() |
| moved_files = [] |
| for filename in ps.listdir(current_dir): |
| full_path = ps.path.join(current_dir, filename) |
| if ps.path.isfile(full_path) and filename != 'server.py': |
| new_path = ps.path.join(assets_path, filename) |
| shutil.move(full_path, new_path) |
| moved_files.append(filename) |
| return moved_files |
|
|
|
|
| def generate_links(filenames): |
| base_url = ps.environ['BASE_URL'] |
| if(base_url == ''): |
| return |
| base_url = base_url + '/assets/pyassets' |
| links = [f"" for filename in filenames] |
| return links |
|
|
|
|
| async def handle_client(websocket, path): |
| try: |
| code = await websocket.recv() |
| result = execute_code(code) |
| filenames = move_files() |
| links = generate_links(filenames) |
| final_result = f"{result}\n" |
| final_result += "\n".join(links) |
| await websocket.send(final_result) |
| except websockets.exceptions.ConnectionClosed: |
| pass |
|
|
| async def server(websocket, path): |
| print(f"Incoming connection from {websocket.remote_address}") |
| while True: |
| await handle_client(websocket, path) |
|
|
| start_server = websockets.serve(server, '0.0.0.0', 3380) |
|
|
| print('Server started, listening on localhost:3380') |
|
|
| asyncio.get_event_loop().run_until_complete(start_server) |
| asyncio.get_event_loop().run_forever() |
