Function generate_database_security_report

web/pgadmin/llm/__init__.py:1302–1360

Generate a security report for the specified database. Uses the multi-stage pipeline to analyze database security.

(sid, did)

)
@pga_login_required
def generate_database_security_report(sid, did):
    """
    Generate a security report for the specified database.
    Uses the multi-stage pipeline to analyze database security.
    """
    from pgadmin.llm.utils import is_llm_enabled
    from pgadmin.llm.reports.generator import generate_report_sync
    from pgadmin.utils.driver import get_driver

    # Check if LLM is configured
    if not is_llm_enabled():
        return make_json_response(
            success=0,
            errormsg=gettext(
                'LLM is not configured. Please configure an LLM provider '
                'in Preferences > AI.'
            )
        )

    # Get database connection
    try:
        driver = get_driver(config.PG_DEFAULT_DRIVER)
        manager = driver.connection_manager(sid)
        conn = manager.connection(did=did)

        if not conn.connected():
            return make_json_response(
                success=0,
                errormsg=gettext('Database is not connected.')
            )

        # Generate report using pipeline
        context = {
            'database_name': conn.db
        }
        success, result = generate_report_sync(
            report_type='security',
            scope='database',
            conn=conn,
            manager=manager,
            context=context
        )

        if success:
            return make_json_response(
                success=1,
                data={'report': result}
            )
        else:
            return make_json_response(
                success=0,
                errormsg=result
            )

    except Exception as e:
        return make_json_response(
            success=0,
            errormsg=gettext('Failed to generate report: ') + str(e)