from .attribute_group_value_count import AttributeGroupValueCount


class AttributeGroupValuePercent(AttributeGroupValueCount):
    def get_query(self, *args, **kwargs) -> str:  # type: ignore

        """Using the outcome of AttirbuteGroupValueCount to get percentage values instead (full stacked)"""

        result_query = super().get_query(*args, **kwargs)

        start = """
            SELECT "group",
                   "label",
                   ROUND(100 * ("value" / sum("value") OVER (PARTITION BY "group")), 1) "value"
            FROM (
        """

        final_query = start + result_query
        final_query = final_query.replace(";", ") temp;")
        return final_query