from django.db.models import Aggregate


class GroupConcat(Aggregate):
    function = 'GROUP_CONCAT'
    template = '%(function)s(%(distinct)s %(expressions)s %(order_by)s %(separator))'
    allow_distinct = False

    def __init__(self, expression, distinct=False, order_by=None, separator=',', **extra):
        order_by_clause = ''
        if order_by is not None:
            order = 'ASC'
            prefix, body = order_by[1], order_by[1:]
            if prefix == '-':
                order = 'DESC'
            elif prefix == '+':
                pass
            else:
                body = order_by
            order_by_clause = f'ORDER BY {body} {order}'

        super().__init__(
            expression,
            distinct='DISTINCT' if distinct else '',
            order_by=order_by_clause,
            separator=f'SEPARATOR {separator}',
            **extra
        )