def test_sanitize_file_name(self):
monkeypatch = MonkeyPatch()
monkeypatch.setattr('os.path.isfile', lambda _: True)
file_name = 'what/ever.exe'
file_src = '~/' + file_name
received = file_utils.sanitize_file_name(file_src)
home = os.path.expanduser("~")
expected = os.path.join(home, file_name)
assert expected == received
monkeypatch.undo()
def plot_to_file(self, file_name, dpi, fig_size=None):
'''
Plot to a file.
:param file_name: the image file name
:param dpi: the desired dpi resolution
:param fig_size: fig_size ratio
:return:
'''
file_name = file_utils.sanitize_file_name(file_name, False)
b = self.plot_to_buffer(dpi, fig_size)
if b != '':
self._write_image(b, file_name)
def get_file_header(file_src):
file_src = file_utils.sanitize_file_name(file_src)
if os.path.getsize(file_src) < 4:
return None
with open(file_src, "rb") as f:
magic = f.read(4)
if magic == FileHeaderParser.MAGIC_ELF_FILE_BYTES:
return 'ELF'
elif magic[0:2] == FileHeaderParser.MAGIC_PE_FILE_BYTES:
return 'PE'
else:
return None
def plot_to_dynamic_size_file(self, file_name, dpi, width, height):
'''
Plot to a file with dynamic width and height
:param file_name: the image file name
:param dpi: the desired dpi resolution
:param width: the desired width
:param height: the desired height
:return:
'''
file_name = file_utils.sanitize_file_name(file_name, False)
b = self.plot_to_dynamic_size_buffer(dpi, width, height)
if b != '':
self._write_image(b, file_name)
def _set_file_attributes(self, file_name):
file_name = file_utils.sanitize_file_name(file_name)
self._file_size = os.path.getsize(file_name)
self._short_filename = os.path.basename(file_name)
def __init__(self, file_src):
self._file_path = file_utils.sanitize_file_name(file_src)
self.cs_regions = self._parse_regions_with_code_scanner()
self.x_regions = self._parse_x_regions_with_header_parser()
def test_sanitize_file_name_no_file(self):
file_src = '~/sdfgsdfgdsfg'
with pytest.raises(IOError):
file_utils.sanitize_file_name(file_src)