Merge pull request #76 from SelfhostedPro/yaml-templates

added removal of whitespace to the right of a template URL to prevent…
SelfhostedPro · Sep 16, 2020 · 1779228 · 1779228
2 parents f432cc2 + 043e2c6
commit 1779228
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/api/db/crud/templates.py b/backend/api/db/crud/templates.py
@@ -47,9 +47,9 @@ def add_template(db: Session, template: models.containers.Template):
         _template = models.containers.Template(
             title=template.title, url=template.url)
         with urllib.request.urlopen(template.url) as file:
-            if ext in (".yml", '.yaml'):
+            if ext.rstrip() in (".yml", '.yaml'):
                 loaded_file = yaml.load(file, Loader=yaml.SafeLoader)
-            elif ext in (".json"):
+            elif ext.rstrip() in (".json", "json"):
                 loaded_file = json.load(file)
             else:
                 print('Invalid filetype')