Mini Shell

# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT

from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql import expression
from sqlalchemy.types import Numeric


class greatest(expression.FunctionElement):
    type = Numeric()
    name = "greatest"


class least(expression.FunctionElement):
    type = Numeric()
    name = "least"


@compiles(greatest)
def default_greatest(element, compiler, **kw):
    return compiler.visit_function(element)


@compiles(least)
def default_least(element, compiler, **kw):
    return compiler.visit_function(element)


@compiles(greatest, "sqlite")
def case_greatest(element, compiler, **kw):
    arg1, arg2 = list(element.clauses)
    return f"max({compiler.process(arg1)}, {compiler.process(arg2)})"


@compiles(least, "sqlite")
def case_least(element, compiler, **kw):
    arg1, arg2 = list(element.clauses)
    return f"min({compiler.process(arg1)}, {compiler.process(arg2)})"


class div(expression.FunctionElement):
    type = Numeric
    name = "div"


@compiles(div)
def default_div(element, compiler, **kw):
    arg1, arg2 = list(element.clauses)
    return f"{compiler.process(arg1)} / {compiler.process(arg2)}"


@compiles(div, "mysql")
def case_div(element, compiler, **kw):
    arg1, arg2 = list(element.clauses)
    return f"{compiler.process(arg1)} div {compiler.process(arg2)}"