Enhancement/projects file upload

backend/project/endpoints/projects/endpoint_parser.py

@@ -3,18 +3,34 @@
 """
 
 from flask_restful import reqparse
+import werkzeug
 
 parser = reqparse.RequestParser()
-parser.add_argument('title', type=str, help='Projects title')
-parser.add_argument('descriptions', type=str, help='Projects description')
-parser.add_argument('assignment_file', type=str, help='Projects assignment file')
-parser.add_argument("deadline", type=str, help='Projects deadline')
-parser.add_argument("course_id", type=str, help='Projects course_id')
-parser.add_argument("visible_for_students", type=bool, help='Projects visibility for students')
-parser.add_argument("archieved", type=bool, help='Projects')
-parser.add_argument("test_path", type=str, help='Projects test path')
-parser.add_argument("script_name", type=str, help='Projects test script path')
-parser.add_argument("regex_expressions", type=str, help='Projects regex expressions')
+parser.add_argument('title', type=str, help='Projects title', location="form")
+parser.add_argument('descriptions', type=str, help='Projects description', location="form")
+parser.add_argument(
+    'assignment_file',
+    type=werkzeug.datastructures.FileStorage,
+    help='Projects assignment file',
+    location="form"
+)
+parser.add_argument("deadline", type=str, help='Projects deadline', location="form")
+parser.add_argument("course_id", type=str, help='Projects course_id', location="form")
+parser.add_argument(
+    "visible_for_students",
+    type=bool,
+    help='Projects visibility for students',
+    location="form"
+)
+parser.add_argument("archieved", type=bool, help='Projects', location="form")
+parser.add_argument("test_path", type=str, help='Projects test path', location="form")
+parser.add_argument("script_name", type=str, help='Projects test script path', location="form")
+parser.add_argument(
+    "regex_expressions",
+    type=str,
+    help='Projects regex expressions',
+    location="form"
+)
 
 
 def parse_project_params():

backend/project/endpoints/projects/project_detail.py

@@ -9,9 +9,10 @@
 from flask import request
 from flask_restful import Resource
 
-from project.models.project import Project
 from project.utils.query_agent import query_by_id_from_model, delete_by_id_from_model, \
     patch_by_id_from_model
+from project.models.project import Project
+
 
 
 API_URL = getenv('API_HOST')

backend/project/endpoints/projects/project_endpoint.py

@@ -4,13 +4,11 @@
 """
 
 from flask import Blueprint
-from flask_restful import Api
 
 from project.endpoints.projects.projects import ProjectsEndpoint
 from project.endpoints.projects.project_detail import ProjectDetail
 
 project_bp = Blueprint('project_endpoint', __name__)
