Generación de transaction puzzle

op_sha256.py

# https://en.bitcoin.it/wiki/Script#Transaction_puzzle

from argparse import ArgumentParser
from binascii import b2a_hex
from hashlib import new

# Opcodes
OP_HASH256 = 'aa'
OP_EQUAL = '87'

# Arguments
p = ArgumentParser(description='Transaction puzzle (OP_SHA256)')
p.add_argument('solution', help='Puzzle solution')
args = p.parse_args()

def getScriptPubKey(s):
    hash = new('sha256', new('sha256', s.encode()).digest()).hexdigest()
    lenHash = '{:02x}'.format(int(len(hash)/2))
    return OP_HASH256 + lenHash + hash + OP_EQUAL

def main():
    s = args.solution

    print('> Puzzle solution: %s' % s)
    print('> ScriptPubKey: %s' % getScriptPubKey(s))
    print('> SigScript: %s' %  b2a_hex(s.encode()).decode('utf-8'))

if __name__ == '__main__':
    main()