Source code for anyblok_wms_base.quantity.wms

# -*- coding: utf-8 -*-
# This file is a part of the AnyBlok / WMS Base project
#
#    Copyright (C) 2018 Georges Racinet <gracinet@anybox.fr>
#
# This Source Code Form is subject to the terms of the Mozilla Public License,
# v. 2.0. If a copy of the MPL was not distributed with this file,You can
# obtain one at http://mozilla.org/MPL/2.0/.
from sqlalchemy import func
from anyblok import Declarations


[docs]@Declarations.register(Declarations.Model)
class Wms:
    """Override to replace quantity counting by summation"""

[docs]    @classmethod
    def base_quantity_query(cls):
        """Return a base query fit for summing PhysObj.quantity."""
        PhysObj = cls.registry.Wms.PhysObj
        Avatar = PhysObj.Avatar
        # TODO distinguish quantity on Avatars from those on PhysObj?
        return Avatar.query(func.sum(PhysObj.quantity)).join(Avatar.obj)