dba_databaseSpace
Retrieve database space allocations for a specific database or all databases to monitor storage usage and manage capacity in Teradata systems.
Instructions
Get database space if database name is provided, otherwise get all databases space allocations.
Arguments: database_name - database name
Returns: ResponseType: formatted response with query results + metadata
Input Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| database_name | Yes |
Implementation Reference
- Handler function implementing the dba_databaseSpace tool. Queries DBC.DiskSpaceV to compute space metrics (allocated, used, free in GB, percent used) for a specific database or all databases.def handle_dba_databaseSpace(conn: TeradataConnection, database_name: str | None | None, *args, **kwargs): """ Get database space if database name is provided, otherwise get all databases space allocations. Arguments: database_name - database name Returns: ResponseType: formatted response with query results + metadata """ logger.debug(f"Tool: handle_dba_databaseSpace: Args: database_name: {database_name}") database_name_filter = f"AND objectdatabasename = '{database_name}'" if database_name else "" with conn.cursor() as cur: if not database_name: logger.debug("No database name provided, returning all databases and space information.") rows = cur.execute(""" SELECT DatabaseName, CAST(SUM(MaxPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceAllocated_GB, CAST(SUM(CurrentPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceUsed_GB, CAST((SUM(MaxPerm) - SUM(CurrentPerm))/1024/1024/1024 AS DECIMAL(10,2)) AS FreeSpace_GB, CAST((SUM(CurrentPerm) * 100.0 / NULLIF(SUM(MaxPerm),0)) AS DECIMAL(10,2)) AS PercentUsed FROM DBC.DiskSpaceV WHERE MaxPerm > 0 GROUP BY 1 ORDER BY 5 DESC; """) else: logger.debug(f"Database name: {database_name}, returning space information for this database.") rows = cur.execute(f""" SELECT DatabaseName, CAST(SUM(MaxPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceAllocated_GB, CAST(SUM(CurrentPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceUsed_GB, CAST((SUM(MaxPerm) - SUM(CurrentPerm))/1024/1024/1024 AS DECIMAL(10,2)) AS FreeSpace_GB, CAST((SUM(CurrentPerm) * 100.0 / NULLIF(SUM(MaxPerm),0)) AS DECIMAL(10,2)) AS PercentUsed FROM DBC.DiskSpaceV WHERE MaxPerm > 0 AND DatabaseName = '{database_name}' GROUP BY 1; """) data = rows_to_json(cur.description, rows.fetchall()) metadata = { "tool_name": "dba_databaseSpace", "database_name": database_name, "total_databases": len(data) } logger.debug(f"Tool: handle_dba_databaseSpace: metadata: {metadata}") return create_response(data, metadata)
- src/teradata_mcp_server/app.py:461-486 (registration)Dynamic registration code that discovers handle_* functions via module_loader, wraps them for MCP (injecting DB connection, QueryBand, etc.), and registers as tools if matching profile patterns. Registers handle_dba_databaseSpace as dba_databaseSpace.# Register code tools via module loader module_loader = td.initialize_module_loader(config) if module_loader: all_functions = module_loader.get_all_functions() for name, func in all_functions.items(): if not (inspect.isfunction(func) and name.startswith("handle_")): continue tool_name = name[len("handle_"):] if not any(re.match(p, tool_name) for p in config.get('tool', [])): continue # Skip template tools (used for developer reference only) if tool_name.startswith("tmpl_"): logger.debug(f"Skipping template tool: {tool_name}") continue # Skip BAR tools if BAR functionality is disabled if tool_name.startswith("bar_") and not enableBAR: logger.info(f"Skipping BAR tool: {tool_name} (BAR functionality disabled)") continue # Skip chat completion tools if chat completion functionality is disabled if tool_name.startswith("chat_") and not enableChat: logger.info(f"Skipping chat completion tool: {tool_name} (chat completion functionality disabled)") continue wrapped = make_tool_wrapper(func) mcp.tool(name=tool_name, description=wrapped.__doc__)(wrapped) logger.info(f"Created tool: {tool_name}") logger.debug(f"Tool Docstring: {wrapped.__doc__}")
- src/teradata_mcp_server/tools/dba/__init__.py:1-2 (registration)Imports dba_tools.py into the dba module, making its functions available for dynamic discovery and loading by the module_loader.from .dba_resources import * from .dba_tools import *
- Imports and logger setup used by the dba_databaseSpace handler.import logging from teradatasql import TeradataConnection from teradata_mcp_server.tools.utils import create_response, rows_to_json logger = logging.getLogger("teradata_mcp_server")