Add overwrite files mode

2 files changed, 15 insertions, 4 deletions

diff --git a/deploy.py b/deploy.py
index 75bbabf..4f373ee 100755
--- a/deploy.py
+++ b/deploy.py
@@ -10,6 +10,7 @@ import yaml
 from termcolor import colored
 
 CWD = os.getcwd()
+OVERWRITE_FILES = os.getenv("OVERWRITE_FILES")
 
 
 def get_sections_meeting_preconditions(yaml):
@@ -47,6 +48,10 @@ print("Sections to run: " + ", ".join(
 ))
 print()
 
+if OVERWRITE_FILES == "1":
+    print(colored("Warning: ", "red") + "you have requested overwrite mode!")
+    print()
+
 if not query_yes_no("Proceed with install?", False):
     exit(1)
 
diff --git a/util/runners.py b/util/runners.py
index dcce820..dbfcf1f 100644
--- a/util/runners.py
+++ b/util/runners.py
@@ -4,6 +4,7 @@ from util.exceptions import StepFailedError
 from termcolor import colored
 
 HOME = os.getenv("HOME")
+OVERWRITE_FILES = os.getenv("OVERWRITE_FILES")
 
 
 def run_step(step):
@@ -24,12 +25,17 @@ def do_run(command):
 def do_link(source, destination):
     destination = re.sub(r"^~/", HOME + "/", destination)
     print(f"Linking {source} -> {destination}")
+    if os.path.exists(destination):
+        if OVERWRITE_FILES == "1":
+            os.remove(destination)
+            print(colored("  (File overwritten)", "yellow"))
+        else:
+            print(colored("  (File already exists)", "yellow"))
+            return
+
     try:
         os.link(source, destination)
     except OSError as e:
-        if e.errno == 17:
-            print(colored("  (File already exists)", "yellow"))
-        else:
-            raise StepFailedError("Link raised exeption: " + str(e)) from e
+        raise StepFailedError("Link raised exeption: " + str(e)) from e
     except Exception as e:
         raise StepFailedError("Link raised exeption: " + str(e)) from e