from ..jobs import *
from ..formatters import OutputFormat
from typing import List, Dict, Optional


class StatusException(Exception):
    def __init__(self, status_code: int):
        self.status_code = status_code


class Site(Enum):
    LINKEDIN = "linkedin"
    INDEED = "indeed"
    ZIP_RECRUITER = "zip_recruiter"


class ScraperInput(BaseModel):
    site_type: List[Site]
    search_term: str
    output_format: OutputFormat = OutputFormat.JSON

    location: str = None
    distance: int = None
    is_remote: bool = False
    job_type: JobType = None
    easy_apply: bool = None  # linkedin

    results_wanted: int = 15


class ScraperResponse(BaseModel):
    linkedin: Optional[JobResponse]
    indeed: Optional[JobResponse]
    zip_recruiter: Optional[JobResponse]


class Scraper:
    def __init__(self, site: Site, url: str):
        self.site = site
        self.url = url

    def scrape(self, scraper_input: ScraperInput) -> JobResponse:
        ...