Function generate_security_report

web/pgadmin/llm/__init__.py:1051–1107

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

(sid)

)
@pga_login_required
def generate_security_report(sid):
    """
    Generate a security report for the specified server.
    Uses the multi-stage pipeline to analyze server configuration.
    """
    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()

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

        # Generate report using pipeline
        context = {}
        success, result = generate_report_sync(
            report_type='security',
            scope='server',
            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)
        )