import os, sys, subprocess, timeclass AdvanceQuery(object): def __init__(self): self.TempFile = "media/advance_query" # 文件名 self.FileNum = int(time.time() * 1000) # python编译器位置 self.EXEC = sys.executable # 获取python版本 def get_version(self): v = sys.version_info version = "python %s.%s" % (v.major, v.minor) return version # 获得py文件名 def get_pyname(self): return '%d' % self.FileNum # 接收代码写入文件 def write_file(self, pyname, code): if not os.path.exists(self.TempFile): os.mkdir(self.TempFile) fpath = os.path.join(self.TempFile, '%s.py' % pyname) with open(fpath, 'w', encoding='utf-8') as f: f.write(code) # print('file path: %s' % fpath) return fpath # 编码 def decode(self, s): try: return s.decode('utf-8') except UnicodeDecodeError: return s.decode('gbk') def query_main(self, code): r = dict() r["version"] = self.get_version() pyname = self.get_pyname() fpath = self.write_file(pyname, code) try: # subprocess.check_output 是 父进程等待子进程完成，返回子进程向标准输出的输出结果 # stderr是标准输出的类型 """subprocess.check_output(args, *, stdin=None, stderr=None, shell=False, universal_newlines=False) 父进程直接创建子进程执行程序，以字符串的形式返回子进程的输出。""" outdata = self.decode(subprocess.check_output([self.EXEC, fpath], stderr=subprocess.STDOUT, timeout=5)) except subprocess.CalledProcessError as e: # e.output是错误信息标准输出 # 错误返回的数据 r["code"] = 'Error' r["output"] = self.decode(e.output) return r else: # 成功返回的数据 r['output'] = outdata r["code"] = "Success" return r finally: # 删除文件(其实不用删除临时文件会自动删除) try: os.remove(fpath) except Exception as e: exit(1) # 显示csv列表 def csv_list(self): path = os.getcwd() alist = [] for i in os.listdir(os.path.join(path, "csv")): # print(os.path.join(path, "csv")) if i.split(".")[1] == "csv": alist.append(i) data = {"csv_list": alist} return data # 根据csv返回相应语句显示 def show_statement(self, st): if st: data = 'import pandas as pdndp = pd.read_csv("csv/%s")' % st return data