Exemplo n.º 1
0
def _run():
    """Initializes the state for the script based on command line input."""
    arg_parser = argparse.ArgumentParser()
    args = _setup_args(arg_parser)
    if args.logfile:
        c42api.set_log_file(args.logfile)

    if args.output or args.logfile:
        c42api.set_log_level(logging.DEBUG)
    else:
        c42api.set_log_level(logging.ERROR)
    server = argutil.server_from_args(args)
    is_json = args.format == 'json'
    params = {
        'active': 'true',
        'incBackupUsage': args.backup,
        'incHistory': args.backup
    }
    with c42api.common.smart_open(args.output, overwrite=True) as out:
        if is_json:
            c42api.write_json(
                out,
                c42api.fetch_computers(server,
                                       params,
                                       insert_schema_version=True))
        else:
            c42api.write_csv(out,
                             c42api.fetch_computers(
                                 server, params, insert_schema_version=True),
                             args.header,
                             shallow=True)
Exemplo n.º 2
0
def _run():
    """Initializes the state for the script based on command line input."""
    arg_parser = argparse.ArgumentParser()
    args = _setup_args(arg_parser)
    if args.logfile:
        c42api.set_log_file(args.logfile)

    if args.output or args.logfile:
        c42api.set_log_level(logging.DEBUG)
    else:
        c42api.set_log_level(logging.ERROR)
    server = argutil.server_from_args(args)
    is_json = args.format == 'json'
    with c42api.common.smart_open(args.output, overwrite=True) as out:
        if is_json:
            c42api.write_json(out, c42api.fetch_users(server))
        else:
            c42api.write_csv(out, c42api.fetch_users(server), args.header, shallow=True)
Exemplo n.º 3
0
def setup():
    """
    Initializes basic state for script running.

    :return: The authority server and config_dictionary.
    """
    key = _session_key()
    config_dict = _config_dict(key)
    if not config_dict or not key:
        print('something went wrong.')
        sys.exit(2)
    CONFIG.initialize(key)
    server = c42api.Server(config_dict['hostname'], port=config_dict['port'],
                           username=config_dict['username'], password=config_dict['password'],
                           verify_ssl=config_dict['verify_ssl'])
    if config_dict['collect_analytics']:
        c42api.common.analytics.OUTPUT_DIRECTORY = ANALYTICS_DIR
    log_file = os.path.join(app_home(), 'log', 'code42.log')
    c42api.set_log_file(log_file)
    return server, config_dict
Exemplo n.º 4
0
def _run():
    """Initializes the state for the script based on command line input."""
    arg_parser = argparse.ArgumentParser()
    args = _setup_args(arg_parser)
    if args.logfile:
        c42api.set_log_file(args.logfile)

    if args.output or args.logfile:
        c42api.set_log_level(logging.DEBUG)
    else:
        c42api.set_log_level(logging.ERROR)
    server = argutil.server_from_args(args)
    is_json = args.format == 'json'
    params = {'active': 'true',
              'incBackupUsage': args.backup,
              'incHistory': args.backup}
    with c42api.common.smart_open(args.output, overwrite=True) as out:
        if is_json:
            c42api.write_json(out, c42api.fetch_computers(server, params, insert_schema_version=True))
        else:
            c42api.write_csv(out, c42api.fetch_computers(server, params, insert_schema_version=True),
                             args.header, shallow=True)
Exemplo n.º 5
0
def setup():
    """
    Initializes basic state for script running.

    :return: The authority server and config_dictionary.
    """
    key = _session_key()
    config_dict = _config_dict(key)
    if not config_dict or not key:
        print('something went wrong.')
        sys.exit(2)
    CONFIG.initialize(key)
    server = c42api.Server(config_dict['hostname'],
                           port=config_dict['port'],
                           username=config_dict['username'],
                           password=config_dict['password'],
                           verify_ssl=config_dict['verify_ssl'])
    if config_dict['collect_analytics']:
        c42api.common.analytics.OUTPUT_DIRECTORY = ANALYTICS_DIR
    log_file = os.path.join(app_home(), 'log', 'code42.log')
    c42api.set_log_file(log_file)
    return server, config_dict
Exemplo n.º 6
0
def _run():
    """Initializes the state for the script based on command line input."""
    arg_parser = argparse.ArgumentParser()
    args = _setup_args(arg_parser)
    if args.logfile:
        c42api.set_log_file(args.logfile)

    if args.output or args.logfile:
        c42api.set_log_level(logging.DEBUG)
    else:
        c42api.set_log_level(logging.ERROR)
    server = argutil.server_from_args(args)
    is_json = args.format == 'json'
    device_guids = c42api.devices(server, args.device)

    def generate_detection_events():
        """
        Since write_json() wants to take a generator and fetch_detection_events()
        returns tuples, where the detection events are an array in the tuple, we use
        this auxilary function to allow us to stream the events to write_json()
        """
        event_filters = [
            c42api.create_filter_by_utc_datetime(args.min_date, args.max_date)
        ] * len(device_guids)
        guids_and_filters = zip(device_guids, event_filters)
        for _, _, detection_events in c42api.fetch_detection_events(
                server, guids_and_filters):
            for event in detection_events:
                yield event

    with c42api.common.smart_open(args.output, overwrite=True) as out:
        if is_json:
            c42api.write_json(out, generate_detection_events())
        else:
            c42api.write_csv(out,
                             generate_detection_events(),
                             args.header,
                             shallow=True)