Added -e flag to jwt_producer.

1 files changed, 10 insertions, 3 deletions

diff --git a/tools/jwt_producer.py b/tools/jwt_producer.py
index 9f30439..a024c13 100644
--- a/tools/jwt_producer.py
+++ b/tools/jwt_producer.py
@@ -10,7 +10,8 @@ def usage():
     print(f'Usage: {progname} [-p <path>] [-w <value>] [-r <value>]\n' +
           '  -p <path to private key>\n' +
           '  -w <write, comma separated list of domains>\n' +
-          '  -r <read, comma separated list of domains>')
+          '  -r <read, comma separated list of domains>\n' +
+          '  -e <print export statement')
     sys.exit(0)
 
 
@@ -42,9 +43,10 @@ if __name__ == '__main__':
     read_domains = None
     write_domains = None
     private_key = None
+    export = False
 
     try:
-        opts, args = getopt.getopt(sys.argv[1:], 'p:w:r:h')
+        opts, args = getopt.getopt(sys.argv[1:], 'p:w:r:eh')
     except getopt.GetoptError:
         usage()
 
@@ -55,6 +57,8 @@ if __name__ == '__main__':
             write_domains = arg
         elif opt == '-r':
             read_domains = arg
+        elif opt == '-e':
+            export = True
         elif opt == '-h':
             usage()
         else:
@@ -69,4 +73,7 @@ if __name__ == '__main__':
     token = create_token(private_key, write_domains,
                          read_domains)
 
-    print(f'{token}')
+    if export:
+        print(f'export JWT_TOKEN={token}')
+    else:
+        print(f'{token}')