From 107f7b9afe277ad0a05582d3bf564dbd5fe24414 Mon Sep 17 00:00:00 2001
From: Attila Farkas <ati@mail.muni.cz>
Date: Tue, 28 Jul 2020 09:21:25 +0200
Subject: [PATCH] copy extra_vars.yml to sandbox

---
 modules/file_manager.py      | 8 ++++++++
 modules/vagrant_generator.py | 2 --
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/modules/file_manager.py b/modules/file_manager.py
index 913c47e..f317f12 100644
--- a/modules/file_manager.py
+++ b/modules/file_manager.py
@@ -134,6 +134,14 @@ def prepare_directories(device_definitions, flags):
         cleanup_and_exit("Could not create directory ./" + OUTPUT_DIRECTORY +
                          ".")
 
+    if "extra_vars" in flags and flags["extra_vars"]:
+        try:
+            os.mkdir(OUTPUT_DIRECTORY + "/user_files")
+            shutil.copyfile(flags["extra_vars"],
+                            OUTPUT_DIRECTORY + "/user_files/extra_vars.yml")
+        except IOError:
+            cleanup_and_exit("Could not import extra_vars.")
+
     _create_provisioning_directories("base_provisioning", device_definitions)
     if "provisioning_dir" not in flags or not flags["provisioning_dir"]:
         _create_provisioning_directories("provisioning", device_definitions)
diff --git a/modules/vagrant_generator.py b/modules/vagrant_generator.py
index ff8653a..f6211bb 100644
--- a/modules/vagrant_generator.py
+++ b/modules/vagrant_generator.py
@@ -1,7 +1,5 @@
 """Contains functions for generating a Vagrantfile from input definitions."""
 
-import os
-
 from modules.file_manager import generate_file, open_yaml
 
 VAGRANT_MAPPING = open_yaml("conf/vagrant_mapping.yml")
-- 
GitLab