54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
# This is not a direct clone of lsblk.c, this is a minimal reinterpretation for use on FreeBSD.
|
|
|
|
import subprocess
|
|
|
|
def get_geom_info():
|
|
result = subprocess.run(['geom', 'disk', 'list'], stdout=subprocess.PIPE)
|
|
return result.stdout.decode()
|
|
|
|
def parse_geom_output(output):
|
|
devices = []
|
|
device = {}
|
|
for line in output.splitlines():
|
|
if line.startswith('Geom name:'):
|
|
if device:
|
|
devices.append(device)
|
|
device = {}
|
|
device['name'] = line.split(': ')[1]
|
|
elif 'Mediasize' in line:
|
|
device['size'] = format_size(int(line.split(': ')[1].split(' ')[0]))
|
|
elif 'descr' in line:
|
|
device['description'] = line.split(': ')[1]
|
|
if device:
|
|
devices.append(device)
|
|
return devices
|
|
|
|
def format_size(size):
|
|
# Convert size in bytes to human-readable format
|
|
for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
|
|
if size < 1024:
|
|
return f"{size:.1f}{unit}"
|
|
size /= 1024
|
|
|
|
def get_mount_points():
|
|
result = subprocess.run(['mount'], stdout=subprocess.PIPE)
|
|
mount_output = result.stdout.decode()
|
|
mount_points = {}
|
|
for line in mount_output.splitlines():
|
|
parts = line.split(' ')
|
|
device = parts[0]
|
|
mount_point = parts[2]
|
|
mount_points[device] = mount_point
|
|
return mount_points
|
|
|
|
def print_lsblk_format(devices, mount_points):
|
|
print(f"{'NAME':<10} {'SIZE':<10} {'DESCRIPTION':<20} {'MOUNTPOINT':<20}")
|
|
for device in devices:
|
|
mount_point = mount_points.get(f"/dev/{device['name']}", '-')
|
|
print(f"{device['name']:<10} {device['size']:<10} {device['description']:<20} {mount_point:<20}")
|
|
|
|
if __name__ == "__main__":
|
|
geom_output = get_geom_info()
|
|
devices = parse_geom_output(geom_output)
|
|
mount_points = get_mount_points()
|
|
print_lsblk_format(devices, mount_points)
|