grade_tool.py

import os
import re

def fix_moodle_output(assn_dir, outdir):
    for (_, _, filenames) in os.walk(assn_dir):
        for filename in filenames:
            # Remove spaces so we can use this as a package name
            name = filename.split("_")[0].replace(" ", "").replace("-", "")
            if not os.path.exists("%s/%s" % (outdir, name)):
                os.makedirs("%s/%s" % (outdir, name))
            file_type = filename.split(".")[-1]
            if file_type == "java":
                new_name = filename.split("_")[1]
                source = "%s/%s" % (assn_dir, filename)
                print(filename)
                java_class = open(source, encoding="utf-8", errors="ignore").read()
                modified_class = re.sub(
                    "package (.*?);",
                    "package %s;" % name,
                    java_class
                )

                dest = "%s/%s/%s.java" % (outdir, name, new_name)
                with open(dest, "w") as sink:
                    sink.write(modified_class)

if __name__ == "__main__":
    fix_moodle_output(<assignment_directory>, <package_destination>)