mirror of https://github.com/Bunsly/JobSpy
70 lines
1.4 KiB
Python
70 lines
1.4 KiB
Python
from typing import Union
|
|
from datetime import date
|
|
from enum import Enum
|
|
|
|
from pydantic import BaseModel, validator
|
|
|
|
|
|
class JobType(Enum):
|
|
FULL_TIME = "fulltime"
|
|
PART_TIME = "parttime"
|
|
CONTRACT = "contract"
|
|
TEMPORARY = "temporary"
|
|
INTERNSHIP = "internship"
|
|
|
|
PER_DIEM = "perdiem"
|
|
NIGHTS = "nights"
|
|
OTHER = "other"
|
|
SUMMER = "summer"
|
|
|
|
|
|
class Location(BaseModel):
|
|
country: str = "USA"
|
|
city: str = None
|
|
state: str = None
|
|
|
|
|
|
class CompensationInterval(Enum):
|
|
YEARLY = "yearly"
|
|
MONTHLY = "monthly"
|
|
WEEKLY = "weekly"
|
|
DAILY = "daily"
|
|
HOURLY = "hourly"
|
|
|
|
|
|
class Compensation(BaseModel):
|
|
interval: CompensationInterval
|
|
min_amount: int
|
|
max_amount: int
|
|
currency: str = "USD"
|
|
|
|
|
|
class JobPost(BaseModel):
|
|
title: str
|
|
company_name: str
|
|
job_url: str
|
|
location: Location
|
|
|
|
description: str = None
|
|
job_type: JobType = None
|
|
compensation: Compensation = None
|
|
# why is 08-28-2023 a validiation error for type date? how do I fix this?
|
|
date_posted: date = None
|
|
|
|
|
|
class JobResponse(BaseModel):
|
|
success: bool
|
|
error: str = None
|
|
|
|
total_results: int = None
|
|
|
|
jobs: list[JobPost] = []
|
|
|
|
returned_results: int = None
|
|
|
|
@validator("returned_results", pre=True, always=True)
|
|
def set_returned_results(cls, v, values):
|
|
if v is None and values.get("jobs"):
|
|
return len(values["jobs"])
|
|
return v
|