Resize Disk Partitions in Python (Mock Script)

#!/usr/bin/env python3
"""Mock script to demonstrate disk partition resize logic."""
import json
from dataclasses import dataclass
from typing import Dict


@dataclass
class Partition:
    name: str
    size_gb: int
    mount_point: str

    def to_dict(self) -> Dict[str, object]:
        return {
            "name": self.name,
            "size_gb": self.size_gb,
            "mount_point": self.mount_point,
        }


def resize_partition(partition: Partition, new_size_gb: int) -> Partition:
    """Resize a partition to the specified size (mock operation)."""
    if new_size_gb <= 0:
        raise ValueError("New size must be positive")
    if new_size_gb < 2:
        print(f"Warning: {partition.name} is very small ({new_size_gb} GB)")

    old_size = partition.size_gb
    partition.size_gb = new_size_gb
    print(f"Resized {partition.name} from {old_size} GB to {new_size_gb} GB")
    return partition


def main() -> None:
    # Simulate inspecting the disk and performing the resize
    disk = {
        "/dev/sda": Partition(name="/dev/sda1", size_gb=50, mount_point="/"),
        "/dev/sda": Partition(name="/dev/sda2", size_gb=100, mount_point="/home"),
    }

    # Perform resize on the root partition
    target = disk["/dev/sda"]
    resized = resize_partition(target, 80)

    # Output the updated disk layout as JSON for verification
    layout = [p.to_dict() for p in disk.values()]
    print(json.dumps({"disk": "/dev/sda", "partitions": layout}, indent=2))


if __name__ == "__main__":
    main()