You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 

39 lines
1.2 KiB

#!/usr/bin/python
import os
import sys
import re
import json
import serpent
def main():
if len(sys.argv) == 1:
print "serpent <command> <arg1> <arg2> ..."
else:
cmd = sys.argv[2] if sys.argv[1][0] == '-' else sys.argv[1]
if sys.argv[1] == '-s':
args = re.findall(r'\S\S*', sys.stdin.read()) + sys.argv[3:]
elif sys.argv[1] == '-B':
args = [sys.stdin.read()] + sys.argv[3:]
elif sys.argv[1] == '-b':
args = [sys.stdin.read()[:-1]] + sys.argv[3:] # remove trailing \n
elif sys.argv[1] == '-j':
args = [json.loads(sys.stdin.read())] + sys.argv[3:]
elif sys.argv[1] == '-J':
args = json.loads(sys.stdin.read()) + sys.argv[3:]
elif sys.argv[1] == '-h':
args = [x.decode('hex') for x in sys.argv[3:]]
else:
cmd = sys.argv[1]
args = sys.argv[2:]
if args[0] in os.listdir(os.getcwd()):
args[0] = open(args[0]).read()
o = getattr(serpent, cmd)(*args)
if isinstance(o, (list, dict)):
print json.dumps(o)
else:
print o.encode('hex')
if __name__ == '__main__':
main()