1#@!#!123s

: / proc / thread-self / root / opt / alt / python37 / lib / python3.7 / site-packages / lvestats / lib /
Filename : db_functions.py
back
# 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 __future__ import absolute_import
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 "max(%s, %s)" % (
        compiler.process(arg1),
        compiler.process(arg2),
    )


@compiles(least, "sqlite")
def case_least(element, compiler, **kw):
    arg1, arg2 = list(element.clauses)
    return "min(%s, %s)" % (
        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 "%s / %s" % (
        compiler.process(arg1),
        compiler.process(arg2),
    )


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