#5 Create course endpoint. Added messages to all db migrations

backend/app/routes.py

@@ -5,7 +5,7 @@ from app.errors import error_response
 from flask_login import current_user
 
 from app import login, db
-from app.models import User
+from app.models import Course, User
 
 
 @login.user_loader
@@ -13,6 +13,13 @@ def load_user(user_id):
     return User.query.get(user_id)
 
 
+def check_data(data, required_fields):
+    for f in required_fields:
+        if f not in data:
+            return f
+    return None
+
+
 @bp.route("/login", methods=["POST"])
 def login_route():
     data = request.get_json()
@@ -46,11 +53,9 @@ def logout_route():
 @bp.route("/register", methods=["POST"])
 def register():
     data = request.get_json()
-
     required_fields = ["role", "username", "email", "password", "password2"]
-    for f in required_fields:
-        if f not in data:
-            return error_response(400, f"Must supply {f}")
+    if f := check_data(data, required_fields):
+        return error_response(400, f"Must supply {f}")
 
     if User.query.filter_by(username=data["username"]).first():
         return error_response(409, "User with that username already exists")
@@ -65,3 +70,28 @@ def register():
 
     resp = jsonify(u.to_dict())
     return resp
+
+
+@bp.route("/course", methods=["POST"])
+def create_course():
+    data = request.get_json()
+
+    required_fields = ["name", "description", "instructor"]
+
+    if f := check_data(data, required_fields):
+        return error_response(400, f"Must supply {f}")
+
+    u = User.query.get(data["instructor"])
+    if not u:
+        return error_response(400, f"User with id {data['instructor']} does not exist")
+
+    if u.role != "instructor":
+        return error_response(400, "User is not instructor")
+
+    c = Course()
+    c.from_dict(data)
+    db.session.add(c)
+    db.session.commit()
+
+    return jsonify(c.to_dict())
+