logo

live-bootstrap

Mirror of <https://github.com/fosslinux/live-bootstrap>

target.py (1581B)

    

  1. #!/usr/bin/env python3
  3. # SPDX-FileCopyrightText: 2023 fosslinux <fosslinux@aussies.space>
  4. # SPDX-License-Identifier: GPL-3.0-or-later
  6. """
  7. Contains a class that represents a target directory
  8. """
  10. import enum
  11. import os
  13. from lib.utils import mount, create_disk
  15. class TargetType(enum.Enum):
  16.     """Different types of target dirs we can have"""
  17.     NONE = 0
  18.     TMPFS = 1
  20. class Target:
  21.     """
  22.     Represents a target directory
  23.     """
  25.     _disks = {}
  26.     _mountpoints = {}
  28.     def __init__(self, path="target"):
  29.         self.path = os.path.abspath(path)
  30.         self._type = TargetType.NONE
  32.         if not os.path.exists(self.path):
  33.             os.mkdir(self.path)
  35.     def tmpfs(self, size="8G"):
  36.         """Mount a tmpfs"""
  37.         print(f"Mounting tmpfs on {self.path}")
  38.         mount("tmpfs", self.path, "tmpfs", f"size={size}")
  39.         self._type = TargetType.TMPFS
  41.     # pylint: disable=too-many-arguments,too-many-positional-arguments
  42.     def add_disk(self,
  43.                  name,
  44.                  size="16G",
  45.                  filesystem="ext4",
  46.                  tabletype="msdos",
  47.                  bootable=False,
  48.                  mkfs_args=None):
  49.         """Add a disk"""
  50.         disk_path = os.path.join(self.path, f"{name}.img")
  51.         create_disk(disk_path,
  52.                     tabletype,
  53.                     filesystem,
  54.                     size,
  55.                     bootable,
  56.                     mkfs_args)
  57.         self._disks[name] = disk_path
  59.     def get_disk(self, name):
  60.         """Get the path to a device of a disk"""
  61.         return self._disks.get(name)