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