Added endpoint to return lei data from list of leis

src/entities/repos/institutions_repo.py

@@ -13,7 +13,7 @@
 
 
 async def get_institutions(
-    session: AsyncSession, domain: str = "", page: int = 0, count: int = 100
+    session: AsyncSession, domain: str = "", page: int = 0, count: int = 100, leis: List[str] = []
 ) -> List[FinancialInstitutionDao]:
     async with session.begin():
         stmt = (
@@ -22,6 +22,10 @@ async def get_institutions(
             .limit(count)
             .offset(page * count)
         )
+        if leis:
+            stmt = stmt.join(FinancialInstitutionDomainDao,
+                FinancialInstitutionDao.lei == FinancialInstitutionDomainDao.lei,
+            ).filter(FinancialInstitutionDao.lei.in_(leis))
         if d := domain.strip():
             search = "%{}%".format(d)
             stmt = stmt.join(

src/routers/institutions.py

@@ -1,7 +1,8 @@
-from fastapi import Depends, Request, HTTPException
+from fastapi import Depends, Request, HTTPException, Query
 from http import HTTPStatus
 from oauth2 import oauth2_admin
 from util import Router
+from util.parsers import parse_leis
 from typing import List, Tuple
 from entities.engine import get_session
 from entities.repos import institutions_repo as repo
@@ -24,9 +25,10 @@ async def get_institutions(
     domain: str = "",
     page: int = 0,
     count: int = 100,
+    leis: List[str] = Depends(parse_leis),
     session: AsyncSession = Depends(get_session),
 ):
-    return await repo.get_institutions(session, domain, page, count)
+    return await repo.get_institutions(session, domain, page, count, leis)
 
 
 @router.post("/", response_model=Tuple[str, FinancialInstitutionDto])

src/util/parsers.py

@@ -0,0 +1,16 @@
+from typing import List, Optional
+
+from fastapi import Query
+
+
+def parse_leis(leis: List[str] = Query(None)) -> Optional[List]:
+    """
+    Parses leis from list of one string to list of multiple distinct lei strings
+    Returns empty list when nothing is passed in
+    Ex: ['lei1,lei2'] -> ['lei1', 'lei2']
+    """
+
+    if leis is None or len(leis) == 0 or leis[0] == "":
+        return []
+    else:
+        return leis[0].split(',')

tests/entities/repos/test_institutions_repo.py

@@ -37,6 +37,14 @@ async def test_get_institutions_by_domain_not_existing(self, session: AsyncSessi
         res = await repo.get_institutions(session, domain="testing.bank")
         assert len(res) == 0
 
+    async def test_get_institutions_by_lei_list(self, session: AsyncSession):
+        res = await repo.get_institutions(session, leis=['TESTBANK123'])
+        assert len(res) == 1
+
+    async def test_get_institutions_by_lei_list_item_not_existing(self, session: AsyncSession):
+        res = await repo.get_institutions(session, leis=["NOTTESTBANK"])
+        assert len(res) == 0
+
     async def test_add_institution(self, session: AsyncSession):
         await repo.upsert_institution(
             session, FinancialInstitutionDao(name="New Bank 123", lei="NEWBANK123")