39 lines
753 B
Python
39 lines
753 B
Python
|
from typing import Optional, List
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
|
||
|
|
from . import models
|
||
|
|
from . import schemas
|
||
|
|
|
||
|
|
|
||
|
|
def get_user(
|
||
|
|
db: Session,
|
||
|
|
user_id: int) -> Optional[models.User]:
|
||
|
|
|
||
|
|
return db \
|
||
|
|
.query(models.User) \
|
||
|
|
.filter(models.User.id == user_id) \
|
||
|
|
.first()
|
||
|
|
|
||
|
|
|
||
|
|
def get_users(
|
||
|
|
db: Session,
|
||
|
|
skip: int = 0,
|
||
|
|
limit: int = 100) -> List[Optional[models.User]]:
|
||
|
|
|
||
|
|
return db \
|
||
|
|
.query(models.User) \
|
||
|
|
.offset(skip) \
|
||
|
|
.limit(limit) \
|
||
|
|
.all()
|
||
|
|
|
||
|
|
|
||
|
|
def create_user(
|
||
|
|
db: Session,
|
||
|
|
user: schemas.User) -> models.User:
|
||
|
|
|
||
|
|
user_model = models.User(**user.dict())
|
||
|
|
db.add(user_model)
|
||
|
|
db.commit()
|
||
|
|
db.refresh(user_model)
|
||
|
|
return user_model
|