Fix template extension retrivial

lukeseawalker · demartinofra · commit d38e6feac02f · 2020-09-11T14:07:14.000+02:00
Signed-off-by: Luca Carrogu &lt;carrogu@amazon.com&gt;
diff --git a/util/upload-cfn-templates.py b/util/upload-cfn-templates.py
@@ -32,7 +32,7 @@ def get_template_extension(templates_dir, template_name):
             "Found 0 or multiple matching files for template name {}: {}".format(template_name, matching_files)
         )
     file_name = os.path.basename(matching_files[0])
-    extension = file_name[len(template_name) :]  # noqa: E203
+    extension = os.path.splitext(file_name)[1]
     if extension not in {".cfn.json", ".cfn.yaml"}:
         raise Exception("Found invalid extension for template {}: {}".format(template_name, extension))
     return extension

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ def get_template_extension(templates_dir, template_name):`
`32`	`32`	`"Found 0 or multiple matching files for template name {}: {}".format(template_name, matching_files)`
`33`	`33`	`)`
`34`	`34`	`file_name = os.path.basename(matching_files[0])`
`35`		`- extension = file_name[len(template_name) :] # noqa: E203`
	`35`	`+ extension = os.path.splitext(file_name)[1]`
`36`	`36`	`if extension not in {".cfn.json", ".cfn.yaml"}:`
`37`	`37`	`raise Exception("Found invalid extension for template {}: {}".format(template_name, extension))`
`38`	`38`	`return extension`