extension support added for yaml_git storage as well
diff --git a/reclass/storage/yaml_git/__init__.py b/reclass/storage/yaml_git/__init__.py
index a28079b..78eed04 100644
--- a/reclass/storage/yaml_git/__init__.py
+++ b/reclass/storage/yaml_git/__init__.py
@@ -34,7 +34,7 @@
from reclass.storage import ExternalNodeStorageBase
from reclass.storage.yamldata import YamlData
-FILE_EXTENSION = '.yml'
+FILE_EXTENSION = ('.yml', '.yaml')
STORAGE_NAME = 'yaml_git'
def path_mangler(inventory_base_uri, nodes_uri, classes_uri):
@@ -213,7 +213,7 @@
branch = {}
files = self.files_in_branch(bname)
for file in files:
- if fnmatch.fnmatch(file.name, '*{0}'.format(FILE_EXTENSION)):
+ if str(file.name).endswith(FILE_EXTENSION):
name = os.path.splitext(file.name)[0]
relpath = os.path.dirname(file.path)
if callable(self._class_name_mangler):