shadowbrokers-exploits/windows/Resources/Ops/PyScripts/pcstate.py
2017-04-14 11:45:07 +02:00

27 lines
No EOL
777 B
Python

import ops.cmd
import ops.env
import ops.parseargs
from argparse import ArgumentTypeError
PCFIX_VAR = 'OPS_PCFIX'
def main():
def validate(x):
try:
x = int(x)
except ValueError as e:
raise ArgumentTypeError(e)
except:
raise
if ((x < 0) or (x > 5)):
raise ArgumentTypeError, 'Value must be on interval [0,5]'
return x
parser = ops.parseargs.ArgumentParser()
parser.add_argument('pcfix', type=validate, help='Number on interval [0,5]')
options = parser.parse_args()
ops.env.set(PCFIX_VAR, options.pcfix)
prob_cmd = ops.cmd.getDszCommand(('problem PCFIX %s' % options.pcfix))
prob_cmd.execute()
if (__name__ == '__main__'):
main()