- import os, sys, subprocess, time
-
-
- class 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 pd\ndp = pd.read_csv("csv/%s")' % st
- return data