commit 6259d0c77b333f2ffb0e59fd68bd380822c98743
author Treehugger Robot <treehugger-gerrit@google.com> Wed Sep 16 06:53:16 2020 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Wed Sep 16 06:53:16 2020 +0000
tree adb5357454f5dd239481d2e493c745d8988d8c8a
parent 50a927857edbd20711f613fefbb28e6b7049b6cd
parent c11bf5f8c1049af2bcff1d9b2d372375c4cd0316

Merge "Add new error type for GCE quota issue error."

public/actions/common_operations.py

diff --git a/public/actions/common_operations.py b/public/actions/common_operations.py
index f5dbdb1..07a4a1e 100644
--- a/public/actions/common_operations.py
+++ b/public/actions/common_operations.py
@@ -39,6 +39,8 @@
     constants.STAGE_ARTIFACT: "ACLOUD_DOWNLOAD_ARTIFACT_ERROR",
     constants.STAGE_BOOT_UP: "ACLOUD_BOOT_UP_ERROR",
 }
+# Error type of GCE quota error.
+_GCE_QUOTA_ERROR = "GCE_QUOTA_ERROR"
 
 
 def CreateSshKeyPairIfNecessary(cfg):
@@ -275,7 +277,9 @@
                 reporter.AddError(str(failures[device.instance_name]))
             else:
                 reporter.AddData(key="devices", value=device_dict)
-    except errors.DriverError as e:
+    except (errors.DriverError, errors.CheckGCEZonesQuotaError) as e:
+        if isinstance(e, errors.CheckGCEZonesQuotaError):
+            reporter.UpdateData({_ERROR_TYPE: _GCE_QUOTA_ERROR})
         reporter.AddError(str(e))
         reporter.SetStatus(report.Status.FAIL)
     return reporter