chore(python): support Stacks, Templates in generator (#41)

Author: bednar

openapi-generator/src/main/java/com/influxdb/codegen/InfluxCSharpGenerator.java

@@ -215,6 +215,12 @@ public boolean usesOwnAuthorizationSchema()
 		return false;
 	}
 
+	@Override
+	public boolean supportsStacksTemplates()
+	{
+		return false;
+	}
+
 	@NotNull
 	@Override
 	public Collection<String> getTypeAdapterImports()

openapi-generator/src/main/java/com/influxdb/codegen/InfluxGenerator.java

@@ -26,6 +26,8 @@ public interface InfluxGenerator
 
 	boolean usesOwnAuthorizationSchema();
 
+	boolean supportsStacksTemplates();
+
 	@Nonnull
 	Collection<String> getTypeAdapterImports();
 }

openapi-generator/src/main/java/com/influxdb/codegen/InfluxJavaGenerator.java

@@ -445,6 +445,12 @@ public boolean usesOwnAuthorizationSchema()
 		return false;
 	}
 
+	@Override
+	public boolean supportsStacksTemplates()
+	{
+		return false;
+	}
+
 	@NotNull
 	@Override
 	public Collection<String> getTypeAdapterImports()

openapi-generator/src/main/java/com/influxdb/codegen/InfluxPhpGenerator.java

@@ -203,6 +203,12 @@ public boolean usesOwnAuthorizationSchema()
 		return true;
 	}
 
+	@Override
+	public boolean supportsStacksTemplates()
+	{
+		return false;
+	}
+
 	@NotNull
 	@Override
 	public Collection<String> getTypeAdapterImports()

openapi-generator/src/main/java/com/influxdb/codegen/InfluxPythonGenerator.java

@@ -127,6 +127,22 @@ public String toApiFilename(String name) {
         return StringUtils.underscore(name) + "_service";
     }
 
+	@Override
+	public String toVarName(String name)
+	{
+		name = name.replaceAll("orgIDs", "orgIds");
+
+		return super.toVarName(name);
+	}
+
+	@Override
+	public String toModelFilename(String name)
+	{
+		name = name.replaceAll("orgIDs", "orgIds");
+
+		return super.toModelFilename(name);
+	}
+
 	@Override
 	public void processOpenAPI(OpenAPI openAPI) {
 
@@ -183,6 +199,12 @@ public boolean usesOwnAuthorizationSchema()
 		return true;
 	}
 
+	@Override
+	public boolean supportsStacksTemplates()
+	{
+		return true;
+	}
+
 	@NotNull
 	@Override
 	public Collection<String> getTypeAdapterImports()

openapi-generator/src/main/java/com/influxdb/codegen/PostProcessHelper.java

@@ -11,6 +11,7 @@
 import java.util.Map;
 import java.util.Set;
 import java.util.function.BiConsumer;
+import java.util.function.Consumer;
 import java.util.function.Function;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
@@ -187,17 +188,64 @@ void postProcessOpenAPI()
 		//
 		{
 			dropSchemas("Geo(.*)View(.*)");
+			dropSchemas("LatLon(.*)");
 			((ComposedSchema) openAPI.getComponents().getSchemas().get("ViewProperties"))
 					.getOneOf()
 					.removeIf(schema -> schema.get$ref().endsWith("GeoViewProperties"));
 		}
 
 		//
-		// Drop supports for Templates, Stack
+		// Templates, Stack API
 		//
 		{
-			dropSchemas("Stack(.*)|Template(.*)|LatLon(.*)");
-			dropPaths("/stacks(.*)|/templates(.*)");
+			// Fix naming for ViewLink and ViewsLink schema
+			{
+				Schema viewSchema = openAPI.getComponents().getSchemas().get("View");
+				Schema links = (Schema) viewSchema.getProperties().get("links");
+				((Schema) links.getProperties().get("self")).setTitle("ViewLinks");
+
+				Schema viewsSchema = openAPI.getComponents().getSchemas().get("Views");
+				Schema viewsSchemaLinks = (Schema) viewsSchema.getProperties().get("links");
+				((Schema) viewsSchemaLinks.getProperties().get("self")).setTitle("ViewLinks");
+			}
+
+			// Fix naming for 'new' and 'old' property of TemplateSummary.diff
+			{
+				Schema schema = openAPI.getComponents().getSchemas().get("TemplateSummary");
+				ObjectSchema diff = (ObjectSchema) schema.getProperties().get("diff");
+				for (Map.Entry<String, Schema> entry : diff.getProperties().entrySet())
+				{
+					String path = entry.getKey();
+					ArraySchema pathSchema = (ArraySchema) entry.getValue();
+					Schema newSchema = (Schema) pathSchema.getItems().getProperties().get("new");
+					if (newSchema != null && newSchema.get$ref() == null)
+					{
+						newSchema.setTitle("TemplateSummary_Diff_" + path + "_new_old");
+					}
+				}
+			}
+
+			// Fix schema name for UpdateStack.additionalResources
+			{
+				Schema stackPatch = openAPI.getPaths()
+						.get("/stacks/{stack_id}")
+						.getPatch()
+						.getRequestBody()
+						.getContent()
+						.get("application/json")
+						.getSchema();
+				Schema additionalResources = ((ArraySchema) stackPatch.getProperties().get("additionalResources"))
+						.getItems();
+				additionalResources.setTitle("PatchStackRequest_additionalResources");
+			}
+
+			//
+			// Drop if not supported
+			//
+			if (!generator.supportsStacksTemplates())
+			{
+				dropPaths("/stacks(.*)|/templates(.*)");
+			}
 		}
 
 		//
@@ -304,6 +352,24 @@ void postProcessOpenAPI()
 				parameter.setDescription(description.trim());
 			}
 		});
+		for (PathItem path : openAPI.getPaths().values())
+		{
+			for (Operation operation : path.readOperations())
+			{
+				List<Parameter> parameters = operation.getParameters();
+				if (parameters != null)
+				{
+					for (Parameter parameter : parameters)
+					{
+						String description = parameter.getDescription();
+						if (description != null)
+						{
+							parameter.setDescription(description.trim());
+						}
+					}
+				}
+			}
+		}
 	}
 
 	public void postProcessModel(final CodegenModel model, final Schema modelSchema, final Map<String, Schema> allDefinitions)