# 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)}"