Copy openapi java stuff

pantsbuild · alonsodomin · Dec 22, 2024 · Aug 10, 2024 · Aug 11, 2024 · Aug 17, 2024
commit e0dd5dd985c42d200c206c78f8bdb3f0d53c85d6
diff --git a/src/python/pants/backend/experimental/openapi/codegen/python/BUILD b/src/python/pants/backend/experimental/openapi/codegen/python/BUILD
@@ -0,0 +1,4 @@
+# Copyright 2022 Pants project contributors (see CONTRIBUTORS.md).
+# Licensed under the Apache License, Version 2.0 (see LICENSE).
+
+python_sources()
diff --git a/src/python/pants/backend/experimental/openapi/codegen/python/__init__.py b/src/python/pants/backend/experimental/openapi/codegen/python/__init__.py
diff --git a/src/python/pants/backend/experimental/openapi/codegen/python/register.py b/src/python/pants/backend/experimental/openapi/codegen/python/register.py
@@ -0,0 +1,18 @@
+# Copyright 2022 Pants project contributors (see CONTRIBUTORS.md).
+# Licensed under the Apache License, Version 2.0 (see LICENSE).
+
+from __future__ import annotations
+
+from pants.backend.experimental.python.register import rules as python_rules
+from pants.backend.experimental.python.register import target_types as python_target_types
+from pants.backend.experimental.openapi.register import rules as openapi_rules
+from pants.backend.experimental.openapi.register import target_types as openapi_target_types
+from pants.backend.openapi.codegen.python.rules import rules as openapi_python_codegen_rules
+
+
+def target_types():
+    return [*python_target_types(), *openapi_target_types()]
+
+
+def rules():
+    return [*python_rules(), *openapi_rules(), *openapi_python_codegen_rules()]
diff --git a/src/python/pants/backend/openapi/codegen/python/BUILD b/src/python/pants/backend/openapi/codegen/python/BUILD
@@ -0,0 +1,8 @@
+# Copyright 2022 Pants project contributors (see CONTRIBUTORS.md).
+# Licensed under the Apache License, Version 2.0 (see LICENSE).
+
+python_sources()
+
+python_tests(name="tests", dependencies=[":lockfiles"])
+
+resources(name="lockfiles", sources=["*.test.lock"])
diff --git a/src/python/pants/backend/openapi/codegen/python/extra_fields.py b/src/python/pants/backend/openapi/codegen/python/extra_fields.py
@@ -0,0 +1,35 @@
+from pants.backend.openapi.target_types import OpenApiDocumentGeneratorTarget, OpenApiDocumentTarget
+from pants.engine.target import BoolField, StringField
+from pants.jvm.target_types import PrefixedJvmJdkField, PrefixedJvmResolveField
+
+
+class OpenApiPythonModelPackageField(StringField):
+    alias = "python_model_package"
+    help = "Root package for generated model code"
+
+
+class OpenApiPythonApiPackageField(StringField):
+    alias = "python_api_package"
+    help = "Root package for generated API code"
+
+
+class OpenApiPythonSkipField(BoolField):
+    alias = "skip_python"
+    default = False
+    help = "If true, skips generation of Python sources from this target"
+
+
+def rules():
+    return [
+        OpenApiDocumentTarget.register_plugin_field(OpenApiPythonSkipField),
+        OpenApiDocumentTarget.register_plugin_field(OpenApiPythonModelPackageField),
+        OpenApiDocumentTarget.register_plugin_field(OpenApiPythonApiPackageField),
+        OpenApiDocumentGeneratorTarget.register_plugin_field(OpenApiPythonSkipField),
+        OpenApiDocumentGeneratorTarget.register_plugin_field(OpenApiPythonModelPackageField),
+        OpenApiDocumentGeneratorTarget.register_plugin_field(OpenApiPythonApiPackageField),
+        # Default Pants JVM fields
+        # OpenApiDocumentTarget.register_plugin_field(PrefixedJvmJdkField),
+        # OpenApiDocumentTarget.register_plugin_field(PrefixedJvmResolveField),
+        # OpenApiDocumentGeneratorTarget.register_plugin_field(PrefixedJvmJdkField),
+        # OpenApiDocumentGeneratorTarget.register_plugin_field(PrefixedJvmResolveField),
+    ]