-project_endpoint = Api(project_bp)
 
 project_bp.add_url_rule(
     '/projects',

backend/project/endpoints/projects/projects.py

@@ -1,16 +1,34 @@
 """
 Module that implements the /projects endpoint of the API
 """
+import os
 from os import getenv
 from urllib.parse import urljoin
+import zipfile
 
 from flask import request
 from flask_restful import Resource
 
 from project.models.project import Project
-from project.utils.query_agent import query_selected_from_model, insert_into_model
+from project.utils.query_agent import query_selected_from_model, create_model_instance
+
+from project.endpoints.projects.endpoint_parser import parse_project_params
 
 API_URL = getenv('API_HOST')
+UPLOAD_FOLDER = getenv('UPLOAD_URL')
+ALLOWED_EXTENSIONS = {'zip'}
+
+
+
+
+
+def allowed_file(filename: str):
+    """
+    check if file extension is allowed for upload
+    """
+    return '.' in filename and \
+        filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
 
 class ProjectsEndpoint(Resource):
     """
@@ -40,7 +58,40 @@ def post(self):
         using flask_restfull parse lib
         """
 
-        return insert_into_model(
-            Project,request.json,
+        file = request.files["assignment_file"]
+        project_json = parse_project_params()
+        filename = file.filename.split("/")[-1]
+
+        # save the file that is given with the request
+
+        new_project = create_model_instance(
+            Project,
+            project_json,
             urljoin(API_URL, "/projects"),
-            "project_id")
+            required_fields=[
+                "title",
+                "descriptions",
+                "course_id",
+                "visible_for_students",
+                "archieved"]
+        )
+
+        project_upload_directory = f"{UPLOAD_FOLDER}{new_project.project_id}"
+
+        file_location = "." + os.path.join(project_upload_directory)
+
+        if not os.path.exists(project_upload_directory):
+            os.makedirs(file_location, exist_ok=True)
+
+        file.save(file_location + "/" + filename)
+        try:
+            with zipfile.ZipFile(file_location + "/" + filename) as upload_zip:
+                upload_zip.extractall(file_location)
+        except zipfile.BadZipfile:
+            return {"message": "Please provide a .zip file for uploading the instructions"}, 400
+
+        return {
+            "message": "Project created succesfully",
+            "data": new_project,
+            "url": f"{API_URL}/projects/{new_project.project_id}"
+        }, 201

backend/project/utils/query_agent.py

@@ -50,6 +50,31 @@ def delete_by_id_from_model(
         return {"error": "Something went wrong while deleting from the database.",
                 "url": base_url}, 500
 
+
+def create_model_instance(model: DeclarativeMeta,
+                      data: Dict[str, Union[str, int]],
+                      response_url_base: str,
+                      required_fields: List[str] = None):
+    """
+    Create an instance of a model
+    """
+    if required_fields is None:
+        required_fields = []
+    # Check if all non-nullable fields are present in the data
+    missing_fields = [field for field in required_fields if field not in data]
+
+    if missing_fields:
+        return {"error": f"Missing required fields: {', '.join(missing_fields)}",
+                "url": response_url_base}, 400
+
+    filtered_data = filter_model_fields(model, data)
+    new_instance: DeclarativeMeta = model(**filtered_data)
+    db.session.add(new_instance)
+    db.session.commit()
+
+    return new_instance
+
+
 def insert_into_model(model: DeclarativeMeta,
                       data: Dict[str, Union[str, int]],
                       response_url_base: str,
@@ -69,26 +94,21 @@ def insert_into_model(model: DeclarativeMeta,
         a message indicating that something went wrong while inserting into the database.
     """
     try:
-        if required_fields is None:
-            required_fields = []
-        # Check if all non-nullable fields are present in the data
-        missing_fields = [field for field in required_fields if field not in data]
-
-        if missing_fields:
-            return {"error": f"Missing required fields: {', '.join(missing_fields)}",
-                    "url": response_url_base}, 400
-
-        filtered_data = filter_model_fields(model, data)
-        new_instance: DeclarativeMeta = model(**filtered_data)
-        db.session.add(new_instance)
-        db.session.commit()
-        return jsonify({
+        new_instance = create_model_instance(model, data, response_url_base, required_fields)
+        # if its a tuple the model instance couldn't be created so it already
+        # is the right format of error message and we just need to return
+        if isinstance(new_instance, tuple):
+            return new_instance
+
+        return (jsonify({
                 "data": new_instance,
                 "message": "Object created succesfully.",
-                "url": urljoin(
-                    f"{response_url_base}/",
-                    str(getattr(new_instance, url_id_field)))}), 201
+                "url":
+                    urljoin(response_url_base + "/",
+                            str(getattr(new_instance, url_id_field)))}),
+                201)
     except SQLAlchemyError:
+        db.session.rollback()
         return jsonify({"error": "Something went wrong while inserting into the database.",
                 "url": response_url_base}), 500
 

backend/tests.yaml

@@ -29,6 +29,7 @@ services:
       POSTGRES_PASSWORD: test_password
       POSTGRES_DB: test_database
       API_HOST: http://api_is_here
+      UPLOAD_URL: /project/endpoints/uploads/
     volumes:
       - .:/app
     command: ["pytest"]

backend/tests/endpoints/conftest.py

@@ -33,7 +33,6 @@ def project(course):
         title="Project",
         descriptions="Test project",
         course_id=course.course_id,
-        assignment_file="testfile",
         deadline=date,
         visible_for_students=True,
         archieved=False,

backend/tests/endpoints/project_test.py

@@ -23,9 +23,16 @@ def test_post_project(db_session, client, course_ad, course_teacher_ad, project_
     db_session.commit()
 
     project_json["course_id"] = course_ad.course_id
+    # cant be done with 'with' because it autocloses then
+    # pylint: disable=R1732
+    project_json["assignment_file"] = open("testzip.zip", "rb")
 
     # post the project
-    response = client.post("/projects", json=project_json)
+    response = client.post(
+        "/projects",
+        data=project_json,
+        content_type='multipart/form-data'
+    )
     assert response.status_code == 201
 
     # check if the project with the id is present
@@ -34,7 +41,6 @@ def test_post_project(db_session, client, course_ad, course_teacher_ad, project_
 
     assert response.status_code == 200
 
-
 def test_remove_project(db_session, client, course_ad, course_teacher_ad, project_json):
     """Test removing a project to the datab and fetching it, testing if it's not present anymore"""
 
@@ -46,8 +52,11 @@ def test_remove_project(db_session, client, course_ad, course_teacher_ad, projec
 
     project_json["course_id"] = course_ad.course_id
 
+    # cant be done with 'with' because it autocloses then
+    # pylint: disable=R1732
+    project_json["assignment_file"] = open("testzip.zip", "rb")
     # post the project
-    response = client.post("/projects", json=project_json)
+    response = client.post("/projects", data=project_json)
 
     # check if the project with the id is present
     project_id = response.json["data"]["project_id"]
@@ -59,7 +68,6 @@ def test_remove_project(db_session, client, course_ad, course_teacher_ad, projec
     response = client.delete(f"/projects/{project_id}")
     assert response.status_code == 404
 
-
 def test_patch_project(db_session, client, course_ad, course_teacher_ad, project):
     """
     Test functionality of the PUT method for projects