#10 Made endpoint to create a new assignment

backend/app/routes.py

@@ -6,7 +6,7 @@ from app.errors import error_response
 from flask_login import current_user
 
 from app import login, db
-from app.models import Course, User
+from app.models import Course, User, Assignment
 
 
 @login.user_loader
@@ -125,6 +125,7 @@ def get_courses(id):
     resp = jsonify(d)
     return resp
 
+
 @bp.route("/course/<int:id>/students", methods=["GET"])
 @login_required
 def get_students_in_course(id):
@@ -138,6 +139,7 @@ def get_students_in_course(id):
         resp["students"].append(s.to_dict())
     return jsonify(resp)
 
+
 @bp.route("/course/<int:id>", methods=["GET"])
 @login_required
 def get_course(id):
@@ -147,6 +149,7 @@ def get_course(id):
     resp = jsonify(c.to_dict())
     return resp
 
+
 @bp.route("/user/<int:uid>/enroll/<int:cid>", methods=["POST", "DELETE"])
 @login_required
 @instructor_required
@@ -172,6 +175,7 @@ def enroll_student(uid, cid):
     resp = {"user": u.to_dict(), "course": c.to_dict()}
     return jsonify(resp)
 
+
 @bp.route("/user/<string:username>/enroll/<int:cid>", methods=["POST"])
 @login_required
 @instructor_required
@@ -197,3 +201,22 @@ def enroll_student_by_username(username, cid):
     resp = {"user": u.to_dict(), "course": c.to_dict()}
     return jsonify(resp)
 
+
+@bp.route("/assignment", methods=["POST"])
+@login_required
+@instructor_required
+def create_assignment():
+    data = request.get_json()
+    required_fields = ["name", "description", "course_id", "due_date"]
+    if f := check_data(data, required_fields):
+        return error_response(400, f"Must supply {f}")
+
+    c = Course.query.get(data["course_id"])
+    if not c:
+        return error_response(400, f"Course with id {data['course_id']} does not exist")
+
+    a = Assignment()
+    a.from_dict(data)
+    db.session.add(a)
+    db.session.commit()
+    return jsonify(a.to_dict())