moss/prune_files.py at master · sniyaz/moss · GitHub

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
import json
import sys
import os

from shutil import copyfile

import pdb

# NOTE: Edit files for CSE 373 HW7 at UW.
# TODO: Change this for your courses HW.
EDIT_FILES = [
    "src/main/java/datastructures/concrete/ArrayDisjointSet.java",
    "src/main/java/misc/graphs/Graph.java",
    "src/main/java/mazes/generators/maze/KruskalMazeCarver.java"
]

def copy_files(submission_dir, out_dir, join_out):
    student_dirs = os.listdir(submission_dir)
    # Gotta prune out files. Only dirs.
    student_dirs = [d for d in student_dirs if os.path.isdir(os.path.join(submission_dir, d))]

    for cur_dir in student_dirs:
        full_dir = os.path.join(submission_dir, cur_dir)
        # Make the output dir.
        cur_copy_dir = os.path.join(out_dir, cur_dir)
        os.mkdir(cur_copy_dir)

        student_files = []

        for cur_edit_file in EDIT_FILES:
            cur_edit_path = os.path.join(full_dir, cur_edit_file)

            # Copy this edit file over.
            edit_basename = os.path.basename(cur_edit_file)
            cur_copy_target = os.path.join(cur_copy_dir, edit_basename)
            copyfile(cur_edit_path, cur_copy_target)

            student_files.append(cur_copy_target)

        # NOTE: Moss takes the skeleton as a single file. So just join all files
        # into a single one.
        # TODO: Correct the file extension if you aren't using java.
        cur_join_file = cur_dir + "_join.java"
        cur_join_target = os.path.join(join_out, cur_join_file)

        # NOTE: Little hack so they all get joined in the same order. MOSS
        # shouldn't care, but hey!
        student_files.sort()
        cat_command = ["cat"] + student_files + [">"] + [cur_join_target]
        cat_command = " ".join(cat_command)
        os.system(cat_command)


if __name__ == '__main__':

    orig_repos = sys.argv[1]
    pruned_out_dir = sys.argv[2]
    joined_out_dir = sys.argv[3]

    copy_files(orig_repos, pruned_out_dir, joined_out_dir)