helper.py

#!/usr/bin/python3
import os, sys
import random
import string
import subprocess
import utils.parser as p
import utils.db as db
import json
from flask import Flask, request
from flask_restful import Resource, Api

# Set the languages allowed
allowedLanguages = {"java17", "java15", "java15dm",
                     "java12", "java11", "python3",
                      "c/c++", "c#-1.2", "char",
                       "text", "scheme"}

# Don't touch. This is for the api
app = Flask(__name__)
api = Api(app)

"""
Simple helper class to catch the data
"""
class Helper(Resource):
    def get(self):
        # Fetch needed variables
        language = request.args.get('language')
        directory = request.args.get('directory')
        studentID =  request.args.get('studentID')
        threshold = request.args.get('threshold')
        language = checkLanguage(language)  #checks to make sure language is properly parsed
        folder = executejplag(language, directory, threshold)
        data = p.parseFile(folder)
        db.storeData(data, config, directory)
        subprocess.check_output(["rm", "-r" , folder])
        dump = json.dumps(studentInfo(data,studentID))
        return dump

"""
Replaces language variable so it can be passed properly through api.
"""
def checkLanguage(language):
    if(language == "c/cpp"):
        language = "c/c++"
    return language

"""
Generate folder name
"""
def genName():
    folder = ''
    while True:
        folder = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(8))
        if(not(os.path.isdir(folder))):
            break
    return folder

"""
Fetch student specific information
"""
def studentInfo(data, studentID):
    if(studentID == None):
        return data
    studentData = []
    for i in range(len(data)):
        d = data[i]
        if(d[0] == studentID or d[1] == studentID):
            studentData.append(d)
    return studentData
"""
Executes the jplag script
"""
def executejplag(lang, direct, threshold):
    # generates a random folder name so chance of conflicting folders if run on the same time is minimal
    folder = genName()
    if(threshold == None):
        threshold = "20"
    threshold = threshold + "%"
    subprocess.check_output(["java", "-jar", "jplag.jar", "-l", lang, "-s", direct, "-r", folder, "-m", threshold])
    return folder


api.add_resource(Helper, '/')

if __name__ == "__main__":
    with open('config/config.json', 'r') as f:
        config = json.load(f)
    app.run(debug=False)