def process_roster(imported_file, user):
  try:
    csv_file = cStringIO.StringIO(imported_file)
    # Read the first kb to ensure the file is a valid CSV file.
    csv.Sniffer().sniff(csv_file.read(1024), ",")
    csv_file.seek(0)
    reader = csv.DictReader(csv_file, dialect="excel")
  except:
    raise Exception("Invalid CSV file")
  reader.fieldnames = [re.compile('[\W_]+', flags=re.UNICODE).sub('', field).lower()
                       for field in reader.fieldnames]
  for row in reader:
    rose_username = row.get("username", None)
    new_student = Student(parent=utils.get_parent_key(user),
                          id=rose_username,
                          first_name=row.get("first", None),
                          last_name=row.get("last", None),
                          team=row.get("team", None),
                          rose_username=rose_username)
    new_student.put()
Exemplo n.º 2
0
def get_all_notifications_for_user(user):
    """ Gets all of the contacts for this user. """
    parent_key = utils.get_parent_key(user)
    return Notification.query(ancestor=parent_key)
def get_csv_export_lists(user, export_student_name, export_rose_username,
                         export_team, urlsafe_assignment_keys):
  table_data = []
  student_row_index_map = {} # Map of student_key to row in the table_data
  assignment_col_index_map = {} # Map of assignment_key to column in the table_data
  header_row = []
  table_data.append(header_row)
  num_columns = 0

  # Student Header
  if export_student_name:
    header_row.append("First")
    header_row.append("Last")
    num_columns += 2
  if export_rose_username:
    header_row.append("Username")
    num_columns += 1
  if export_team:
    header_row.append("Team")
    num_columns += 1

  # Assignment Prep
  assignment_keys = []
  for urlsafe_assignment_key in urlsafe_assignment_keys:
    assignment_keys.append(ndb.Key(urlsafe=urlsafe_assignment_key))
  assignments = ndb.get_multi(assignment_keys)
  assignments.sort(key=lambda assignment: assignment.name)
  num_assignments_found = 0
  for assignment in assignments:
    if assignment:
      header_row.append(assignment.name)
      assignment_col_index_map[assignment.key] = num_columns
      num_columns += 1
      num_assignments_found += 1

  # Student Data + assignment placeholders
  num_rows = 1
  students = Student.query(ancestor=utils.get_parent_key(user)).order(Student.rose_username)
  for student in students:
    current_row = []
    if export_student_name:
      current_row.append(student.first_name)
      current_row.append(student.last_name)
    if export_rose_username:
      current_row.append(student.rose_username)
    if export_team:
      current_row.append(student.team)
    for i in range(num_assignments_found):
      current_row.append("-")
    table_data.append(current_row)
    student_row_index_map[student.key] = num_rows
    num_rows += 1

  # Add the grades
  grade_query = GradeEntry.query(ancestor=utils.get_parent_key(user))
  for grade in grade_query:
    if grade.student_key in student_row_index_map and grade.assignment_key in assignment_col_index_map:
      row = student_row_index_map[grade.student_key]
      col = assignment_col_index_map[grade.assignment_key]
      table_data[row][col] = grade.score

  # Removing rows with no grades (allows for data merging)
  for row_index in reversed(range(1, num_rows)):
    row = table_data[row_index]
    blank_grades = row.count("-")
    if blank_grades == num_assignments_found:
      table_data.remove(row)

  return table_data