From 5da73a7ec1216d8f2dd93f100c5b1612cb716b17 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.anjos@idiap.ch>
Date: Tue, 20 Dec 2016 11:54:20 +0100
Subject: [PATCH] Use files() to check for download/upload/create commands for
 'all' special case (c.f. issue #11)

---
 bob/db/base/manage.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/bob/db/base/manage.py b/bob/db/base/manage.py
index 4ebbaa3..88f0958 100644
--- a/bob/db/base/manage.py
+++ b/bob/db/base/manage.py
@@ -21,7 +21,7 @@ def files_all(args):
 def upload_all(args):
   """Executes all the 'upload' commands from SQLite databases"""
 
-  for name in [k.name() for k in args.modules if k.type() in ('sqlite',)]:
+  for name in [k.name() for k in args.modules if k.files()]:
     parsed = args.parser.parse_args([name, 'upload'])
     parsed.destination = args.destination
     parsed.func(parsed)
@@ -30,7 +30,7 @@ def upload_all(args):
 def download_all(args):
   """Executes all the 'download' commands from SQLite databases"""
 
-  for name in [k.name() for k in args.modules if k.type() in ('sqlite',)]:
+  for name in [k.name() for k in args.modules if k.files()]:
     parsed = args.parser.parse_args([name, 'download'])
     parsed.source = args.source
     parsed.force = args.force
@@ -44,12 +44,12 @@ def create_all(args):
   databases = 0
   total_start = time.time()
 
-  sqlite_dbs = [k.name() for k in args.modules if k.type() in ('sqlite',)]
+  create_dbs = [k.name() for k in args.modules if k.files()]
 
   if args.verbose >= 1:
-    print('### Running %d SQLite database creation commands...' % len(sqlite_dbs))
+    print('### Running %d database creation commands...' % len(create_dbs))
 
-  for name in sqlite_dbs:
+  for name in create_dbs:
 
     start_time = time.time()
     databases += 1
-- 
GitLab