예제 #1
0
 def test_version_higher(self):
     req_ver = "100.100.100"
     e = (
         "Script and API Version mismatch.*version {}, required {}"
     ).format(pytan.__version__, req_ver)
     with self.assertRaisesRegexp(Exception, e):
         utils.version_check(req_ver)
def csvdictwriter(rows_list, **kwargs):
    """returns the rows_list (list of dicts) as a CSV string"""
    csv_io = io.BytesIO()
    headers = kwargs.get('headers', []) or get_all_headers(rows_list)
    writer = csv.DictWriter(
        csv_io, fieldnames=headers, quoting=csv.QUOTE_NONNUMERIC,
    )
    writer.writerow(dict((h, h) for h in headers))
    writer.writerows(rows_list)
    csv_str = csv_io.getvalue()
    return csv_str


if __name__ == "__main__":

    utils.version_check(__version__)
    parser = utils.CustomArgParse(
        description=__doc__,
        add_help=True,
        formatter_class=utils.CustomArgFormat,
    )
    auth_group = parser.add_argument_group('Tanium Authentication')
    auth_group.add_argument(
        '-u',
        '--username',
        required=False,
        action='store',
        dest='username',
        default=None,
        help='Name of user',
    )
예제 #3
0
 def test_version_lower(self):
     self.assertTrue(utils.version_check('0.0.0'))
예제 #4
0
    csv_file = '{}_{}.csv'.format(my_name, utils.get_now())
    csv_file = filter_filename(csv_file)
    csv_file_path = os.path.join(report_dir, csv_file)

    csv_fh = open(csv_file_path, 'wb')
    csv_fh.write(csv_str)
    csv_fh.close()

    mylog.info("Final CSV results of {} written to: {}".format(
        story, csv_file_path))


if __name__ == "__main__":

    utils.version_check(__version__)
    parser = utils.CustomArgParse(
        description=__doc__,
        add_help=True,
        formatter_class=utils.CustomArgFormat,
    )
    auth_group = parser.add_argument_group('Tanium Authentication')
    auth_group.add_argument(
        '-u',
        '--username',
        required=False,
        action='store',
        dest='username',
        default=None,
        help='Name of user',
    )