processed = super.postProcessAllModels(objs);
+ postProcessUpdateImports(processed);
+ return processed;
+ }
+
+ /**
+ * Update/clean up model imports
+ *
+ * append '._" if the import is a Enum class, otherwise
+ * remove model imports to avoid warnings for importing class in the same
+ * package in Scala
+ *
+ * @param models processed models to be further processed
+ */
+ @SuppressWarnings("unchecked")
+ private void postProcessUpdateImports(final Map models) {
+ final String prefix = modelPackage() + ".";
+
+ enumRefs = getEnumRefs(models);
+
+ for (String openAPIName : models.keySet()) {
+ CodegenModel model = ModelUtils.getModelByName(openAPIName, models);
+ if (model == null) {
+ LOGGER.warn(
+ "Expected to retrieve model {} by name, but no model was found. Check your -Dmodels inclusions.",
+ openAPIName);
+ continue;
+ }
+
+ ModelsMap objs = models.get(openAPIName);
+ List