feat: add new folder
This commit is contained in:
58
gig-poc/apps/api/app/repositories/job_repository.py
Normal file
58
gig-poc/apps/api/app/repositories/job_repository.py
Normal file
@@ -0,0 +1,58 @@
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import Session, selectinload
|
||||
|
||||
from app.domain.models import Job, JobSkill
|
||||
from app.domain.schemas import JobCard
|
||||
|
||||
|
||||
class JobRepository:
|
||||
def __init__(self, db: Session):
|
||||
self.db = db
|
||||
|
||||
def upsert(self, job_card: JobCard) -> Job:
|
||||
instance = self.db.get(Job, job_card.job_id)
|
||||
if instance is None:
|
||||
instance = Job(id=job_card.job_id)
|
||||
self.db.add(instance)
|
||||
|
||||
instance.title = job_card.title
|
||||
instance.category = job_card.category
|
||||
instance.description = job_card.description
|
||||
instance.city = job_card.city
|
||||
instance.region = job_card.region
|
||||
instance.location_detail = job_card.location_detail
|
||||
instance.start_time = job_card.start_time
|
||||
instance.duration_hours = job_card.duration_hours
|
||||
instance.headcount = job_card.headcount
|
||||
instance.salary_type = job_card.salary.type.value
|
||||
instance.salary_amount = job_card.salary.amount
|
||||
instance.salary_currency = job_card.salary.currency
|
||||
instance.work_mode = job_card.work_mode
|
||||
instance.tags_json = job_card.tags
|
||||
instance.confidence = job_card.confidence
|
||||
instance.skills.clear()
|
||||
instance.skills.extend(
|
||||
[
|
||||
JobSkill(skill_name=skill_name, weight=1.0, is_required=index < 2)
|
||||
for index, skill_name in enumerate(job_card.skills)
|
||||
]
|
||||
)
|
||||
self.db.commit()
|
||||
self.db.refresh(instance)
|
||||
return instance
|
||||
|
||||
def list(self, limit: int = 100) -> list[Job]:
|
||||
stmt = select(Job).options(selectinload(Job.skills)).order_by(Job.created_at.desc()).limit(limit)
|
||||
return list(self.db.scalars(stmt))
|
||||
|
||||
def get(self, job_id: str) -> Job | None:
|
||||
stmt = select(Job).options(selectinload(Job.skills)).where(Job.id == job_id)
|
||||
return self.db.scalar(stmt)
|
||||
|
||||
def get_many(self, ids: list[str]) -> list[Job]:
|
||||
if not ids:
|
||||
return []
|
||||
stmt = select(Job).options(selectinload(Job.skills)).where(Job.id.in_(ids))
|
||||
result = list(self.db.scalars(stmt))
|
||||
order = {item_id: index for index, item_id in enumerate(ids)}
|
||||
return sorted(result, key=lambda item: order[item.id])
|
||||
Reference in New Issue
Block a user