diff options
| author | Gabi Davar <grizzly.nyo@gmail.com> | 2015-04-26 18:11:26 +0300 |
|---|---|---|
| committer | Mark Adams <mark@markadams.me> | 2015-05-17 22:44:53 -0500 |
| commit | c3e82547980d2f70440b1e198670f7ac1605418a (patch) | |
| tree | 9be1ee0a43da711ddac327959172caa367024321 /jwt | |
| parent | b4ede91a85be4befb4075da15a1ded8bf6874bbb (diff) | |
| download | pyjwt-c3e82547980d2f70440b1e198670f7ac1605418a.tar.gz | |
refactor scripts, testing, travis and tox
Diffstat (limited to 'jwt')
| -rw-r--r-- | jwt/__main__.py | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/jwt/__main__.py b/jwt/__main__.py new file mode 100644 index 0000000..8b20439 --- /dev/null +++ b/jwt/__main__.py @@ -0,0 +1,135 @@ +#!/usr/bin/env python
+
+from __future__ import absolute_import, print_function
+
+import json
+import optparse
+import sys
+import time
+
+from . import DecodeError, __package__, __version__, decode, encode
+
+
+def main():
+
+ usage = '''Encodes or decodes JSON Web Tokens based on input.
+
+ %prog [options] input
+
+Decoding examples:
+
+ %prog --key=secret json.web.token
+ %prog --no-verify json.web.token
+
+Encoding requires the key option and takes space separated key/value pairs
+separated by equals (=) as input. Examples:
+
+ %prog --key=secret iss=me exp=1302049071
+ %prog --key=secret foo=bar exp=+10
+
+The exp key is special and can take an offset to current Unix time.\
+'''
+ p = optparse.OptionParser(
+ usage=usage,
+ prog=__package__,
+ version='%s %s' % (__package__, __version__),
+ )
+
+ p.add_option(
+ '-n', '--no-verify',
+ action='store_false',
+ dest='verify',
+ default=True,
+ help='ignore signature verification on decode'
+ )
+
+ p.add_option(
+ '--key',
+ dest='key',
+ metavar='KEY',
+ default=None,
+ help='set the secret key to sign with'
+ )
+
+ p.add_option(
+ '--alg',
+ dest='algorithm',
+ metavar='ALG',
+ default='HS256',
+ help='set crypto algorithm to sign with. default=HS256'
+ )
+
+ options, arguments = p.parse_args()
+
+ if len(arguments) > 0 or not sys.stdin.isatty():
+ if len(arguments) == 1 and (not options.verify or options.key):
+ # Try to decode
+ try:
+ if not sys.stdin.isatty():
+ token = sys.stdin.read()
+ else:
+ token = arguments[0]
+
+ token = token.encode('utf-8')
+ data = decode(token, key=options.key, verify=options.verify)
+
+ print(json.dumps(data))
+ sys.exit(0)
+ except DecodeError as e:
+ print(e)
+ sys.exit(1)
+
+ # Try to encode
+ if options.key is None:
+ print('Key is required when encoding. See --help for usage.')
+ sys.exit(1)
+
+ # Build payload object to encode
+ payload = {}
+
+ for arg in arguments:
+ try:
+ k, v = arg.split('=', 1)
+
+ # exp +offset special case?
+ if k == 'exp' and v[0] == '+' and len(v) > 1:
+ v = str(int(time.time()+int(v[1:])))
+
+ # Cast to integer?
+ if v.isdigit():
+ v = int(v)
+ else:
+ # Cast to float?
+ try:
+ v = float(v)
+ except ValueError:
+ pass
+
+ # Cast to true, false, or null?
+ constants = {'true': True, 'false': False, 'null': None}
+
+ if v in constants:
+ v = constants[v]
+
+ payload[k] = v
+ except ValueError:
+ print('Invalid encoding input at {}'.format(arg))
+ sys.exit(1)
+
+ try:
+ token = encode(
+ payload,
+ key=options.key,
+ algorithm=options.algorithm
+ )
+
+ print(token)
+ sys.exit(0)
+ except Exception as e:
+ print(e)
+ sys.exit(1)
+ else:
+ p.print_help()
+
+if __name__ == '__main__':
+ main()
|