33 lines
1,019 B
Python
33 lines
1,019 B
Python
from sqlalchemy.orm import joinedload, selectinload
|
|
from .models import User, Room, Inventory, WorkLog
|
|
|
|
def eager_load_user_relationships(query):
|
|
return query.options(
|
|
joinedload(User.supervisor),
|
|
joinedload(User.location).joinedload(Room.room_function)
|
|
)
|
|
|
|
def eager_load_inventory_relationships(query):
|
|
return query.options(
|
|
joinedload(Inventory.owner),
|
|
joinedload(Inventory.brand),
|
|
joinedload(Inventory.item),
|
|
joinedload(Inventory.location).joinedload(Room.room_function)
|
|
)
|
|
|
|
def eager_load_room_relationships(query):
|
|
return query.options(
|
|
joinedload(Room.area),
|
|
joinedload(Room.room_function),
|
|
selectinload(Room.inventory),
|
|
selectinload(Room.users)
|
|
)
|
|
|
|
def eager_load_worklog_relationships(query):
|
|
return query.options(
|
|
joinedload(WorkLog.contact),
|
|
joinedload(WorkLog.work_item)
|
|
)
|
|
|
|
def chunk_list(lst, chunk_size):
|
|
return [lst[i:i + chunk_size] for i in range(0, len(lst), chunk_size)]
|