def __init__(self, **kwargs): # Simply invokes the behavior of the superclass, but sets the filename keyword argument if it's not already set. if 'filename' in kwargs: super(MagicsFileHandler, self).__init__(**kwargs) else: magics_home_path = get_magics_home_path() logs_folder_name = "logs" log_file_name = "log_{}.log".format(get_instance_id()) directory = FileSystemReaderWriter(join_paths(magics_home_path, logs_folder_name)) directory.ensure_path_exists() super(MagicsFileHandler, self).__init__(filename=join_paths(directory.path, log_file_name), **kwargs)
def test_read(): path = "test" if os.path.isfile(path): os.remove(path) expected_lines = ["a\n", "b"] rw = FileSystemReaderWriter(path) with open("test", "w") as f: f.writelines(expected_lines) read_lines = rw.read_lines() assert expected_lines == read_lines os.remove(path)
def __init__(self, **kwargs): # Simply invokes the behavior of the superclass, but sets the filename keyword argument if it's not already set. if 'filename' in kwargs: super(MagicsFileHandler, self).__init__(**kwargs) else: magics_home_path = get_magics_home_path() logs_folder_name = "logs" log_file_name = "log_{}.log".format(get_instance_id()) directory = FileSystemReaderWriter( join_paths(magics_home_path, logs_folder_name)) directory.ensure_path_exists() super(MagicsFileHandler, self).__init__(filename=join_paths(directory.path, log_file_name), **kwargs)
def test_write_non_existent_file(): path = "test" if os.path.isfile(path): os.remove(path) expected_line = "hi" rw = FileSystemReaderWriter(path) rw.overwrite_with_line(expected_line) with open("test", "r") as f: lines = f.readlines() assert len(lines) == 1 assert lines[0] == expected_line os.remove(path)
def __init__(self, ipython_display, serialize_path=None): self.logger = Log("SparkController") self.ipython_display = ipython_display if serialize_path is not None: serializer = ClientManagerStateSerializer( FileSystemReaderWriter(serialize_path)) self.client_manager = ClientManager(serializer) else: self.client_manager = ClientManager()