删除临时代码文件，并设置回复信息的最大行数与最大长度

shenxgan · May 29, 2024 · cd667c9 · cd667c9
1 parent 8896c9b
commit cd667c9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/server.py b/python/server.py
@@ -1,5 +1,6 @@
 import asyncio
 import datetime
+import os
 
 from sanic import Sanic
 from sanic import text
@@ -26,9 +27,12 @@ async def run_code(request):
     )
     stdout, stderr = await proc.communicate()
     msg = stdout.decode() + stderr.decode()
+    os.remove(fname)
 
-    _msg = msg.split('\n')[:15]
-    res = '\n'.join(_msg)
+    max_line = os.getenv('OUTPUT_MAX_LINE', 15)
+    max_len = os.getenv('OUTPUT_MAX_LEN', 256)
+    _msg = msg.split('\n')[:max_line]
+    res = '\n'.join(_msg)[:max_len]
     return text(res)