#!/usr/bin/python3
"""
Generate an image of the database schema
"""
from sqlalchemy_schemadisplay import create_schema_graph

from bodhi.server import models


def write_graph(filename):
    graph = create_schema_graph(
        metadata=models.metadata,
        show_datatypes=False,
        show_indexes=False,
        rankdir='LR',
        concentrate=False
    )
    graph.write_png(filename)


if __name__ == '__main__':
    write_graph('generated/database.png')
