Add: add helpers for storing, retrieving tokens for users

1 files changed, 11 insertions, 1 deletions

diff --git a/yarns/lib.py b/yarns/lib.py
index 0b95a81..b508389 100644
--- a/yarns/lib.py
+++ b/yarns/lib.py
@@ -75,12 +75,22 @@ def cat(filename):
             return open(filename, 'r').read()
 
 
+def store_token(user, token):
+    filename = '{}.jwt'.format(user)
+    write(filename, token)
+
+
+def get_token(user):
+    filename = '{}.jwt'.format(user)
+    return cat(filename)
+
+
 def get(url, token):
     headers = {
         'Authorization': 'Bearer {}'.format(token),
     }
     r = requests.get(url, headers=headers, verify=False)
-    return r.status_code, r.headers['Content-Type'], r.text
+    return r.status_code, r.headers['Content-Type'], dict(r.headers), r.text
 
 
 def post(url, body_text, token):