Merge pull request #2 from colin-combe/python3

speedup json encoding (?) using orjson lib

Pipfile

@@ -18,6 +18,7 @@ psycopg2-binary = "*"
 sqlalchemy = "2.0.21"
 sqlalchemy-utils = "*"
 obonet = "*"
+orjson = "*"
 #Security modules
 python-multipart = "*"
 python-jose = "*"

app/routes/pdbdev.py

@@ -1,7 +1,9 @@
 import json
 
 import psycopg2
-from fastapi import APIRouter, Depends, Path
+from fastapi import APIRouter, Depends, Path, Response
+import orjson
+from fastapi import APIRouter, Depends
 from psycopg2.extras import RealDictCursor
 import logging
 from sqlalchemy import text
@@ -165,7 +167,7 @@ async def get_psm_level_residue_pairs(project_id: Annotated[str, Path(...,
         if conn is not None:
             conn.close()
             print('Database connection closed.')
-        return data
+        return Response(orjson.dumps(data), media_type='application/json')
 
 #
 # @pdb_dev_router.get('/projects/{project_id}/residue-pairs/reported')

app/routes/xiview.py

@@ -5,7 +5,8 @@
 
 import fastapi
 import psycopg2
-from fastapi import APIRouter, Depends, Request
+from fastapi import APIRouter, Depends, Request, Response
+import orjson
 from psycopg2 import sql
 from psycopg2.extras import RealDictCursor
 from sqlalchemy.orm import session, Session
@@ -117,7 +118,7 @@ async def get_data_object(ids, pxid):
             logger.debug('Database connection closed.')
         if error is not None:
             raise error
-        return data
+        return Response(orjson.dumps(data), media_type='application/json')
 
 
 async def get_pride_api_info(cur, pxid):