From 8a3c2e754b38cf5d66605e7cdcd30cda11e07362 Mon Sep 17 00:00:00 2001 From: wing328 Date: Sat, 2 Jul 2016 17:14:13 +0800 Subject: [PATCH] fix python flask to handle comment block --- .../codegen/languages/FlaskConnexionCodegen.java | 12 ++++++++++++ 1 file changed, 12 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 b2c1492631b..153d85f1596 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 @@ -336,4 +336,16 @@ public class FlaskConnexionCodegen extends DefaultCodegen implements CodegenConf // addPet => add_pet return underscore(operationId); } + + @Override + public String escapeQuotationMark(String input) { + // remove ' to avoid code injection + return input.replace("'", ""); + } + + @Override + public String escapeUnsafeCharacters(String input) { + // remove multiline comment + return input.replace("'''", "'_'_'"); + } }