mcp-airflow-api 0.0.2__tar.gz

mcp_airflow_api-0.0.2/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: mcp-airflow-api
+Version: 0.0.2
+Summary: Model Context Protocol (MCP) server for Apache Airflow API integration. Provides comprehensive tools for managing Airflow clusters including service operations, configuration management, status monitoring, and request tracking.
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.116.1
+Requires-Dist: python-dotenv>=1.1.1
+Requires-Dist: requests>=2.32.4
+Requires-Dist: uvicorn>=0.35.0

mcp_airflow_api-0.0.2/pyproject.toml

@@ -0,0 +1,12 @@
+[project]
+name = "mcp-airflow-api"
+version = "0.0.2"
+description = "Model Context Protocol (MCP) server for Apache Airflow API integration. Provides comprehensive tools for managing Airflow clusters including service operations, configuration management, status monitoring, and request tracking."
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "fastapi>=0.116.1",
+    "python-dotenv>=1.1.1",
+    "requests>=2.32.4",
+    "uvicorn>=0.35.0",
+]

mcp_airflow_api-0.0.2/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

mcp_airflow_api-0.0.2/src/mcp_airflow_api/airflow_api.py

@@ -0,0 +1,47 @@
+"""
+Main feature definitions for Airflow MCP.
+"""
+import os
+from fastapi import APIRouter, HTTPException
+from dotenv import load_dotenv
+import requests
+from .functions import airflow_request
+
+load_dotenv(dotenv_path=os.getenv("MCP_CONFIG", "config"))
+
+AIRFLOW_API_URL = os.getenv("AIRFLOW_API_URL")
+AIRFLOW_API_TOKEN = os.getenv("AIRFLOW_API_TOKEN")
+
+router = APIRouter()
+
+# MCP tool: DAG 목록 조회
+@router.get("/dags", tags=["airflow"])
+def list_dags():
+    resp = airflow_request("GET", f"{AIRFLOW_API_URL}/dags")
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+
+# MCP tool: 실행 중인 DAG 조회
+@router.get("/dags/running", tags=["airflow"])
+def running_dags():
+    resp = airflow_request("GET", f"{AIRFLOW_API_URL}/dags?state=running")
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+
+# MCP tool: 최근 실패한 DAG 조회
+@router.get("/dags/failed", tags=["airflow"])
+def failed_dags():
+    resp = airflow_request("GET", f"{AIRFLOW_API_URL}/dags?state=failed")
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+
+# MCP tool: DAG 실행 트리거
+@router.post("/dags/{dag_id}/trigger", tags=["airflow"])
+def trigger_dag(dag_id: str):
+    resp = airflow_request("POST", f"{AIRFLOW_API_URL}/dags/{dag_id}/dagRuns")
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()

mcp_airflow_api-0.0.2/src/mcp_airflow_api/functions.py

@@ -0,0 +1,12 @@
+"""
+Airflow MCP auxiliary utility functions definition file
+"""
+import os
+import requests
+
+def airflow_request(method, url, **kwargs):
+    user = os.getenv("AIRFLOW_API_USER")
+    password = os.getenv("AIRFLOW_API_PASSWORD")
+    headers = kwargs.pop("headers", {})
+    auth = (user, password) if user and password else None
+    return requests.request(method, url, headers=headers, auth=auth, **kwargs)

mcp_airflow_api-0.0.2/src/mcp_airflow_api.egg-info/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: mcp-airflow-api
+Version: 0.0.2
+Summary: Model Context Protocol (MCP) server for Apache Airflow API integration. Provides comprehensive tools for managing Airflow clusters including service operations, configuration management, status monitoring, and request tracking.
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.116.1
+Requires-Dist: python-dotenv>=1.1.1
+Requires-Dist: requests>=2.32.4
+Requires-Dist: uvicorn>=0.35.0

mcp_airflow_api-0.0.2/src/mcp_airflow_api.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+README.md
+pyproject.toml
+src/mcp_airflow_api/airflow_api.py
+src/mcp_airflow_api/functions.py
+src/mcp_airflow_api.egg-info/PKG-INFO
+src/mcp_airflow_api.egg-info/SOURCES.txt
+src/mcp_airflow_api.egg-info/dependency_links.txt
+src/mcp_airflow_api.egg-info/requires.txt
+src/mcp_airflow_api.egg-info/top_level.txt

mcp_airflow_api-0.0.2/src/mcp_airflow_api.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

mcp_airflow_api-0.0.2/src/mcp_airflow_api.egg-info/requires.txt

@@ -0,0 +1,4 @@
+fastapi>=0.116.1
+python-dotenv>=1.1.1
+requests>=2.32.4
+uvicorn>=0.35.0

mcp_airflow_api-0.0.2/src/mcp_airflow_api.egg-info/top_level.txt

@@ -0,0 +1 @@
+mcp_airflow_api