Fast api conversion subsemester

src/api/Updating to FastAPI

@@ -20,7 +20,7 @@ UPDATING YOUR ENDPOINTS
             depending on your database call's response types
 5. Updating pytests:
     * AFTER running your pytests as before to make sure your endpoint's functionality has not changed:
-        * Client becomes TestClient. Replace from util import Client with from fastapi import TestClient
+        * Client becomes TestClient. Replace from util import Client with from fastapi.testclient import TestClient
             and replace 'client: Client' with 'client: TestClient'
         * Mark your pytests as updated- import pytest and add the decorator (@pytest.mark.testclient)
             in front of each test case

src/api/api_models.py

@@ -1,4 +1,6 @@
 from pydantic import BaseModel
+from typing import Optional
+
 
 class SessionPydantic(BaseModel):
     email: str
@@ -7,3 +9,5 @@ class SessionPydantic(BaseModel):
 class SessionDeletePydantic(BaseModel):
     sessionID: str
 
+class SubsemesterPydantic(BaseModel):
+    semester: Optional[str] = None

src/api/app.py

@@ -5,6 +5,7 @@
 from fastapi_cache.backends.inmemory import InMemoryBackend
 from fastapi_cache.decorator import cache
 from fastapi_cache.coder import PickleCoder
+from fastapi import Depends
 
 from api_models import *
 import db.connection as connection
@@ -93,34 +94,38 @@ async def get_departments():
     return departments if not error else Response(content=error, status_code=500)
 
 
-# @app.route('/api/subsemester', methods=['GET'])
-# @cache.cached(timeout=Constants.HOUR_IN_SECONDS, query_string=True)
-# def get_subsemesters():
-#     """
-#     GET /api/subsemester?semester={}
-#     Cached: 1 Hour
-#
-#     Get list of departments i.e. COGS, CIVL, CSCI, BIOL
-#     (Used in dropdown in "Course Search"
-#     """
-#     semester = request.args.get("semester", default=None)
-#     if semester:
-#         subsemesters, error = class_info.get_subsemesters(semester)
-#         return jsonify(subsemesters) if not error else Response(error, status=500)
-#     # Some cases, we do want all subsemesters across all semesters like in Admin Panel
-#     subsemesters, error = class_info.get_subsemesters()
-#     return jsonify(subsemesters) if not error else Response(error, status=500)
-#
-# @app.route('/api/semester', methods=['GET'])
-# @cache.cached(timeout=Constants.DAY_IN_SECONDS)
-# def get_semesters():
-#     """
-#     GET /api/semester
-#     Cached: 24 Hours
-#     """
-#     semesters, error = class_info.get_semesters()
-#     return jsonify(semesters) if not error else Response(error, status=500)
-#
+@app.get('/api/subsemester')
+@cache(expire=Constants.HOUR_IN_SECONDS, coder=PickleCoder, namespace="API_CACHE")
+async def get_subsemesters(subsemester: SubsemesterPydantic = Depends(SubsemesterPydantic)):
+    """
+    GET /api/subsemester?semester={}
+    Cached: 1 Hour
+
+    Get list of departments i.e. COGS, CIVL, CSCI, BIOL
+    (Used in dropdown in "Course Search"
+    """
+    if subsemester.semester:
+        subsemesters, error = class_info.get_subsemesters(subsemester.semester)
+        # for i in subsemesters:
+        #     print(i)
+        db_list = [dict(r) for r in subsemesters]
+        return db_list if not error else Response(error, status_code=500)
+    # Some cases, we do want all subsemesters across all semesters like in Admin Panel
+    subsemesters, error = class_info.get_subsemesters()
+    db_list = [dict(r) for r in subsemesters]
+    return db_list if not error else Response(error, status_code=500)
+
+@app.get('/api/semester')
+@cache(expire=Constants.DAY_IN_SECONDS, coder=PickleCoder, namespace="API_CACHE")
+async def get_semesters():
+    """
+    GET /api/semester
+    Cached: 24 Hours
+    """
+    semesters, error = class_info.get_semesters()
+    db_list = [dict(r) for r in semesters]
+    return db_list if not error else Response(error, status_code=500)
+
 # @app.route('/api/semesterInfo', methods=['GET'])
 # def get_all_semester_info():
 #     all_semester_info, error = class_info.get_all_semester_info()

src/api/db/connection.py

@@ -9,7 +9,6 @@
 DB_PORT = os.environ.get('DB_PORT', None)
 DB_PASS = os.environ.get('DB_PASS', None)
 
-
 class database():
     def connect(self):
         self.conn = psycopg2.connect(

src/api/pytest.ini

@@ -6,6 +6,4 @@ markers =
     testclient: tests updated to use the FastAPI TestClient object
     incompletedependency: marks tests that depend on an api that has not yet been completed
 testpaths =
-    tests/test_root.py
-    tests/test_session.py
-    tests/test_bulk_upload.py
+    tests/

src/api/tests/test.sh

@@ -1,14 +1,11 @@
 #!/usr/bin/env bash
-
 export DB_HOST=${DB_HOST:-localhost}
 export DB_USER=${DB_USER:-yacs}
 export DB_PASS=${DB_PASS:-easy_dev_pass}
 export DB_NAME=${DB_NAME:-yacs}
 export DB_PORT=${DB_PORT:-5432}
-
 SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
-
 export TEST_APP_DIR=$SCRIPT_DIR/../
 
 #pip3 install -r $TEST_APP_DIR/requirements.txt
-python3 -m pytest -s -m "testclient and not incompletedependency" "${SCRIPT_DIR}"
+python3 -m pytest -s -m "testclient and not incompletedependency" "${SCRIPT_DIR}"

src/api/tests/test_semester.py

@@ -1,7 +1,8 @@
+import pytest
+from fastapi.testclient import TestClient
 
-from .util import Client
-
-def test_semester(upload, client):
+@pytest.mark.testclient
+def test_semester(upload, client: TestClient):
     """
     semester endpoint should get all of the semesters in the upload
     in the case of our test data, we should be getting get 2 semesters: "SUMMER 2020" and "SPRING 2020"

src/api/tests/test_subsemester.py

@@ -1,7 +1,8 @@
-from .util import Client
+import pytest
+from fastapi.testclient import TestClient
 
-
-def test_subsemeseter_spring2020(client, upload):
+@pytest.mark.testclient
+def test_subsemeseter_spring2020(client: TestClient, upload):
     """
     when subsemester endpoint is given an input such as Spring 2020 
     it should only return data for that subsemester i.e. data where the parent semester name 
@@ -14,6 +15,7 @@ def test_subsemeseter_spring2020(client, upload):
     assert len(data) == 1
     assert data[0]["parent_semester_name"] == "SPRING 2020"
 
+@pytest.mark.testclient
 def test_subsemester_nosemester(client, upload):
     """   
     when no subsemester is taken as input the subsemester endpoint should return all of the subsemesters
@@ -25,6 +27,7 @@ def test_subsemester_nosemester(client, upload):
     data = r.json()
     assert len(data) == 4
 
+@pytest.mark.testclient
 def test_subsemester_invalid_semester(client):
     """
     invalid input to subsemester to subsemester endpoint such as "moon 2050"