statblk 1.21__py3-none-any.whl → 1.22__py3-none-any.whl

{statblk-1.21.dist-info → statblk-1.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: statblk
-Version: 1.21
+Version: 1.22
 Summary: Gather essential disk and partition info for block devices and print it in a nice table
 Home-page: https://github.com/yufei-pan/statblk
 Author: Yufei Pan

statblk-1.22.dist-info/RECORD

@@ -0,0 +1,6 @@
+statblk.py,sha256=C8_7zjjE0b5hKj7GtK73jXcQcaRr96uhtWb-RDenPug,27212
+statblk-1.22.dist-info/METADATA,sha256=rVGQOV-ggZp3EHWxPqm9dp7ZaSWfYUkdhaPmhz0P-Cg,1465
+statblk-1.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+statblk-1.22.dist-info/entry_points.txt,sha256=JDz-sa6FIdaOckmlz9NbnhZXQaB5Yle-cTKgUQmAV40,41
+statblk-1.22.dist-info/top_level.txt,sha256=dBdU6_PD4tG_7uquWEs6YremqudiePASv3u3G59scf4,8
+statblk-1.22.dist-info/RECORD,,

statblk.py

@@ -276,7 +276,7 @@ except :
 	def cache_decorator(func):
 		return func
 
-version = '1.21'
+version = '1.22'
 VERSION = version
 __version__ = version
 COMMIT_DATE = '2025-09-10'
@@ -293,11 +293,11 @@ def read_text(path):
 def read_int(path):
 	s = read_text(path)
 	if s is None:
-		return None
+		return 0
 	try:
 		return int(s)
 	except Exception:
-		return None
+		return 0
 
 def build_symlink_dict(dir_path):
 	"""
@@ -337,7 +337,7 @@ def read_discard_support(sysfs_block_path):
 		return 'N/A'
 	dmbytes = read_int(os.path.join(sysfs_block_path, "queue", "discard_max_bytes"))
 	try:
-		if (dmbytes or 0) > 0:
+		if dmbytes > 0:
 			return 'Yes'
 		else:
 			return 'No'
@@ -345,7 +345,7 @@ def read_discard_support(sysfs_block_path):
 		return 'N/A'
 
 @cache_decorator
-def get_parent_device_sysfs(sysfs_block_path):
+def get_real_sysfs_device_path(sysfs_block_path):
 	"""
 	Return the sysfs 'device' directory for this block node (resolves partition
 	to its parent device as well).
@@ -360,7 +360,7 @@ def get_parent_device_sysfs(sysfs_block_path):
 def read_model_and_serial(sysfs_block_path):
 	if not sysfs_block_path or not os.path.isdir(sysfs_block_path):
 		return '', ''
-	device_path = get_parent_device_sysfs(sysfs_block_path)
+	device_path = get_real_sysfs_device_path(sysfs_block_path)
 	model = read_text(os.path.join(device_path, "model"))
 	serial = read_text(os.path.join(device_path, "serial"))
 	if serial is None:
@@ -375,12 +375,20 @@ def read_model_and_serial(sysfs_block_path):
 		serial = ''
 	return model, serial
 
+def read_size(sysfs_block_path):# -> tuple[int | None, Any] | Literal['']:
+	if not sysfs_block_path or not os.path.isdir(sysfs_block_path):
+		return ''
+	size = read_int(os.path.join(sysfs_block_path, "size"))
+	return size
+
 MountEntry = namedtuple("MountEntry", ["MOUNTPOINT", "FSTYPE", "OPTIONS"])
 def parseMount():
 	rtn = multiCMD.run_command('mount',timeout=1,quiet=True)
 	mount_table = defaultdict(list)
 	for line in rtn:
 		device_name, _, line = line.partition(' on ')
+		if device_name.startswith(os.path.sep):
+			device_name = os.path.realpath(device_name)
 		mount_point, _, line = line.partition(' type ')
 		fstype, _ , options = line.partition(' (')
 		options = options.rstrip(')').split(',')
@@ -429,7 +437,7 @@ def get_sector_size(sysfs_block_path):
 	if get_partition_parent_name(sysfs_block_path):
 		sysfs_block_path = os.path.join('/sys/class/block', os.path.basename(get_partition_parent_name(sysfs_block_path)))
 	sector_size = read_int(os.path.join(sysfs_block_path, "queue", "hw_sector_size"))
-	if sector_size is None:
+	if sector_size == 0:
 		sector_size = read_int(os.path.join(sysfs_block_path, "queue", "logical_block_size"))
 	return sector_size if sector_size else 512
 
@@ -503,7 +511,7 @@ def get_drives_info(print_bytes = False, use_1024 = False, mounted_only=False, b
 			if parent_name not in smart_infos:
 				smart_infos[parent_name] = multiCMD.run_command(f'{SMARTCTL_PATH} -H {parent_name}',timeout=2,quiet=True,wait_for_return=False,return_object=True)
 		if block_device not in tptDict:
-			sysfs_block_path = os.path.realpath(os.path.join('/sys/class/block', os.path.basename(block_device)))
+			sysfs_block_path = os.path.join('/sys/class/block', os.path.basename(block_device))
 			tptDict[block_device] = get_read_write_rate_throughput_iter(sysfs_block_path)
 	mount_table = parseMount()
 	target_devices = set(block_devices)
@@ -568,6 +576,7 @@ def get_drives_info(print_bytes = False, use_1024 = False, mounted_only=False, b
 					elif "denied" in line:
 						smart = 'DENIED'
 						break
+			size_bytes = read_size(os.path.join('/sys/class/block', os.path.basename(device_name)))
 		if device_name in tptDict:
 			try:
 				rtpt, wtpt = next(tptDict[device_name])
@@ -603,13 +612,14 @@ def get_drives_info(print_bytes = False, use_1024 = False, mounted_only=False, b
 				else:
 					size = multiCMD.format_bytes(size_bytes, use_1024_bytes=use_1024, to_str=True) + 'B'
 				if not full:
-					device_name = device_name.lstrip('/dev/')
+					device_name = device_name.replace('/dev/', '')
 				output.append([device_name, fstype, size, fsusepct, mountpoint, smart, label, uuid, model, serial, discard, rtpt, wtpt])
 		else:
 			if formated_only and device_name not in fstype_dict:
 				continue
 			fstype = fstype_dict.get(device_name, '')
-			size_bytes = size_dict.get(device_name, 0)
+			if not size_bytes:
+				size_bytes = size_dict.get(device_name, 0)
 			if size_bytes == 0 and not show_zero_size_devices:
 				continue
 			if print_bytes:
@@ -617,7 +627,7 @@ def get_drives_info(print_bytes = False, use_1024 = False, mounted_only=False, b
 			else:
 				size = multiCMD.format_bytes(size_bytes, use_1024_bytes=use_1024, to_str=True) + 'B'
 			if not full:
-				device_name = device_name.lstrip('/dev/')
+				device_name = device_name.replace('/dev/', '')
 			output.append([device_name, fstype, size, fsusepct, mountpoint, smart, label, uuid, model, serial, discard, rtpt, wtpt])
 	return output
 

statblk-1.21.dist-info/RECORD

@@ -1,6 +0,0 @@
-statblk.py,sha256=KzueRFpQZTcTJzrLPuC2kIRX0DilrnyyJJcGTfN2pEw,26804
-statblk-1.21.dist-info/METADATA,sha256=Zxv0Jkuz1hsATIkOG1cnRhOiFx-VF5SID3rWVEPGWMw,1465
-statblk-1.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-statblk-1.21.dist-info/entry_points.txt,sha256=JDz-sa6FIdaOckmlz9NbnhZXQaB5Yle-cTKgUQmAV40,41
-statblk-1.21.dist-info/top_level.txt,sha256=dBdU6_PD4tG_7uquWEs6YremqudiePASv3u3G59scf4,8
-statblk-1.21.dist-info/RECORD,,