Example #1
0
        print(version.version)
        sys.exit(0)

    if sys.argv[1] == '-r':
        if len(sys.argv) <= 2:
            print(sys.argv[0] +
                  ': `-r` option requires code as argument: -r [code...]')
            sys.exit(1)
        script_commands = parser.parse(sys.argv[2], filepath='-')
        filename = '-r'
    else:
        filename = sys.argv[1]

        if sys.argv[1] == '-':
            script_content = ''
            for line in sys.stdin.readlines():
                script_content += line
            script_commands = parser.parse(script_content, filepath=filename)
        elif not os.path.isfile(filename):
            print(sys.argv[0] + ': file "' + filename + '" not found')
            sys.exit(1)
        else:
            # read content of file and parse it with the parser
            script_commands = jit.load(filename, code_location=filename)

    # make pashmak program object
    prog = program.Program(args=sys.argv[1:])
    prog.main_filename = filename
    prog.set_commands(script_commands)
    prog.start()
Example #2
0
def main():
    """ The main entry point """
    # set signal handler
    signal.signal(signal.SIGINT, signal_handler)

    # set the python path
    try:
        py_path = os.environ['PYTHONPATH'].split(';') if os.name.lower(
        ) == "nt" else os.environ['PYTHONPATH'].split(':')
        py_path = [p for p in py_path if p != '']
        sys.path = [*py_path, *sys.path]
    except KeyError:
        pass

    # validate arguments
    if len(sys.argv) <= 1:
        print(sys.argv[0] +
              ': script file name is required: pashmak [filename]')
        sys.exit(1)

    if sys.argv[1] == '--info':
        print('Pashmak Version: ' + version.version)
        print('Python Version: ', end='')
        print(sys.version.replace('\n', ''))
        sys.exit(0)

    if sys.argv[1] == '--version' or sys.argv[1] == '-v':
        print(version.version)
        sys.exit(0)

    is_module_run = False
    if sys.argv[1][0] == '@':
        module_name = sys.argv[1]
        is_module_run = module_name
        sys.argv[1] = '-r'
        sys.argv.insert(2, 'import_run ' + module_name)

    if sys.argv[1] == '-r':
        if len(sys.argv) <= 2:
            print(sys.argv[0] +
                  ': `-r` option requires code as argument: -r [code...]')
            sys.exit(1)
        script_commands = parser.parse(sys.argv[2], filepath='-')
        filename = '-r'
    else:
        filename = sys.argv[1]

        if sys.argv[1] == '-':
            filename = '<stdin>'
            script_content = ''.join(sys.stdin.readlines())
            script_commands = parser.parse(script_content, filepath=filename)
        elif not os.path.isfile(filename):
            print(sys.argv[0] + ': file "' + filename + '" not found')
            sys.exit(1)
        else:
            # read content of file and parse it with the parser
            script_commands = jit.load(filename,
                                       code_location=filename,
                                       ismain_default=True)

    # make pashmak program object
    if is_module_run:
        sys.argv[2] = is_module_run
        prog = program.Program(args=sys.argv[2:])
    else:
        prog = program.Program(args=sys.argv[1:])
    prog.main_filename = filename
    prog.set_commands(script_commands)
    prog.start()