flask-ast-openapi 0.1.0__py3-none-any.whl

flask_ast_openapi/__init__.py

@@ -0,0 +1,13 @@
+"""Generate OpenAPI specifications from Flask source code using Python AST."""
+
+from .generator import FlaskASTOpenAPI, PathParameter, RouteDefinition
+from .swagger import create_swagger_blueprint
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "FlaskASTOpenAPI",
+    "PathParameter",
+    "RouteDefinition",
+    "create_swagger_blueprint",
+]

flask_ast_openapi/cli.py

@@ -0,0 +1,70 @@
+"""Command-line interface for flask-ast-openapi."""
+
+import argparse
+from pathlib import Path
+
+from .generator import FlaskASTOpenAPI
+
+
+def build_parser() -> argparse.ArgumentParser:
+    """Build the command-line argument parser."""
+
+    parser = argparse.ArgumentParser(
+        prog="flask-ast-openapi",
+        description=(
+            "Generate OpenAPI specifications from Flask source code."
+        ),
+    )
+
+    parser.add_argument(
+        "source_dir",
+        type=Path,
+        help="Path to the Flask source directory.",
+    )
+
+    parser.add_argument(
+        "--output",
+        "-o",
+        type=Path,
+        default=Path("openapi.json"),
+        help="Output file path.",
+    )
+
+    parser.add_argument(
+        "--title",
+        default="Flask API",
+        help="OpenAPI document title.",
+    )
+
+    parser.add_argument(
+        "--version",
+        default="1.0.0",
+        help="OpenAPI document version.",
+    )
+
+    return parser
+
+
+def main() -> None:
+    """Generate an OpenAPI specification from Flask source code."""
+
+    parser = build_parser()
+    args = parser.parse_args()
+
+    generator = FlaskASTOpenAPI(
+        args.source_dir
+    )
+
+    spec = generator.generate(
+        title=args.title,
+        version=args.version,
+    )
+
+    generator.write_json(
+        spec,
+        args.output,
+    )
+
+
+if __name__ == "__main__":
+    main()