def test_sizeof_fmt(self):
"""Format filesize in human-readable format."""
readable_size = helpers.sizeof_fmt(100)
self.assertEqual(readable_size, "100B")
readable_size = helpers.sizeof_fmt(10240)
self.assertEqual(readable_size, "10.0K")
huge_size = 1024**8 * 12
readable_size = helpers.sizeof_fmt(huge_size)
self.assertEqual(readable_size, "12.0Y")
def test_sizeof_fmt(self):
"""Format filesize in human-readable format."""
readable_size = helpers.sizeof_fmt(100)
self.assertEqual(readable_size, "100B")
readable_size = helpers.sizeof_fmt(10240)
self.assertEqual(readable_size, "10.0K")
huge_size = 1024**8 * 12
readable_size = helpers.sizeof_fmt(huge_size)
self.assertEqual(readable_size, "12.0Y")
def filelist_create(self, directory="."):
"""Create a filelist from all files in directory.
Args:
directory: Directory of which the filelist is created.
"""
# Get data from ls -lh and parse it correctly
files = listdir_wrapper(directory, self.show_hidden)
self.filesize = {}
for fil in files:
# Catch broken symbolic links
if os.path.islink(fil) and \
not os.path.exists(os.path.realpath(fil)):
continue
# Number of images in directory as filesize
if os.path.isdir(fil):
try:
subfiles = listdir_wrapper(fil, self.show_hidden)
# Necessary to keep acceptable speed in library
many = False
if len(subfiles) > self.file_check_amount:
many = True
subfiles = [subfile
for subfile in subfiles[:self.file_check_amount]
if is_image(os.path.join(fil, subfile))]
amount = str(len(subfiles))
if subfiles and many:
amount += "+"
self.filesize[fil] = amount
except:
self.filesize[fil] = "N/A"
else:
self.filesize[fil] = sizeof_fmt(os.path.getsize(fil))
return files
def filelist_create(self, directory="."):
"""Create a filelist from all files in directory.
Args:
directory: Directory of which the filelist is created.
"""
# Get data from ls -lh and parse it correctly
files = listdir_wrapper(directory, self.show_hidden)
self.filesize = {}
for fil in files:
# Number of images in directory as filesize
if os.path.isdir(fil):
try:
subfiles = listdir_wrapper(fil, self.show_hidden)
# Necessary to keep acceptable speed in library
many = False
if len(subfiles) > self.file_check_amount:
many = True
subfiles = [
subfile
for subfile in subfiles[:self.file_check_amount]
if is_image(os.path.join(fil, subfile))
]
amount = str(len(subfiles))
if subfiles and many:
amount += "+"
self.filesize[fil] = amount
except:
self.filesize[fil] = "N/A"
else:
self.filesize[fil] = sizeof_fmt(os.path.getsize(fil))
return files
def _filelist_create(self, directory="."):
"""Create a filelist from all files in directory.
Args:
directory: Directory of which the filelist is created.
Return:
filelist, filesize: List of files, dictionary with filesize info
"""
# Get data from ls -lh and parse it correctly
files = listdir_wrapper(directory, settings["show_hidden"].get_value())
filesize = {}
file_check_amount = settings["file_check_amount"].get_value()
for fil in files:
# Catch broken symbolic links
if os.path.islink(fil) and \
not os.path.exists(os.path.realpath(fil)):
continue
# Number of images in directory as filesize
if os.path.isdir(fil):
try:
subfiles = listdir_wrapper(
fil, settings["show_hidden"].get_value())
# Necessary to keep acceptable speed in library
many = False
if len(subfiles) > file_check_amount:
many = True
subfiles = [
sub for sub in subfiles[:file_check_amount]
if is_image(os.path.join(fil, sub))
]
amount = str(len(subfiles))
if subfiles and many:
amount += "+"
filesize[fil] = amount
except PermissionError:
filesize[fil] = "N/A"
else:
filesize[fil] = sizeof_fmt(os.path.getsize(fil))
return files, filesize
