def test_make_8dot3_name_cut_ext():
"""Test that make_8dot3_filename generates valid 8dot3 filenames."""
fde = mock.MagicMock()
fde.get_entries.return_value = ([], [], [])
sfn = EightDotThree()
lfn = sfn.make_8dot3_name("This is a long filename.TeXT", fde)
assert "THISIS.TEX" == lfn
assert sfn.is_8dot3_conform(lfn)
def test_make_8dot3_name_unicode():
"""Test that make_8dot3_filename generates valid 8dot3 filenames."""
fde = mock.MagicMock()
fde.get_entries.return_value = ([], [], [])
sfn = EightDotThree()
lfn = sfn.make_8dot3_name("Ъци.Ъци", fde)
assert "_._" == lfn
assert sfn.is_8dot3_conform(lfn)
def test_make_8dot3_name():
"""Test that make_8dot3_filename generates valid 8dot3 filenames."""
fde = mock.MagicMock()
fde.get_entries.return_value = ([], [], [])
sfn = EightDotThree()
n = sfn.make_8dot3_name("This is a long filename.txt", fde)
sfn.set_str_name(n)
assert "THISIS.TXT" == n
assert sfn.is_8dot3_conform(sfn.get_unpadded_filename())
def test_make_8dot3_name_collision():
"""Test that make_8dot3_filename generates valid 8dot3 filenames."""
fde = mock.MagicMock()
fde_sub = mock.MagicMock()
fde_sub.get_short_name.side_effect = [
"THISIS.TXT", "THISIS~1.TXT", "THISIS~2.TXT"
]
fde.get_entries.return_value = ([fde_sub, fde_sub], [fde_sub], [])
sfn = EightDotThree()
lfn = sfn.make_8dot3_name("This is a long filename.txt", fde)
assert "THISIS~3.TXT" == lfn
assert sfn.is_8dot3_conform(lfn)
def makedir(self, path: str, permissions: Permissions = None,
recreate: bool = False):
"""Create directory on filesystem.
:param path: Path of new directory on filesystem
:param permissions: Currently not implemented
:param recreate: Ignore if directory already exists
"""
path = normpath(path)
base = split(path)[0]
dirname = split(path)[1]
# Plausability checks
try:
self.opendir(base)
except DirectoryExpected:
raise ResourceNotFound(path)
base = self._get_dir_entry(base)
try:
self._get_dir_entry(path)
except ResourceNotFound:
pass
else:
if not recreate:
raise DirectoryExists(path)
else:
# TODO: Update mtime
return SubFS(self, path)
parent_is_root = base == self.fs.root_dir
# Determine 8DOT3 file name + LFN
short_name = EightDotThree()
n = short_name.make_8dot3_name(dirname, base)
short_name.set_str_name(n)
dt = DosDateTime.now(tz=self.tz)
newdir = FATDirectoryEntry(DIR_Name=short_name,
DIR_Attr=FATDirectoryEntry.ATTR_DIRECTORY,
DIR_NTRes=0,
DIR_CrtTimeTenth=0,
DIR_CrtTime=dt.serialize_time(),
DIR_CrtDate=dt.serialize_date(),
DIR_LstAccessDate=dt.serialize_date(),
DIR_FstClusHI=0x00,
DIR_WrtTime=dt.serialize_time(),
DIR_WrtDate=dt.serialize_date(),
DIR_FstClusLO=0x00,
DIR_FileSize=0,
encoding=self.fs.encoding)
# Create LFN entry if required
_sfn = short_name.get_unpadded_filename()
if _sfn != dirname.upper() or (_sfn != dirname and self.preserve_case):
lfn_entry = make_lfn_entry(dirname, short_name)
newdir.set_lfn_entry(lfn_entry)
# Create . and .. directory entries
first_cluster = self.fs.allocate_bytes(
FATDirectoryEntry.FAT_DIRECTORY_HEADER_SIZE * 2,
erase=True)[0]
newdir.set_cluster(first_cluster)
dot_sn = EightDotThree()
dot_sn.set_byte_name(b". ")
dot = FATDirectoryEntry(DIR_Name=dot_sn,
DIR_Attr=FATDirectoryEntry.ATTR_DIRECTORY,
DIR_NTRes=newdir.ntres,
DIR_CrtTimeTenth=newdir.crttimetenth,
DIR_CrtTime=newdir.crttime,
DIR_CrtDate=newdir.crtdate,
DIR_LstAccessDate=newdir.lstaccessdate,
DIR_FstClusHI=newdir.fstclushi,
DIR_WrtTime=newdir.wrttime,
DIR_WrtDate=newdir.wrtdate,
DIR_FstClusLO=newdir.fstcluslo,
DIR_FileSize=newdir.filesize,
encoding=self.fs.encoding)
dotdot_sn = EightDotThree()
dotdot_sn.set_byte_name(b".. ")
base_fstclushi = base.fstclushi if not parent_is_root else 0x0
base_fstcluslo = base.fstcluslo if not parent_is_root else 0x0
dotdot = FATDirectoryEntry(DIR_Name=dotdot_sn,
DIR_Attr=FATDirectoryEntry.ATTR_DIRECTORY,
DIR_NTRes=base.ntres,
DIR_CrtTimeTenth=base.crttimetenth,
DIR_CrtTime=base.crttime,
DIR_CrtDate=base.crtdate,
DIR_LstAccessDate=base.lstaccessdate,
DIR_FstClusHI=base_fstclushi,
DIR_WrtTime=base.wrttime,
DIR_WrtDate=base.wrtdate,
DIR_FstClusLO=base_fstcluslo,
DIR_FileSize=base.filesize,
encoding=self.fs.encoding)
newdir.add_subdirectory(dot)
newdir.add_subdirectory(dotdot)
# Write new directory contents
self.fs.update_directory_entry(newdir)
# Write parent directory
base.add_subdirectory(newdir)
self.fs.update_directory_entry(base)
# Flush FAT(s) to disk
self.fs.flush_fat()
return SubFS(self, path)
def create(self, path: str, wipe: bool = False) -> bool:
"""Create a new file.
:param path: Path of new file on filesystem
:param wipe: Overwrite existing file contents
"""
basename = "/".join(path.split("/")[:-1])
dirname = path.split("/")[-1]
# Plausability checks
try:
self.opendir(basename)
except DirectoryExpected:
raise ResourceNotFound(path)
base = self._get_dir_entry(basename)
try:
dentry = self._get_dir_entry(path)
except ResourceNotFound:
pass
else:
if not wipe:
return False
else:
# Clean up existing file contents
# TODO: touch {a,m}time
dentry.set_size(0)
old_cluster = dentry.get_cluster()
dentry.set_cluster(0)
self.fs.free_cluster_chain(old_cluster)
return True
# Determine 8DOT3 file name + LFN
short_name = EightDotThree()
n = short_name.make_8dot3_name(dirname, base)
short_name.set_str_name(n)
dt = DosDateTime.now(tz=self.tz)
newdir = FATDirectoryEntry(DIR_Name=short_name,
DIR_Attr=0,
DIR_NTRes=0,
DIR_CrtTimeTenth=0,
DIR_CrtTime=dt.serialize_time(),
DIR_CrtDate=dt.serialize_date(),
DIR_LstAccessDate=dt.serialize_date(),
DIR_FstClusHI=0x00,
DIR_WrtTime=dt.serialize_time(),
DIR_WrtDate=dt.serialize_date(),
DIR_FstClusLO=0x00,
DIR_FileSize=0,
encoding=self.fs.encoding)
# Create LFN entry if required
_sfn = short_name.get_unpadded_filename()
if _sfn != dirname.upper() or (_sfn != dirname and self.preserve_case):
lfn_entry = make_lfn_entry(dirname, short_name)
newdir.set_lfn_entry(lfn_entry)
# Write reference to parent directory
base.add_subdirectory(newdir)
self.fs.update_directory_entry(base)
# Flush FAT(s) to disk
self.fs.flush_fat()
return True