def patch_sole(course_id, sole_id): """ Join or leave a sole """ user, err = auth.get_user_from_request(db, request) if not user: return json_error("User not found") sole = Sole.find_by_id(db, sole_id) if not sole: return json_error("that study group doesn't exist") # if the new set of student ids makes sense, use them data = json.loads(request.data) current_sids = sole.get(Sole.A_STUDENT_IDS) new_sids = data.get(Sole.A_STUDENT_IDS) user_id = user.get('id') resp = None action = None if (set(current_sids) - set(new_sids)) == set([user_id]): resp = Sole.leave_sole_by_id(db, sole_id, user_id) action = 'left' elif (set(new_sids) - set(current_sids)) == set([user_id]): resp = Sole.join_sole_by_id(db, sole_id, user_id) action = 'joined' else: return json_error("Invalid new student_ids") # return the new model if resp: s = Sole.find_by_id(db, sole_id) ns = User.update_sole_with_students(db, s) resq.enqueue(EmailGroup, action, sole_id, user_id) return json.dumps(ns) else: return json_error("Some other error")