Avoid key error for custom machine types

appengine/swarming/swarming_bot/api/os_utilities.py

Index: appengine/swarming/swarming_bot/api/os_utilities.py
diff --git a/appengine/swarming/swarming_bot/api/os_utilities.py b/appengine/swarming/swarming_bot/api/os_utilities.py
index b95d7fec21f530bd70014facc7f226f1fb4ca9c8..28710d916c0135d0bbd602314a48aaca87677906 100644
--- a/appengine/swarming/swarming_bot/api/os_utilities.py
+++ b/appengine/swarming/swarming_bot/api/os_utilities.py
@@ -412,9 +412,9 @@ def get_cost_hour():
   machine_type = get_machine_type()
   if platforms.is_gce():
     if platforms.gce.get_zone().startswith('us-'):
-      machine_cost = GCE_MACHINE_COST_HOUR_US[machine_type]
+      machine_cost = GCE_MACHINE_COST_HOUR_US.get(machine_type, 0.)

[borenet, 2017/04/07 12:19:25]

I suspect we'll want a default, eg. GCE_MACHINE_COST_HOUR_US['n1-standard-8']

[M-A Ruel, 2017/04/07 12:20:16]

Bah, I prefer 0 than something really wrong. I don't mind much.

     else:
-      machine_cost = GCE_MACHINE_COST_HOUR_EUROPE_ASIA[machine_type]
+      machine_cost = GCE_MACHINE_COST_HOUR_EUROPE_ASIA.get(machine_type, 0.)
   else:
     # Guess an equivalent machine_type.
     machine_cost = GCE_MACHINE_COST_HOUR_US.get(machine_type, 0.)