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