def write_csv(list_of_dicts, report_dir, my_name, story):
headers = sorted(list_of_dicts[0].keys())
csv_str = csvdictwriter(list_of_dicts, headers=headers)
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))
def write_csv(list_of_dicts, report_dir, my_name, story):
headers = sorted(list_of_dicts[0].keys())
csv_str = csvdictwriter(list_of_dicts, headers=headers)
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))
parent_dir = os.path.dirname(my_dir)
lib_dir = os.path.join(parent_dir, 'lib')
path_adds = [lib_dir]
for aa in path_adds:
if aa not in sys.path:
sys.path.append(aa)
import pytan
import taniumpy
from pytan import utils
from pytan import constants
examples = []
OUTPUT_DIR = os.path.join(os.getcwd(), 'TSAT_OUTPUT', utils.get_now())
mylog = logging.getLogger(os.path.basename(my_file))
mylog.setLevel(logging.INFO)
def filter_filename(filename):
valid_chars = "-_.() %s%s" % (string.ascii_letters, string.digits)
filename = ''.join(c for c in filename if c in valid_chars)
return filename
def remove_file_log(logfile):
basename = os.path.basename(logfile)
root_logger = logging.getLogger()
try:
for x in root_logger.handlers:
parent_dir = os.path.dirname(my_dir)
lib_dir = os.path.join(parent_dir, 'lib')
path_adds = [lib_dir]
for aa in path_adds:
if aa not in sys.path:
sys.path.append(aa)
import pytan
import taniumpy
from pytan import utils
from pytan import constants
examples = []
OUTPUT_DIR = os.path.join(os.getcwd(), 'TUAT_OUTPUT', utils.get_now())
mylog = logging.getLogger("TUAT")
mylog.setLevel(logging.INFO)
def fix_mac(m):
return str(m).upper().replace('-', ':')
def dictify_resultset_row(rs_row):
d = dict(zip(
[x.display_name for x in rs_row.columns],
[x[0] for x in rs_row.vals if type(x) == list]
))
return d
def test_get_now(self):
self.assertTrue(utils.get_now())
parent_dir = os.path.dirname(my_dir)
lib_dir = os.path.join(parent_dir, 'lib')
path_adds = [lib_dir]
for aa in path_adds:
if aa not in sys.path:
sys.path.append(aa)
import pytan
import taniumpy
from pytan import utils
from pytan import constants
examples = []
OUTPUT_DIR = os.path.join(os.getcwd(), 'TUAT_OUTPUT', utils.get_now())
mylog = logging.getLogger("TUAT")
mylog.setLevel(logging.INFO)
def fix_mac(m):
return str(m).upper().replace('-', ':')
def dictify_resultset_row(rs_row):
d = dict(
zip([x.display_name for x in rs_row.columns],
[x[0] for x in rs_row.vals if type(x) == list]))
return d
