def delete_currentdata(): """Delete the rundata file when exiting of the script.""" logger = getlogger() singletest = get_from_currentdata(SINGLETEST) if singletest: container = get_from_currentdata('container') cdir = get_container_dir(container) shutil.rmtree('%s/snapshots' % cdir) logger.critical("END: Completed the run and cleaning up.") runctx = get_currentdata() runctx['end'] = int(time.time() * 1000) runctx['log'] = FWLOGFILENAME if 'start' in runctx: runctx['duration'] = '%d seconds' % int( (runctx['end'] - runctx['start']) / 1000) runctx['start'] = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(runctx['start'] / 1000)) runctx['end'] = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(runctx['end'] / 1000)) for field in exclude_list: if field in runctx: del runctx[field] logger.info("\033[92m Run Stats: %s\033[00m" % json.dumps(runctx, indent=2)) run_file = framework_currentdata() remove_file(run_file)
def terraform_to_json_main(arg_vals=None): """Main driver utility for converting terraform to json files.""" logger = getlogger() logger.info("Comand: '%s %s'", sys.executable.rsplit('/', 1)[-1], ' '.join(sys.argv)) cmd_parser = argparse.ArgumentParser("Convert terraform to json files") cmd_parser.add_argument('terraform', action='store', help='Full path of the terraform file.') cmd_parser.add_argument('--output', action='store', default=None, help='Path to store the file.') args = cmd_parser.parse_args(arg_vals) # Delete the rundata at the end of the script. atexit.register(delete_currentdata) logger.info(args) init_currentdata() convert_terraform_to_json(args.terraform, args.output) return 0
""" Filesystem snapshot specific functionality using the base methods. """ import json from processor.helper.file.file_utils import exists_file from processor.logging.log_handler import getlogger from processor.helper.json.json_utils import get_field_value, json_from_file,\ TEST, MASTERTEST, SNAPSHOT, MASTERSNAPSHOT, make_snapshots_dir,\ store_snapshot, get_field_value_with_default, get_json_files, get_container_snapshot_json_files from processor.helper.config.config_utils import config_value, framework_dir, get_test_json_dir from processor.connector.snapshot_base import Snapshot from processor.connector.snapshot_exception import SnapshotsException logger = getlogger() class FSSnapshot(Snapshot): """ Filesystem snapshot utilities. """ def __init__(self, container, snapshot_refactored_fns, singleTest=None): """ Default isDb is false, singletest shall be set to the test that needs to be run.""" super().__init__(container, snapshot_refactored_fns) self.singleTest = singleTest reporting_path = config_value('REPORTING', 'reportOutputFolder') self.container_dir = '%s/%s/%s' % (framework_dir(), reporting_path, container) def get_structure_data(self, snapshot_object): """ Get the structure from the filesystem.""" structure_data = {}
def test_getlogger(): logger = getlogger() assert logger is not None logger1 = getlogger() assert logger1 is not None assert id(logger) == id(logger1)