From 2cec42004eb62adee573fb802c85502de9c2a9e3 Mon Sep 17 00:00:00 2001
From: Jagraj Aulakh <aulakhjagraj@gmail.com>
Date: Thu, 13 Apr 2023 18:44:43 -0400
Subject: [PATCH] #10 Made endpoint to create a new assignment

---
 backend/app/models.py | 23 +++++++++++++++++++++++
 backend/app/routes.py | 25 ++++++++++++++++++++++++-
 2 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/backend/app/models.py b/backend/app/models.py
index c1a59bc..de4d95b 100644
--- a/backend/app/models.py
+++ b/backend/app/models.py
@@ -105,3 +105,26 @@ class Assignment(db.Model):
     description = sa.Column(sa.Text, index=True)
     due_date = sa.Column(sa.DateTime)
     created_at = sa.Column(sa.DateTime)
+
+    def from_dict(self, data) -> None:
+        for field in ["name", "course_id",  "description", "due_date"]:
+            if field in data:
+                setattr(self, field, data[field])
+
+        if not self.created_at:
+            self.created_at = datetime.now()
+
+    def from_dict(self, data) -> None:
+        for field in ["name", "course_id",  "description", "due_date"]:
+            if field in data:
+                setattr(self, field, data[field])
+
+        if not self.created_at:
+            self.created_at = datetime.now()
+
+    def to_dict(self) -> dict:
+        d = {}
+        for f in ["id", "name", "course_id",  "description", "due_date", "created_at"]:
+            d[f] = getattr(self, f)
+
+        return d
diff --git a/backend/app/routes.py b/backend/app/routes.py
index 5c57a2e..bf63657 100644
--- a/backend/app/routes.py
+++ b/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())