1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
|
#!/usr/bin/env python
from __future__ import absolute_import
from __future__ import division
import subprocess
import pwnlib.args
pwnlib.args.free_form = False
from pwn import *
from pwnlib.commandline import common
p = common.parser_commands.add_parser(
'update',
help = 'Check for pwntools updates',
description = 'Check for pwntools updates'
)
p.add_argument('--install', action='store_true', help='''
Install the update automatically.
''')
p.add_argument('--pre', action='store_true', help='''
Check for pre-releases.
''')
def main(a):
result = pwnlib.update.perform_check(prerelease=a.pre)
if result and a.install:
subprocess.check_call(result, shell=False)
if __name__ == '__main__':
pwnlib.commandline.common.main(__file__, main)
|