From 4704524fbadeb1b5350ae4ffd86f51015006af01 Mon Sep 17 00:00:00 2001
From: Ville Rantanen <ville.rantanen@reaktor.com>
Date: Tue, 20 Jun 2023 16:58:24 +0300
Subject: [PATCH] spill get key from file

---
 py-packages/spiller/spiller/__init__.py |  2 +-
 py-packages/spiller/spiller/spiller.py  | 21 ++++++++++++++++++++-
 2 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/py-packages/spiller/spiller/__init__.py b/py-packages/spiller/spiller/__init__.py
index c48efc5..0f9d4c6 100644
--- a/py-packages/spiller/spiller/__init__.py
+++ b/py-packages/spiller/spiller/__init__.py
@@ -1,2 +1,2 @@
-__version__ = "0.2"
+__version__ = "0.3"
 from spiller.spiller import retrieve, store, list_storage
diff --git a/py-packages/spiller/spiller/spiller.py b/py-packages/spiller/spiller/spiller.py
index fdb5abb..4f5b56b 100755
--- a/py-packages/spiller/spiller/spiller.py
+++ b/py-packages/spiller/spiller/spiller.py
@@ -59,6 +59,13 @@ def get_opts():
         default=os.getenv("SPILLER_KEY", None),
         help="Encryption key",
     )
+    set_parser.add_argument(
+        "--key-file",
+        action="store",
+        default=None,
+        type=argparse.FileType('r'),
+        help="Read encryption key stored in a file",
+    )
     get_parser.add_argument(
         "name",
         action="store",
@@ -70,12 +77,24 @@ def get_opts():
         default=os.getenv("SPILLER_KEY", None),
         help="Decryption key",
     )
+    get_parser.add_argument(
+        "--key-file",
+        action="store",
+        default=None,
+        type=argparse.FileType('r'),
+        help="Read encryption key stored in a file",
+    )
     del_parser.add_argument(
         "name",
         action="store",
         help="Name of secret to delete",
     )
-    return parser.parse_args()
+    args = parser.parse_args()
+    if args.key_file:
+        with args.key_file as fp:
+            args.key = fp.read().rstrip("\n")
+    return args
+
 
 
 def list_storage():