Add support for .yaml #104

francescou · Jan 12, 2018 · e7249dc · droogmic · Jan 12, 2018 · e7249dc
1 parent 6bb72f2
commit e7249dc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/scripts/find_files.py b/scripts/find_files.py
@@ -11,7 +11,7 @@ def find_yml_files(path):
     """
     matches = {}
     for root, _, filenames in os.walk(path, followlinks=True):
-        for _ in fnmatch.filter(filenames, 'docker-compose.yml'):
+        for _ in list(set().union(fnmatch.filter(filenames, 'docker-compose.yml'), fnmatch.filter(filenames, 'docker-compose.yaml'))):
             key = root.split('/')[-1]
             matches[key] = os.path.join(os.getcwd(), root)
     return matches

diff --git a/scripts/manage_project.py b/scripts/manage_project.py
@@ -2,15 +2,15 @@
 docker compose project management
 """
 
-from os import rename, makedirs
+from os import rename, makedirs, path
 from time import time
 
 def manage(directory, yml, is_update):
     """
     create or update docker compose project
     """
 
-    file_path = directory + "/docker-compose.yml"
+    file_path = directory + "/docker-compose.yml" if path.exists(directory + "/docker-compose.yml") else "/docker-compose.yaml"
 
     if is_update:
         rename(file_path, file_path + "." + str(int(round(time()))))