summaryrefslogtreecommitdiff
path: root/website/helpers.py
blob: 016f2fd409d5df05808bf71b95bfe3fbb2aaa0de (plain)
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
import os
import re
import time
import subprocess

from soc.settings import PROJECT_DIR
 
def scilab_run(code, token, example_id): 
    #Check for system commands
    system_commands = re.compile(
        'unix\(.*\)|unix_g\(.*\)|unix_w\(.*\)|unix_x\(.*\)|unix_s\(.*\)'
    )
    if system_commands.search(code):
        return "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')

    plot_path = ''
    if p.search(code):
        plot_exists = True
        code = code + '\n'
        current_time = time.time()
        plot_path = PROJECT_DIR + '/static/tmp/{0}.jpg'.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(\'/home/cheese/scimax/loader.sce\');\nmaxinit\n' + code

    file_path = PROJECT_DIR + '/static/tmp/' + token + '.sci'

    #thanks @prathamesh920 github
    #traps even syntax errors eg: endfunton
    f = open(file_path, "w")
    f.write('mode(2);\n')
    f.write(unicode(code))
    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 "lines(0)\nexec(\'{0}\',2);\nquit();"'.format(file_path)
    cmd += ' | /home/cheese/scilab-5.4.1/bin/scilab-adv-cli -nw'

    output = subprocess.Popen(
        cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True
    ).communicate()[0]

    #os.remove(file_path)
    output = trim(output)
    data = {
        'output': output,
        'plot_path': plot_path.replace(PROJECT_DIR, '')
    }
    return data

def trim(output):
    #for future use
    output = re.sub(r'\n \n \n', '\n', output)
    return output