def fs_sink(): # create fs = FileSystemSink(FS_PATH) assert fs.stix_dir == FS_PATH yield fs # remove campaign dir shutil.rmtree(os.path.join(FS_PATH, "campaign"), True)
def test_filesystem_sink_nonexistent_folder(): with pytest.raises(ValueError) as excinfo: FileSystemSink('nonexistent-folder') assert "for STIX data does not exist" in str(excinfo)
print('\n') except Exception as e: print('Some other exception parsing:', o) print(e, '\n') if __name__ == '__main__': from stix2 import FileSystemSink store_path = pathlib.Path('/Users/priezm/repos/GTO/stix_worknench/') path = pathlib.Path('/Users/priezm/repos/GTO/working_stix_corpora') if not store_path.exists(): store_path.mkdir(parents=True) fs = workbench.FileSystemSource(str(store_path)) workbench.add_data_source(fs) sink = FileSystemSink('/Users/priezm/repos/GTO/stix_worknench/', allow_custom=True) if path.is_dir(): for p in tqdm(list(path.glob('**/*.json')), 'Files'): with open(p, 'r') as f: print( '\n\n***********************************************************' ) print('adding file: {}'.format(p)) try: irl = json.load(f) print('our irl: ', irl) print('our objects: ', irl['objects']) for o in tqdm(irl['objects'], 'Objects'): new_obj = obj_to_21(o) print('adding to sink') if new_obj is not None: