Python 2-proof versions.py

This commit is contained in:
Philipp Hagemeister 2013-01-12 18:12:24 +01:00
parent bb28998920
commit 1841f65e64

View File

@ -1,15 +1,20 @@
#!/usr/bin/env python3
#!/usr/bin/env python
import rsa
import json
from binascii import hexlify
try:
input = raw_input
except NameError:
pass
versions_info = json.load(open('update/versions.json'))
if 'signature' in versions_info:
del versions_info['signature']
print('Enter the PKCS1 private key, followed by a blank line:')
privkey = ''
privkey = b''
while True:
try:
line = input()
@ -17,8 +22,7 @@
break
if line == '':
break
privkey += line + '\n'
privkey = bytes(privkey, 'ascii')
privkey += line.encode('ascii') + b'\n'
privkey = rsa.PrivateKey.load_pkcs1(privkey)
signature = hexlify(rsa.pkcs1.sign(json.dumps(versions_info, sort_keys=True).encode('utf-8'), privkey, 'SHA-256')).decode()