Create a gist now

Instantly share code, notes, and snippets.

What would you like to do?
Return nbgrader feedback to students
"""Return feedback to students that was generated with `nbgrader feedback`.
This must be run from the root of the nbgrader directory. You probably need to run
it with sudo since you need to write to other users' directories.
sudo python problem_set_name
import os
import shutil
import pwd
import argparse
import stat
def set_permissions(path, uid, gid):
os.chown(path, uid, gid)
if os.path.isdir(path):
os.chmod(path, stat.S_IRUSR | stat.S_IRGRP | stat.S_IXUSR | stat.S_IXGRP)
os.chmod(path, stat.S_IRUSR | stat.S_IRGRP)
def main(name, student=None, force=False):
feedback_dir = os.path.abspath(os.path.join("feedback", name))
if student is None:
students = sorted(os.listdir(feedback_dir))
students = [student]
for student in students:
src = os.path.join(feedback_dir, student)
dst = os.path.join("/", "home", student, "{} feedback".format(name))
# remove existing feedback if it exists
if os.path.exists(dst):
if force:
print("removing '{}'".format(dst))
print("skipping {}, feedback already exists".format(student))
# copy the feedback
print("'{}' --> '{}'".format(src, dst))
shutil.copytree(src, dst)
# get the uid and gid
pwinfo = pwd.getpwnam(student)
uid = pwinfo.pw_uid
gid = pwinfo.pw_gid
# set the owner to be the student and permissions to be read-only
set_permissions(dst, uid, gid)
for dirname, dirnames, filenames in os.walk(dst):
for f in (dirnames + filenames):
path = os.path.join(dirname, f)
set_permissions(path, uid, gid)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('name', help='the name of the assignment')
parser.add_argument('--student', default=None, help='the name of a specific student')
parser.add_argument('--force', action="store_true", default=False,
help='overwrite existing feedback (use with extreme caution!!)')
args = parser.parse_args()
main(, student=args.student, force=args.force)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment