def initialize_database()

in use-cases/rag-pipeline/alloy-db-setup/src/db_setup.py [0:0]


def initialize_database():
    """Initialize the database"""
    try:
        logger.info("Creating the database...")
        database.create(
            new_database_name=catalog_db_name,
        )
        logger.info("Database created successfully")

        logger.info("Enabling extensions...")
        database.enable_extensions(
            database_name=catalog_db_name,
        )
        logger.info("Extensions enabled successfully")

        logger.info("Granting permissions...")
        database.grant_permissions(
            database_name=catalog_db_name,
            read_users=db_read_users,
            write_users=db_write_users,
        )
        logger.info("Permissions granted successfully")
    except Exception:
        logger.exception(
            "An unhandled exception occurred during database initialization"
        )
        raise