diff --git a/Readme.md b/Readme.md index 523406b..ddf04fc 100755 --- a/Readme.md +++ b/Readme.md @@ -21,6 +21,9 @@ A small packet I use for testing. ```scapy``` needed. ### 6. `cpp-project-gen.py` A small script to generate a c++ project structure. +### 7. `lsblk.py` +This is not a direct clone of lsblk.c, this is a minimal reinterpretation for use on FreeBSD. It works in the sense that it functions. Maybe it'll get added to. + ## Installation No installation is necessary for these scripts. They are standalone Python scripts that can be run from the command line. Ensure you have Python installed on your system, navigate to the directory containing these scripts, and execute them with Python. For example: diff --git a/lsblk.py b/lsblk.py new file mode 100644 index 0000000..3ee8d87 --- /dev/null +++ b/lsblk.py @@ -0,0 +1,54 @@ +# 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)