import typing from hlds.definitions import location_definitions from hlds.models import Dish, Location from models import Order, OrderItem, User from sqlalchemy.sql import desc, func class FatModel: @classmethod def all(cls): return cls.query.all() @classmethod def amount(cls): return cls.query.count() class FatLocation(Location, FatModel): @classmethod def all(cls): return location_definitions @classmethod def amount(cls): return len(location_definitions) class FatOrder(Order, FatModel): # It's hard to add the unique user constraint, # as DISTINCT seems to apply after a GROUP BY and aggregate # So DISTINCT ... count(user_id) ... will count all users, # even if they get reduced by the disctinct afterwards. @classmethod def items_per_order(cls): return ( Order.query.join(OrderItem) .group_by(Order.id) .with_entities(Order.id, func.count(OrderItem.user_id).label("total")) ) class FatUser(User, FatModel): pass class FatOrderItem(OrderItem, FatModel): pass