From 7b8daabdabb000e8c9fb2d8be6f58e889b12d171 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.anjos@idiap.ch>
Date: Fri, 24 Jun 2016 17:12:30 +0200
Subject: [PATCH] [utils] Fix usage of new exclude argument

---
 beat/web/utils/management/commands/restore.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/beat/web/utils/management/commands/restore.py b/beat/web/utils/management/commands/restore.py
index e743eb4b1..1011d694a 100644
--- a/beat/web/utils/management/commands/restore.py
+++ b/beat/web/utils/management/commands/restore.py
@@ -99,6 +99,8 @@ class Command(BaseCommand):
             with tarfile.open(arguments['backup']) as tar:
                 tar.extractall(tmpdir)
 
+            exclude = arguments.get('exclude')
+
             arguments = dict(
                     verbosity=arguments.get('verbosity'),
                     interactive=False,
@@ -119,7 +121,7 @@ class Command(BaseCommand):
 
             # and loads the apps respecting the imposed order
             for app in APPS:
-                if app in arguments.get('exclude'): continue
+                if app in exclude: continue
                 # copy prefix data
                 path = os.path.join(settings.PREFIX, app)
                 srcdir = os.path.join(tmpdir, 'prefix', app)
-- 
GitLab