Function render_reference

pydbml/renderer/sql/default/reference.py:61–82

Returns SQL of the reference: ALTER TABLE "orders" ADD FOREIGN KEY ("customer_id") REFERENCES "customers ("id");

(model: Reference)

@DefaultSQLRenderer.renderer_for(Reference)
def render_reference(model: Reference) -> str:
    '''
    Returns SQL of the reference:

    ALTER TABLE "orders" ADD FOREIGN KEY ("customer_id") REFERENCES "customers ("id");

    '''
    validate_for_sql(model)

    if model.type == MANY_TO_MANY:
        return generate_many_to_many_sql(model)

    result = ''
    func = generate_inline_sql if model.inline else generate_not_inline_sql
    if model.type in (MANY_TO_ONE, ONE_TO_ONE):
        result = func(model=model, source_col=model.col1, ref_col=model.col2)
    elif model.type == ONE_TO_MANY:
        result = func(model=model, source_col=model.col2, ref_col=model.col1)

    c = f'CONSTRAINT "{model.name}" ' if model.name else ''

    return result.format(c=c)