From 12a85bb09d9e8a68bb6c39929d11ec8972b18321 Mon Sep 17 00:00:00 2001
From: Logan Patino <loganpatino10@gmail.com>
Date: Sun, 15 Oct 2017 23:34:31 -0400
Subject: [PATCH] Add enum support for flask (#6684)

---
 .../codegen/languages/FlaskConnexionCodegen.java       |  6 ++++++
 .../src/main/resources/flaskConnexion/model.mustache   | 10 ++++++++++
 2 files changed, 16 insertions(+)

diff --git a/modules/swagger-codegen/src/main/java/io/swagger/codegen/languages/FlaskConnexionCodegen.java b/modules/swagger-codegen/src/main/java/io/swagger/codegen/languages/FlaskConnexionCodegen.java
index efaee9eb05f..adab7d64d16 100644
--- a/modules/swagger-codegen/src/main/java/io/swagger/codegen/languages/FlaskConnexionCodegen.java
+++ b/modules/swagger-codegen/src/main/java/io/swagger/codegen/languages/FlaskConnexionCodegen.java
@@ -662,6 +662,12 @@ public class FlaskConnexionCodegen extends DefaultCodegen implements CodegenConf
         postProcessPattern(property.pattern, property.vendorExtensions);
     }
 
+    @Override
+    public Map<String, Object> postProcessModels(Map<String, Object> objs) {
+        // process enum in models
+        return postProcessModelsEnum(objs);
+    }
+
     @Override
     public void postProcessParameter(CodegenParameter parameter){
         postProcessPattern(parameter.pattern, parameter.vendorExtensions);
diff --git a/modules/swagger-codegen/src/main/resources/flaskConnexion/model.mustache b/modules/swagger-codegen/src/main/resources/flaskConnexion/model.mustache
index 9a0b4f89247..761492870bf 100644
--- a/modules/swagger-codegen/src/main/resources/flaskConnexion/model.mustache
+++ b/modules/swagger-codegen/src/main/resources/flaskConnexion/model.mustache
@@ -16,6 +16,16 @@ class {{classname}}(Model):
     NOTE: This class is auto generated by the swagger code generator program.
     Do not edit the class manually.
     """
+
+{{#allowableValues}}
+    """
+    allowed enum values
+    """
+{{#enumVars}}
+    {{name}} = {{{value}}}
+{{/enumVars}}
+{{/allowableValues}}
+
     def __init__(self{{#vars}}, {{name}}{{^supportPython2}}: {{datatype}}{{/supportPython2}}={{#defaultValue}}{{{defaultValue}}}{{/defaultValue}}{{^defaultValue}}None{{/defaultValue}}{{/vars}}):
         """
         {{classname}} - a model defined in Swagger