From c7de51ca9a99928c976246f77cc276512e7c8392 Mon Sep 17 00:00:00 2001
From: Yeuoly <45712896+Yeuoly@users.noreply.github.com>
Date: Wed, 17 Apr 2024 19:49:53 +0800
Subject: [PATCH] enhance: preload general packages (#3567)

---
 .../code_executor/python_transformer.py       | 24 +++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/api/core/helper/code_executor/python_transformer.py b/api/core/helper/code_executor/python_transformer.py
index ca758c1efa..d6a13e9923 100644
--- a/api/core/helper/code_executor/python_transformer.py
+++ b/api/core/helper/code_executor/python_transformer.py
@@ -20,8 +20,28 @@ result = f'''<<RESULT>>
 print(result)
 """
 
-PYTHON_PRELOAD = """"""
-
+PYTHON_PRELOAD = """
+# prepare general imports
+import json
+import datetime
+import math
+import random
+import re
+import string
+import sys
+import time
+import traceback
+import uuid
+import os
+import base64
+import hashlib
+import hmac
+import binascii
+import collections
+import functools
+import operator
+import itertools
+"""
 
 class PythonTemplateTransformer(TemplateTransformer):
     @classmethod