mirror of
https://github.com/zylon-ai/private-gpt.git
synced 2025-12-22 23:22:57 +01:00
16 lines
No EOL
667 B
Python
16 lines
No EOL
667 B
Python
from typing import Optional
|
|
|
|
from private_gpt.users.crud.base import CRUDBase
|
|
from private_gpt.users.models.company import Company
|
|
from private_gpt.users.schemas.company import CompanyCreate, CompanyUpdate
|
|
from sqlalchemy.orm import Session
|
|
|
|
|
|
class CRUDCompany(CRUDBase[Company, CompanyCreate, CompanyUpdate]):
|
|
def get_by_id(self, db: Session, *, id: str) -> Optional[Company]:
|
|
return db.query(self.model).filter(Company.id == id).first()
|
|
|
|
def get_by_company_name(self, db: Session, *, company_name: str) -> Optional[Company]:
|
|
return db.query(self.model).filter(Company.name == company_name).first()
|
|
|
|
company = CRUDCompany(Company) |