Пример #1
0
import os
os.environ['OMP_NUM_THREADS'] = '1'

from doom.game import play
# from doom.runners import run_basic, run_deadly_corridor
from utils.args import parse_arguments
import argparse

if __name__ == '__main__':
    # parser = argparse.ArgumentParser(description='AI runner')
    # parser.add_argument("--scenario", type=str, help="Scenario")
    # args, remaining = parser.parse_known_args()
    #
    # if args.scenario is not None:
    #     run_scenario(args.scenario)

    game_args = parse_arguments()
    play(game_args)
Пример #2
0
from utils.args import parse_arguments

params = parse_arguments()
print(params)

import numpy as np

if params.scenario == 'deadly_corridor':
    resize = (100, 181)
    crop = (30, -35, 1, -1)

    if params.actions == 'all':
        action_size = 10
    elif params.actions == 'single':
        action_size = 6

    state_size = np.prod(resize)

elif params.scenario == 'basic':
    resize = (84, 84)
    crop = (10, -10, 30, -30)

    if params.actions == 'all':
        action_size = 6
    elif params.actions == 'single':
        action_size = 3

    state_size = np.prod(resize)

elif params.scenario == 'defend_the_center':
    resize = (84, 159)
Пример #3
0
def main():
    banner("0.1")
    hr = "------------------------------------"
    # counter below needs to be added as part of the Sheepl Object
    # this should get automatically incremented either based on the
    # length of the task list or a counter tracker
    id = 1
    # Main Parser Setup
    args = parse_arguments()
#>> DEBUG
#    print(args)

    # assign colour output
    colour_output = args.colour
    print("[!] Colour output is set to : {}".format(str(colour_output).upper()))
    cl = ColourText(colour_output)

    # gets global header declarations for autoIT script
    # >> Global Calls

    # manual check for Name
    if not args.name and args.interactive:
        interactive = True
        console = SheeplConsole(cl)
    elif not args.name:
        print(cl.red("[!] You need to give this Sherson a name (--name <NAME>) - bah"))
        sys.exit(0)


    ## TODO: this needs to be pushed into a class from below to make S
    # Sheepl
    print("[:] - Creating the person called > {}".format(cl.green(args.name)))

    # first check for spaces
    file_name = args.name.replace(' ', '_').lower() + '.au3'
    # ------->
    # Parse time
    # goes through supplied arguments and creates tasks
    csh = Sheepl(args.name, args.total_time, args.type, cl, args.loop)
    csh.say_hello()

    # add the tea break task
    csh.add_task("tea_break", csh.tea_break_task())


    #############################################
    ##  Word
    #############################################

    if args.wordfile:
        from actions.office.word import WordDocument
        id += 1
        print("[!] - Word Interaction")
        print('[-] - Word output file is "{}" and user will type contents of "{}"'.format(args.wordfile, args.inputtext))
        word_file = WordDocument(args.inputtext , args.wordfile, str(id))
        csh.add_task('Word_' + str(id), word_file.create())


    #############################################
    ##  Excel
    #############################################

    # if args.wordfile:
    #     print("[!] - Excel Interaction")
    #     print("[!] - Adding Excel Header")
    #     print('[-] - Excel output file is "{}" and user will type contents of "{}"'.format(args.excelfile, args.inputcsv))
    #     # import word module
    #     from actions.office import excel
    #     # # TODO: get this from Headers
    #     # header = add_header('word', headers)
    #     # of.write(header + '\n')
    #     of.write('#include <Excel.au3>\n')
    #     of.write(excel.create_new_document())
    #     with open(args.inputtext) as st:
    #         poem = st.read()
    #         of.write(word.typing_block(poem) + '\n')
    #         of.write(word.save_file(args.wordfile))
    #     # # TODO: this will come from time variable based on some counting of some args
    #     # this will determine the wait time until the next the command
    #     # # TODO: break this out into a master function that decrements some cound and randomly
    #     # distributest the remaining time in milliseconds
    #     of.write('sleep(3000)' + '\n')


    #############################################
    ##  PowerShell
    #############################################

    if args.powershell:
        from actions.shell.powershell import PowerShell
        id += 1
        print(cl.green("[!] - Powershell Interaction"))
        print("[^] - Issuing the following powershell commands")
        ps_shell = PowerShell(args.pc, str(id))
        csh.add_task('Powershell_' + str(id), ps_shell.create())
        for command in args.pc:
            print('[-] - They will type "{}" into the PowerShell prompt'.format(command))


    #############################################
    ##  CMD
    #############################################

    if args.cmd:
        from actions.shell.cmd import CMDShell
        id += 1
        print(cl.green("[!] - Command Shell Interaction"))
        print("[!] - Issuing the following cmd commands")
        cmd_shell = CMDShell(args.cc, str(id))
        csh.add_task('CMD_' + str(id), cmd_shell.create())
        for command in args.cc:
            print('[-] - They will type "{}" into the CMD Shell'.format(command))


    #############################################
    ##  Internet Explorer
    #############################################

    if args.ie:
        from actions.browsing.ie import IEBrowser
        id += 1
        print(cl.green("[!] - Internet Explorer Interaction"))
        ie = IEBrowser(args.url, str(id))
        csh.add_task('IE_' + str(id), ie.create())
        print('[-] - They will browse to "{}" using Internet Explorer'.format(args.url))


    #############################################
    ##  RDP
    #############################################

    if args.rdp:
        from actions.network.remotedesktop import RDPConnection
        id += 1
        print(cl.green("[!] - Remote Desktop Interaction"))
        print(hr)
        print('[-] - The user "{}" will create a Remote Desktop Connection to "{}"'.format(args.rdp_user, args.rdp))
        if args.rdp_command:
            for command in args.rdp_command:
                print('[-] - They will type "{}" into the remote session'.format(command))
        rdp_conn = RDPConnection(args.rdp , args.rdp_user , args.rdp_password, str(id))
        csh.add_task('RDP_' + str(id), rdp_conn.create())


    #############################################
    ##  WRITE FILE
    #############################################

    print("[*] - Writing output file > {}\n".format(file_name))
    print(hr)
    csh.write_file(file_name)
Пример #4
0
def main():
    game_args = parse_arguments()
    play(game_args)