

@compiles(CreateTable, 'mysql')
def add_partition_scheme(element, compiler, **kw):
    table = element.element
    partition_by = table.kwargs.get('mysql_partition_by', None)
    partitions = table.kwargs.get('mysql_partitions', None)
    ddl = compiler.visit_create_table(element, **kw)
    ddl = ddl.rstrip()
    if partition_by:
        ddl += ('\nPARTITION BY %s' % partition_by)
    if partitions:
        ddl += ('\nPARTITIONS %s' % partitions)
    return ddl

dataset/ETHPy150Open limodou/uliweb/uliweb/orm/patch.py add_partition_scheme/original