1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
import os, re, sys, time, subprocess
from soc.settings import PROJECT_DIR
from timeout import TimerTask
from soc.config import SCILAB_BIN, SCILAB_FLAGS, SCIMAX_LOADER, UPLOADS_PATH
def scilab_run(code, token, book_id, dependency_exists):
#Check for system commands
system_commands = re.compile(
'unix\(.*\)|unix_g\(.*\)|unix_w\(.*\)|unix_x\(.*\)|unix_s\(.*\)|host|newfun|execstr|ascii|mputl|dir\(\)'
)
if system_commands.search(code):
return {
'output': 'System Commands not allowed',
}
#Remove all clear;
code = re.sub(r'clear.*all|clear|clc\(\)|clc', '', code)
plot_exists = False
#Finding the plot and appending xs2jpg function
#p = re.compile(r'.*plot.*\(.*\).*\n|bode\(.*\)|evans\(.*\)')
p = re.compile(r'plot*|.*plot.*\(.*\).*\n|bode\(.*\)|evans\(.*\)')
plot_path = ''
if p.search(code):
plot_exists = True
code = code + '\n'
current_time = time.time()
plot_path = PROJECT_DIR + '/static/tmp/{0}.png'.format(str(current_time))
#code += 'xs2jpg(gcf(), "{0}");\n'.format(plot_path)
#Check whether to load scimax / maxima
if 'syms' in code or 'Syms' in code:
code = code.replace('syms', 'Syms')
code = 'exec(\'{0}\');\nmaxinit\n'.format(SCIMAX_LOADER) + code
file_path = PROJECT_DIR + '/static/tmp/' + token + '.sci'
#traps even syntax errors eg: endfunton
f = open(file_path, "w")
f.write('driver("PNG");\n')
f.write('xinit("{0}");\n'.format(plot_path))
f.write('mode(2);\n')
if dependency_exists:
f.write(
'getd("{0}/{1}/DEPENDENCIES/");'.format(UPLOADS_PATH, book_id)
)
f.write('lines(0);\n')
f.write(unicode(code))
f.write('\nxend();')
f.write('\nquit();')
f.close()
#this makes it possible to execute scilab without the problem of \
#getting stuck in the prompt in case of error
cmd = 'printf "exec(\'{0}\',2);\nquit();"'.format(file_path)
cmd += ' | {0} {1}'.format(SCILAB_BIN, SCILAB_FLAGS)
task = TimerTask(cmd, timeout=15)
output = task.run().communicate()[0]
e = task.wait()
output = trim(output)
data = {
'output': output,
'plot_path': plot_path.replace(PROJECT_DIR, '')
}
return data
def trim(output):
output = [line for line in output.split('\n') if line.strip() != '']
output = '\n'.join(output)
return output
|
