galaxy_schemadisp.py

import os
import sys

from db_shell import *  # noqa
from sqlalchemy import MetaData
from sqlalchemy.orm import class_mapper

try:
    from sqlalchemy_schemadisplay import (
        create_schema_graph,
        create_uml_graph,
    )
except ImportError:
    print("please install sqlalchemy_schemadisplay to use this script (pip install sqlalchemy_schemadisplay)")
    raise


gxy_root = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
sys.path.insert(1, os.path.abspath(os.path.join(gxy_root, "lib")))

from galaxy import model

if __name__ == "__main__":
    gxy_root = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
    sqlitedb = os.path.join(gxy_root, "database/universe.sqlite")
    # Try to build a representation of what's in the sqlite database
    if os.path.exists(sqlitedb):
        graph = create_schema_graph(
            metadata=MetaData("sqlite:///" + sqlitedb),
            show_datatypes=False,
            show_indexes=False,
            rankdir="LR",
            concentrate=False,
        )
        print(f"Writing galaxy_universe.png, built from {sqlitedb}")
        graph.write_png("galaxy_universe.png")
    else:
        print(f"No sqlitedb available at {sqlitedb}, skipping rendering")

    # Build UML graph from loaded mapper
    mappers = []
    for attr in dir(model):
        if attr[0] == "_":
            continue
        try:
            cls = getattr(model, attr)
            mappers.append(class_mapper(cls))
        except Exception:
            pass

    graph = create_uml_graph(
        mappers,
        show_operations=False,
    )
    print("Writing galaxy_uml.png")
    graph.write_png("galaxy_uml.png")  # write out the file