From 839c571f6ffb98171c88acd183e18d93b365f4aa Mon Sep 17 00:00:00 2001
From: Eggert Jung <eggert.s.jung@wischhof13.de>
Date: Mon, 23 Nov 2020 00:34:29 +0100
Subject: [PATCH] add file writing when using --out

---
 src/vigenere/vig.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/vigenere/vig.py b/src/vigenere/vig.py
index d215bd1..9470929 100644
--- a/src/vigenere/vig.py
+++ b/src/vigenere/vig.py
@@ -29,7 +29,12 @@ if __name__ == "__main__":
     t = ''.join([x for x in open(args.FILE, "r").read().lower() if x.isalpha()])
 
     if(args.encrypt != None):
-        print(encrypt(args.encrypt, t))
+        out = encrypt(args.encrypt, t)
 
     if(args.decrypt != None):
-        print(decrypt(args.decrypt, t))
+        out = decrypt(args.decrypt, t)
+
+    if(args.out != None):
+        open(args.out, "w").write(out)
+    else:
+        print(out)