adding multiple search sites

.vscode/launch.json

@@ -0,0 +1,16 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python: Module",
+            "type": "python",
+            "request": "launch",
+            "module": "uvicorn",
+            "args": ["main:app","--reload"]
+        }
+        
+    ]
+}

api/core/scrapers/__init__.py

@@ -1,4 +1,5 @@
 from ..jobs import *
+from typing import List
 
 
 class StatusException(Exception):
@@ -13,7 +14,7 @@ class Site(Enum):
 
 
 class ScraperInput(BaseModel):
-    site_type: Site
+    site_type: List[Site]
     search_term: str
 
     location: str = None

api/v1/jobs/__init__.py

@@ -4,6 +4,7 @@ from api.core.scrapers.indeed import IndeedScraper
 from api.core.scrapers.ziprecruiter import ZipRecruiterScraper
 from api.core.scrapers.linkedin import LinkedInScraper
 from api.core.scrapers import ScraperInput, Site, JobResponse
+from typing import List
 
 router = APIRouter(prefix="/jobs", tags=["jobs"])
 
@@ -14,11 +15,13 @@ SCRAPER_MAPPING = {
 }
 
 
-@router.post("/", response_model=JobResponse)
+@router.post("/", response_model=List[JobResponse])
-async def scrape_jobs(scraper_input: ScraperInput):
+async def scrape_jobs(scraper_input: ScraperInput) -> JobResponse:
-    scraper_class = SCRAPER_MAPPING[scraper_input.site_type]
+    resp = []
-    scraper = scraper_class()
+    for site in scraper_input.site_type:
+        scraper_class = SCRAPER_MAPPING[site]
+        scraper = scraper_class()
+        job_response = scraper.scrape(scraper_input)
+        resp.append(job_response)
 
-    job_response = scraper.scrape(scraper_input)
+    return resp
-
-    return job_response