abhishekbhakat/airflow-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

safe and unsafe are mutually exclusive

Browse Source
This commit is contained in:
abhishekbhakat
2025-02-25 06:10:28 +00:00
parent 355fb55bdb
commit 420b6fc68f
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/airflow_mcp_server/__init__.py
View File

@@ -22,12 +22,15 @@ def main(verbose: int, safe: bool, unsafe: bool) -> None:
logging.basicConfig(level=logging_level, stream=sys.stderr) logging.basicConfig(level=logging_level, stream=sys.stderr)
# Determine server mode with proper precedence
if safe and unsafe: if safe and unsafe:
# CLI argument validation
raise click.UsageError("Options --safe and --unsafe are mutually exclusive") raise click.UsageError("Options --safe and --unsafe are mutually exclusive")
elif safe:
if safe: # CLI argument for safe mode
asyncio.run(serve_safe()) asyncio.run(serve_safe())
else: # Default to unsafe mode else:
# Default to unsafe mode
asyncio.run(serve_unsafe()) asyncio.run(serve_unsafe())