JobSpy/api/v1/jobs/__init__.py

27 lines
714 B
Python
Raw Normal View History

2023-07-09 13:15:39 -07:00
from fastapi import APIRouter, Depends
2023-07-06 16:44:38 -07:00
2023-07-07 19:00:59 -07:00
from api.core.scrapers.indeed import IndeedScraper
2023-07-08 04:57:36 -07:00
from api.core.scrapers.ziprecruiter import ZipRecruiterScraper
2023-07-08 07:34:55 -07:00
from api.core.scrapers.linkedin import LinkedInScraper
2023-07-10 16:04:44 -07:00
from api.core.scrapers import ScraperInput, Site, JobResponse
2023-07-07 19:00:59 -07:00
2023-07-09 16:42:44 -07:00
router = APIRouter(prefix="/jobs")
2023-07-07 19:00:59 -07:00
SCRAPER_MAPPING = {
Site.LINKEDIN: LinkedInScraper,
Site.INDEED: IndeedScraper,
Site.ZIP_RECRUITER: ZipRecruiterScraper,
}
2023-07-07 19:00:59 -07:00
2023-07-08 07:34:55 -07:00
2023-07-10 16:04:44 -07:00
@router.get("/", response_model=JobResponse)
2023-07-10 14:14:05 -07:00
async def scrape_jobs(
2023-07-10 16:04:44 -07:00
scraper_input: ScraperInput
2023-07-10 14:14:05 -07:00
):
2023-07-10 16:04:44 -07:00
scraper_class = SCRAPER_MAPPING[scraper_input.site_type]
scraper = scraper_class()
2023-07-08 07:34:55 -07:00
job_response = scraper.scrape(scraper_input)
2023-07-07 19:00:59 -07:00
return job_response