Function render_enum

pydbml/renderer/sql/default/enum.py:10–27

Returns SQL for enum type: CREATE TYPE "job_status" AS ENUM ( 'created', 'running', 'done', 'failure', );

(model: Enum)

@DefaultSQLRenderer.renderer_for(Enum)
def render_enum(model: Enum) -> str:
    '''
    Returns SQL for enum type:

    CREATE TYPE "job_status" AS ENUM (
      'created',
      'running',
      'done',
      'failure',
    );
    '''

    result = comment_to_sql(model.comment) if model.comment else ''
    result += f'CREATE TYPE {get_full_name_for_sql(model)} AS ENUM (\n'
    enum_body = '\n'.join(f'{indent(DefaultSQLRenderer.render(i), "  ")}' for i in model.items)
    result += enum_body.rstrip(',')
    result += '\n);'
    return result


@DefaultSQLRenderer.renderer_for(EnumItem)