Handle scratch disks (no source attribute)

.github/workflows/install_test.yml

@@ -12,7 +12,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest, macos-latest]
-        python-version: ['3.8', '3.9', '3.10', '3.11']
+        python-version: ['3.9', '3.10', '3.11', '3.12']
 
     steps:
     - uses: actions/checkout@v2

.github/workflows/install_test_coverage.yml

@@ -11,7 +11,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest, macos-latest]
-        python-version: ['3.8', '3.9', '3.10', '3.11']
+        python-version: ['3.9', '3.10', '3.11', '3.12']
 
     steps:
     - uses: actions/checkout@v2

.github/workflows/install_test_setup_py.yml

@@ -11,7 +11,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest, macos-latest]
-        python-version: ['3.8', '3.9', '3.10', '3.11']
+        python-version: ['3.9', '3.10', '3.11', '3.12']
 
     steps:
     - uses: actions/checkout@v2

.github/workflows/pylint_and_mypy.yml

@@ -12,10 +12,10 @@ jobs:
 
     steps:
     - uses: actions/checkout@v2
-    - name: Set up Python 3.8
+    - name: Set up Python 3.12
       uses: actions/setup-python@v1
       with:
-        python-version: 3.8
+        python-version: 3.12
     - name: Install poetry
       run: |
         python -m pip install poetry

.github/workflows/pypi_push.yml

@@ -8,10 +8,10 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@master
-      - name: Set up Python 3.8 🐍🐍🐍
+      - name: Set up Python 3.12 🐍🐍🐍
         uses: actions/setup-python@v1
         with:
-          python-version: 3.8
+          python-version: 3.12
       - name: Install poetry
         run: |
           python -m pip install poetry

libcloudforensics/providers/gcp/internal/compute.py

@@ -1382,10 +1382,16 @@ def GetBootDisk(self) -> 'GoogleComputeDisk':
 
     Raises:
       ResourceNotFoundError: If no boot disk could be found.
+      InvalidNameError: If the boot disk Source could not be found.
     """
 
     for disk in self.GetValue('disks'):
       if disk['boot']:
+        if 'source' not in disk:
+          raise errors.InvalidNameError(
+              'Boot disk has no source attribute: {0:s}'.format(
+                  self.name),
+              __name__)
         disk_name = disk['source'].split('/')[-1]
         return GoogleCloudCompute(self.project_id).GetDisk(disk_name=disk_name)
     raise errors.ResourceNotFoundError(
@@ -1406,7 +1412,7 @@ def GetDisk(self, disk_name: str) -> 'GoogleComputeDisk':
     """
 
     for disk in self.GetValue('disks'):
-      if disk['source'].split('/')[-1] == disk_name:
+      if disk.get('source', '').split('/')[-1] == disk_name:
         return GoogleCloudCompute(self.project_id).GetDisk(disk_name=disk_name)
     raise errors.ResourceNotFoundError(
         'Disk {0:s} was not found in instance {1:s}'.format(
@@ -1423,7 +1429,9 @@ def ListDisks(self) -> Dict[str, 'GoogleComputeDisk']:
 
     disks = {}
     disk_names = [
-        disk['source'].split('/')[-1] for disk in self.GetValue('disks')
+        disk['source'].split('/')[-1]
+        for disk in self.GetValue('disks')
+        if disk.get('source', None)
     ]
     for name in disk_names:
       disks[name] = self.GetDisk(name)
@@ -1507,7 +1515,7 @@ def DetachDisk(self, disk: 'GoogleComputeDisk') -> None:
     gce_instance_client = self.GceApi().instances() # pylint: disable=no-member
     device_name = None
     for disk_dict in self.GetValue('disks'):
-      if disk_dict['source'].split('/')[-1] == disk.name:
+      if disk_dict.get('source', '').split('/')[-1] == disk.name:
         device_name = disk_dict['deviceName']
     request = gce_instance_client.detachDisk(
         instance=self.name,
@@ -1541,7 +1549,9 @@ def Delete(
     disks_to_delete = []
     if delete_disks:
       disks_to_delete = [
-          disk['source'].split('/')[-1] for disk in self.GetValue('disks')
+          disk['source'].split('/')[-1]
+          for disk in self.GetValue('disks')
+          if disk.get('source', None)
       ]
 
     gce_instance_client = self.GceApi().instances() # pylint: disable=no-member

pyproject.toml

@@ -10,7 +10,7 @@ readme = "README.md"
 cloudforensics = "tools.cli:Main"
 
 [tool.poetry.dependencies]
-python = "^3.8"
+python = "^3.9"
 google-api-core = "*"
 azure-common = "^1.1.28"
 azure-core = "^1.29.4"
@@ -40,6 +40,7 @@ urllib3 = [
   {version = ">=1.25.4,<2.1", python = ">=3.10"}
 ]
 google-auth = "^2.22.0"
+setuptools = "^75.8.0"
 
 [tool.poetry.group.dev.dependencies]
 coverage = "^7.2.7"

tests/providers/gcp/gcp_mocks.py

@@ -240,6 +240,18 @@
         'initializeParams': {
             'diskName': FAKE_DISK.name
         }
+    },
+    {
+        'kind': 'compute#attachedDisk',
+        'type': 'SCRATCH',
+        'mode': 'READ_WRITE',
+        'savedState': 'DISK_SAVED_STATE_UNSPECIFIED',
+        'deviceName': 'local-ssd-0',
+        'index': 1,
+        'boot': False,
+        'autoDelete': True,
+        'interface': 'NVME',
+        'diskSizeGb': '375'
     }],
     'networkInterfaces': MOCK_NETWORK_INTERFACES,
     'metadata': {