add issue utility

Author: mosquito

jwt_rsa/issue.py

@@ -1,13 +1,30 @@
 import argparse
+import json
 
 from .token import JWT
 from .rsa import load_private_key
 
 
 parser = argparse.ArgumentParser()
-parser.add_argument()
+parser.add_argument("-K", "--private-key", required=True,
+                    help="Private JWT key", type=load_private_key)
+parser.add_argument("--expired", help="Token expiration",
+                    type=float, default=...)
+parser.add_argument("--nbf", help="Token nbf claim",
+                    type=float, default=...)
 
 
 def main():
     arguments = parser.parse_args()
-    jwt = JWT(private_key=load_private_key(arguments.private_key))
+    jwt = JWT(private_key=arguments.private_key)
+    print(
+        jwt.encode(
+            expired=arguments.expired,
+            nbf=arguments.nbf,
+            **json.loads(input("Paste JSON content here: "))
+        )
+    )
+
+
+if __name__ == '__main__':
+    main()

jwt_rsa/keygen.py

@@ -8,8 +8,8 @@
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument('-b', '--bits', type=int, default=2048)
-    parser.add_argument('-P', '--pem', action='store_true')
+    parser.add_argument('-b', '--bits', dest="bits", type=int, default=2048)
+    parser.add_argument('-P', '--pem', dest="pem", action='store_true')
 
     arguments = parser.parse_args()
 

jwt_rsa/token.py

@@ -1,16 +1,19 @@
 import time
 from datetime import timedelta, datetime
 from operator import sub, add
+from typing import Union
 
 from jwt import PyJWT
 from jwt_rsa.rsa import RSAPrivateKey, RSAPublicKey
 
 
+DateType = Union[timedelta, datetime, float, int, type(Ellipsis)]
+
+
 class JWT:
     __slots__ = ('__private_key', '__public_key', '__jwt',
                  '__expires', '__nbf_delta')
 
-    _nil = object()
     DEFAULT_EXPIRATION = 86400 * 30  # one month
     NBF_DELTA = 20
 
@@ -31,12 +34,12 @@ def _date_to_timestamp(self, value, default, timedelta_func=add):
             return value.timestamp()
         elif isinstance(value, (int, float)):
             return value
-        elif value is self._nil:
+        elif value is Ellipsis:
             return default()
 
         raise ValueError(type(value))
 
-    def encode(self, expired=_nil, nbf=_nil, **claims) -> str:
+    def encode(self, expired: DateType=..., nbf: DateType=..., **claims) -> str:
         if not self.__private_key:
             raise RuntimeError("Can't encode without private key")
 

setup.py

@@ -45,6 +45,7 @@ def load_requirements(fname):
         'console_scripts': [
             'jwt-rsa-keygen = {}.keygen:main'.format(module_name),
             'jwt-rsa-verify = {}.verify:main'.format(module_name),
+            'jwt-rsa-issue= {}.issue:main'.format(module_name),
         ]
     },
     python_requires=">=3.4.*, <4",